✨Add rich click (#307)

* Add rich click * Add rich-click to requirements * Update click version * Remove pyinstaller requirement * Add back pyinstaller --------- Co-authored-by: ayushuk <[email protected]>
purduesigbots · Jan 14, 2024 · 25ea823 · 25ea823
1 parent 26d19b7
commit 25ea823
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 3 deletions.
diff --git a/pros/cli/click_classes.py b/pros/cli/click_classes.py
@@ -1,13 +1,14 @@
 from collections import defaultdict
 from typing import *
 
+from rich_click import RichCommand
 import click.decorators
 from click import ClickException
 from pros.conductor.project import Project as p
 from pros.common.utils import get_version
 
 
-class PROSFormatted(click.BaseCommand):
+class PROSFormatted(RichCommand):
     """
     Common format functions used in the PROS derived classes. Derived classes mix and match which functions are needed
     """

diff --git a/pros/cli/main.py b/pros/cli/main.py
@@ -7,6 +7,7 @@
 import pros.common.sentry
 
 import click
+import ctypes
 import sys
 
 import pros.common.ui as ui
@@ -27,6 +28,10 @@
 import pros.cli.interactive
 import pros.cli.user_script
 
+if sys.platform == 'win32':
+    kernel32 = ctypes.windll.kernel32
+    kernel32.SetConsoleMode(kernel32.GetStdHandle(-11), 7)
+
 root_sources = [
     'build',
     'conductor',

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
-click>=6,<7
+click>=8
+rich-click
 pyserial
 cachetools
 requests
@@ -13,4 +14,4 @@ scan-build==2.0.13
 sentry-sdk
 observable
 pypng==0.0.20
-pyinstaller
+pyinstaller