Use colorama on Windows

wkentaro · wkentaro · commit 83f892794519 · 2020-07-15T12:32:11.000+01:00
diff --git a/labelme/logger.py b/labelme/logger.py
@@ -1,8 +1,14 @@
 import datetime
 import logging
+import os
 
 import termcolor
 
+if os.name == "nt":  # Windows
+    import colorama
+
+    colorama.init()
+
 from . import __appname__
 
 
diff --git a/setup.py b/setup.py
@@ -1,13 +1,15 @@
 from __future__ import print_function
 
 import distutils.spawn
+import os
 import re
-from setuptools import find_packages
-from setuptools import setup
 import shlex
 import subprocess
 import sys
 
+from setuptools import find_packages
+from setuptools import setup
+
 
 def get_version():
     filename = "labelme/__init__.py"
@@ -75,6 +77,9 @@ def get_install_requires():
             QT_BINDING = "pyqt5"
     del QT_BINDING
 
+    if os.name == "nt":  # Windows
+        install_requires.append("colorama")
+
     return install_requires
 
 
