nvbn#414: Install and use win_unicode_console only on windows

setup.py

@@ -22,8 +22,9 @@
 
 VERSION = '3.2'
 
-install_requires = ['psutil', 'colorama', 'six', 'decorator', 'win_unicode_console']
-extras_require = {':python_version<"3.4"': ['pathlib']}
+install_requires = ['psutil', 'colorama', 'six', 'decorator']
+extras_require = {':python_version<"3.4"': ['pathlib'],
+                  ":sys_platform=='win32'": ['win_unicode_console']}
 
 setup(name='thefuck',
       version=VERSION,

thefuck/main.py

@@ -2,8 +2,6 @@
 from warnings import warn
 from pprint import pformat
 import sys
-import win_unicode_console
-win_unicode_console.enable() #https://github.com/tartley/colorama/issues/32
 import colorama
 from . import logs, types, shells
 from .conf import settings
@@ -13,9 +11,18 @@
 from .ui import select_command
 
 
+def init_colorama():
+    if sys.platform == 'win32':
+        #  https://github.com/tartley/colorama/issues/32
+        import win_unicode_console
+
+        win_unicode_console.enable()
+    colorama.init()
+
+
 def fix_command():
     """Fixes previous command. Used when `thefuck` called without arguments."""
-    colorama.init()
+    init_colorama()
     settings.init()
     with logs.debug_time('Total'):
         logs.debug(u'Run with settings: {}'.format(pformat(settings)))
@@ -53,7 +60,7 @@ def how_to_configure_alias():
     It'll be only visible when user type fuck and when alias isn't configured.
 
     """
-    colorama.init()
+    init_colorama()
     settings.init()
     logs.how_to_configure_alias(shells.how_to_configure())
 

thefuck/ui.py

@@ -7,6 +7,7 @@
 
 try:
     import msvcrt
+
     def getch():
         ch = msvcrt.getch()
         if ch in (b'\x00', b'\xe0'):  # arrow or function key prefix?