|
1 | 1 | import os |
2 | 2 | import subprocess |
| 3 | +from platform import system as _curos |
| 4 | +CUR_OS = _curos() |
| 5 | +IS_WIN = CUR_OS in ['Windows', 'cli'] |
| 6 | +IS_NIX = (not IS_WIN) and any( |
| 7 | + CUR_OS.startswith(i) for i in |
| 8 | + ['CYGWIN', 'MSYS', 'Linux', 'Darwin', 'SunOS', 'FreeBSD']) |
| 9 | + |
3 | 10 |
|
4 | 11 | try: # pragma: no cover |
5 | 12 | _range = xrange |
|
17 | 24 | except NameError: # pragma: no cover |
18 | 25 | _unicode = str # in Py3, all strings are unicode |
19 | 26 |
|
| 27 | + |
20 | 28 | try: # pragma: no cover |
21 | | - import colorama |
22 | | - colorama.init() |
| 29 | + if IS_WIN: |
| 30 | + import colorama |
| 31 | + colorama.init() |
| 32 | + else: |
| 33 | + colorama = None |
23 | 34 | except ImportError: # pragma: no cover |
24 | 35 | colorama = None |
25 | 36 | except: # pragma: no cover |
@@ -49,15 +60,12 @@ def _environ_cols_wrapper(): # pragma: no cover |
49 | 60 | Return a function which gets width and height of console |
50 | 61 | (linux,osx,windows,cygwin). |
51 | 62 | """ |
52 | | - import platform |
53 | | - current_os = platform.system() |
54 | 63 | _environ_cols = None |
55 | | - if current_os in ['Windows', 'cli']: |
| 64 | + if IS_WIN: |
56 | 65 | _environ_cols = _environ_cols_windows |
57 | 66 | if _environ_cols is None: |
58 | 67 | _environ_cols = _environ_cols_tput |
59 | | - if any(current_os.startswith(i) for i in |
60 | | - ['CYGWIN', 'MSYS', 'Linux', 'Darwin', 'SunOS', 'FreeBSD']): |
| 68 | + if IS_NIX: |
61 | 69 | _environ_cols = _environ_cols_linux |
62 | 70 | return _environ_cols |
63 | 71 |
|
|
0 commit comments