Formatting

obs_client.py

@@ -1,16 +1,17 @@
-import obsws_python as obs
-import click
-
 from time import sleep
 
-from obs_tools.sc2client import sc2client
-from config import config
-from obs_tools.types import UIInfo, Screen
+import click
+import obsws_python as obs
 from rich import print
 
+from config import config
+from obs_tools.sc2client import sc2client
+from obs_tools.types import Screen, UIInfo
+
 
 @click.command()
 def main():
+    """Monitor SC2 UI through client API and let OBS know when loading screen is active"""
     with obs.ReqClient(
         host="localhost", port=4455, password=config.obs_ws_pw, timeout=3
     ) as cl:

obs_tools/sc2client.py

@@ -9,7 +9,7 @@
 
 from config import config
 
-from .types import GameInfo, Result, ScanResult, UIInfo, Screen
+from .types import GameInfo, Result, ScanResult, Screen, UIInfo
 
 log = logging.getLogger(f"{config.name}.{__name__}")
 
@@ -43,12 +43,12 @@ def get_opponent_name(self, gameinfo=None) -> str:
                 if player.name != config.student.name:
                     return player.name
         return None
-    
+
     def get_screens(self) -> UIInfo:
         try:
             response = requests.get(config.sc2_client_url + "/ui")
             if response.status_code == 200:
-                try: 
+                try:
                     ui = UIInfo.model_validate_json(response.text)
                     return ui
                 except ValidationError as e: