Parcourir la source

Use colorama on Windows

Kentaro Wada il y a 4 ans
Parent
commit
83f8927945
2 fichiers modifiés avec 13 ajouts et 2 suppressions
  1. 6 0
      labelme/logger.py
  2. 7 2
      setup.py

+ 6 - 0
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__
 
 

+ 7 - 2
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