qs: try to fix color output on CI

mandiant · May 24, 2023 · 821d33a · 821d33a
1 parent 9d88936
commit 821d33a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/floss/qs/main.py b/floss/qs/main.py
@@ -13,6 +13,7 @@
 from dataclasses import dataclass
 
 import pefile
+import colorama
 import lancelot
 import vivisect
 import viv_utils
@@ -474,6 +475,7 @@ def main():
         logging.getLogger().setLevel(logging.INFO)
 
     rich.traceback.install()
+    colorama.just_fix_windows_console()
 
     path = pathlib.Path(args.path)
     if not path.exists():

diff --git a/setup.py b/setup.py
@@ -57,6 +57,7 @@
     install_requires=requirements,
     extras_require={
         "qs": [
+            "colorama==0.4.5",
             "intervaltree==3.1.0",
             "python-lancelot==0.8.6",
         ],