From 821d33ac49b79ff695674706136473824b1aa25b Mon Sep 17 00:00:00 2001
From: Willi Ballenthin <willi.ballenthin@gmail.com>
Date: Wed, 24 May 2023 11:57:18 +0200
Subject: [PATCH] qs: try to fix color output on CI

---
 floss/qs/main.py | 2 ++
 setup.py         | 1 +
 2 files changed, 3 insertions(+)

diff --git a/floss/qs/main.py b/floss/qs/main.py
index c4d105c86..5cc1007c1 100644
--- 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
index 58d301536..fc74151ce 100644
--- 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",
         ],