Run black on rewrite

rewrite/config/__init__.py

@@ -1,7 +1,7 @@
 from utils.font import FontCache
 
-class Config:
 
+class Config:
     def __init__(self):
         self.font_cache = FontCache()
 

rewrite/data/__init__.py

@@ -1,7 +1,7 @@
 from data.schedule import Schedule
 
-class Data:
 
+class Data:
     def __init__(self, screen_manager):
         self.screen_manager = screen_manager
 

rewrite/data/game.py

@@ -1,7 +1,7 @@
 from data.status import Status
 
-class Game:
 
+class Game:
     @staticmethod
     def from_schedule(game_data):
         game = Game(game_data)

rewrite/data/schedule.py

@@ -7,9 +7,9 @@
 
 from screens import Screen
 
-class Schedule:
 
-    REFRESH_RATE = 5 * 60 # minutes
+class Schedule:
+    REFRESH_RATE = 5 * 60  # minutes
 
     def __init__(self, data):
         self.data = data
@@ -25,7 +25,7 @@ def update(self):
         self.last_update = time.time()
 
         ScoreboardLogger.log(f"Updating schedule for {datetime.datetime.today()}")
-        
+
         try:
             self.__fetch_updated_schedule(datetime.datetime.today())
         except Exception as exception:
@@ -36,7 +36,7 @@ def update(self):
 
         self.data.request_next_screen(Screen.GAME)
 
-        return Status.SUCCESS 
+        return Status.SUCCESS
 
     def __fetch_updated_schedule(self, date):
         self._games = statsapi.schedule(date.strftime("%Y-%m-%d"))

rewrite/data/screen_request.py

@@ -1,17 +1,17 @@
 from screens import Screen
 
-class ScreenRequest:
 
+class ScreenRequest:
     class InvalidRequest(Exception):
         pass
 
     def __init__(self, type, manager, *args):
         # Required arguments
-        self.type    = type
+        self.type = type
         self.manager = manager
 
         # Screen-specific arguments
-        self.args    = args
+        self.args = args
 
         if self.type not in Screen:
             raise ScreenRequest.InvalidRequest(f"Screen type {self.type} is not valid")

rewrite/data/status.py

@@ -10,5 +10,6 @@ class Status(Enum):
 def ok(status):
     return status in [Status.SUCCESS, Status.DEFERRED]
 
+
 def fail(status):
     return status in [Status.FAIL]

rewrite/driver/__init__.py

@@ -3,6 +3,7 @@
 from utils import args
 from driver.mode import DriverMode
 
+
 class DriverWrapper:
     def __init__(self):
         self.hardware_load_failed = False
@@ -13,10 +14,9 @@ def __init__(self):
         else:
             self.set_mode(DriverMode.HARDWARE)
 
-
     @property
     def __name__(self):
-        return 'driver'
+        return "driver"
 
     def is_hardware(self):
         return self.mode == DriverMode.HARDWARE
@@ -47,4 +47,4 @@ def __getattr__(self, name):
         return getattr(self.driver, name)
 
 
-sys.modules['driver'] = DriverWrapper()
+sys.modules["driver"] = DriverWrapper()

rewrite/driver/mode.py

@@ -2,5 +2,5 @@
 
 
 class DriverMode(Enum):
-    HARDWARE           = 0
+    HARDWARE = 0
     SOFTWARE_EMULATION = 1

rewrite/main.py

@@ -26,7 +26,7 @@
 
 
 def main(matrix):
-    #TODO: Configure matrix dimensions
+    # TODO: Configure matrix dimensions
     ScoreboardLogger.info(f"{SCRIPT_NAME} - v#{SCRIPT_VERSION} (32x32)")
 
     canvas = matrix.CreateFrameCanvas()
@@ -35,11 +35,7 @@ def main(matrix):
     config = Config()
     screen_manager = ScreenManager(matrix, canvas, config, screen_queue)
 
-    render_thread = threading.Thread(
-        target=screen_manager.start,
-        name="render_thread",
-        daemon=True
-    )
+    render_thread = threading.Thread(target=screen_manager.start, name="render_thread", daemon=True)
 
     render_thread.start()
 

rewrite/screens/__init__.py

@@ -1,9 +1,8 @@
 from enum import Enum
 
+
 class Screen(Enum):
-    STATIC  = 0
-    CLOCK   = 1
+    STATIC = 0
+    CLOCK = 1
     WEATHER = 2
-    GAME    = 3
-
-from screens.screen_manager import ScreenManager
+    GAME = 3

rewrite/screens/base.py

@@ -1,13 +1,13 @@
 from datetime import datetime as dt
 
-class ScreenBase():
 
+class ScreenBase:
     def __init__(self, manager):
         self.manager = manager
 
         self.start_time = None
         self.duration = 0
-    
+
     def track_duration(fn):
         def wrapper(self, *args, **kwargs):
             if self.start_time is None:
@@ -21,21 +21,21 @@ def wrapper(self, *args, **kwargs):
 
     def render(self):
         raise NotImplementedError("Subclasses must implement render()")
-    
+
     def ready_to_transition(self):
-        '''
-        Returns True if the screen is ready to transition away.
+        """
+        Returns True if the screen is ready to transition away. By default, always allow the screen to transition.
 
-        Subclasses should implement this method if they desire different behavior.
+        Subclasses should implement this method if they desire different behavior, such as requiring screens to be visible for a certain duration.
 
         NOTE: If this value always returns False, no screen transitions will happen!
-        '''
+        """
         return True
-    
+
     @track_duration
     def _render(self):
         self.render()
-    
+
     @property
     def matrix(self):
         return self.manager.matrix
@@ -47,7 +47,7 @@ def canvas(self):
     @property
     def data(self):
         return self.manager.data
-    
+
     @property
     def config(self):
         return self.manager.config

rewrite/screens/clock.py

@@ -7,7 +7,6 @@
 
 
 class ClockScreen(ScreenBase):
-
     MAX_DURATION_SECONDS = 3
 
     def render(self):

rewrite/screens/game.py

@@ -7,7 +7,6 @@
 
 
 class GameScreen(ScreenBase):
-
     MAX_DURATION_SECONDS = 3
 
     def render(self):

rewrite/screens/screen_manager.py

@@ -8,21 +8,21 @@
 from screens.weather import WeatherScreen
 from screens.game import GameScreen
 
-class ScreenManager():
 
+class ScreenManager:
     SCREENS = {
-        Screen.STATIC:  StaticScreen,
-        Screen.CLOCK:   ClockScreen,
+        Screen.STATIC: StaticScreen,
+        Screen.CLOCK: ClockScreen,
         Screen.WEATHER: WeatherScreen,
-        Screen.GAME:    GameScreen
+        Screen.GAME: GameScreen,
     }
 
     def __init__(self, matrix, canvas, config, queue):
         self.matrix = matrix
         self.canvas = canvas
         self.config = config
         self.data = None
-        self.queue  = queue
+        self.queue = queue
         self.screen = StaticScreen(self)
 
     def start(self):
@@ -46,35 +46,39 @@ def __handle_screen_request(self):
         request = self.queue.get()
 
         if not isinstance(request, ScreenRequest):
-            ScoreboardLogger.warning(f"Screen manager received transition request for instance of {type(request)}, but must be a ScreenRequest")
+            ScoreboardLogger.warning(
+                f"Screen manager received transition request for instance of {type(request)}, but must be a ScreenRequest"
+            )
 
             return False
 
         screen_class = self.SCREENS.get(request.type, None)
 
         if not screen_class:
-            ScoreboardLogger.warning(f"Screen manager received transition request for screen type of {request.type}, but no such screen type exists")
+            ScoreboardLogger.warning(
+                f"Screen manager received transition request for screen type of {request.type}, but no such screen type exists"
+            )
 
             return False
-        
+
         try:
             screen = screen_class(request.manager, *request.args)
         except Exception as exception:
             ScoreboardLogger.exception(exception)
             ScoreboardLogger.exception("Screen manager failed to process screen transition!")
 
             return False
-        
+
         # If not ready to transition, re-queue the request for later.
         if not self.screen.ready_to_transition():
             self.queue.put(request)
 
             return False
-        
+
         self.__perform_transition(screen)
-        
+
         return True
-            
+
     def __perform_transition(self, screen):
         self.screen = screen
 

rewrite/screens/static.py

@@ -4,18 +4,17 @@
 
 from screens.base import ScreenBase
 
+
 class StaticScreen(ScreenBase):
 
-    '''
+    """
     This screen is used to display a static image on startup before real data is loaded.
-    '''
+    """
 
     MIN_DURATION_SECONDS = 3
 
     # TODO: Pull this from config.
-    LOGO_PATH = os.path.abspath(
-        os.path.join(__file__, "./../../assets/logo/mlb-w32h32.png")
-    )
+    LOGO_PATH = os.path.abspath(os.path.join(__file__, "./../../assets/logo/mlb-w32h32.png"))
 
     def __init__(self, *args):
         super(self.__class__, self).__init__(*args)
@@ -27,7 +26,7 @@ def render(self):
         pass
 
     def ready_to_transition(self):
-        '''
+        """
         If the static load screen is displayed, then leave it on screen for a few seconds to avoid flashing.
-        '''
+        """
         return self.duration >= self.MIN_DURATION_SECONDS * 1000

rewrite/screens/weather.py

@@ -7,7 +7,6 @@
 
 
 class WeatherScreen(ScreenBase):
-
     MAX_DURATION_SECONDS = 3
 
     def __init__(self, *args):

rewrite/utils/__init__.py

@@ -2,6 +2,7 @@
 
 from utils import logger as ScoreboardLogger
 
+
 def args():
     parser = argparse.ArgumentParser()
 
@@ -101,7 +102,11 @@ def args():
         type=int,
     )
     parser.add_argument(
-        "--led-pwm-dither-bits", action="store", help="Time dithering of lower bits (Default: 0)", default=0, type=int,
+        "--led-pwm-dither-bits",
+        action="store",
+        help="Time dithering of lower bits (Default: 0)",
+        default=0,
+        type=int,
     )
     parser.add_argument(
         "--config",
@@ -111,10 +116,7 @@ def args():
         type=str,
     )
     parser.add_argument(
-        "--emulated",
-        action="store_const",
-        help="Force using emulator mode over default matrix display.",
-        const=True
+        "--emulated", action="store_const", help="Force using emulator mode over default matrix display.", const=True
     )
     return parser.parse_args()
 

rewrite/utils/font.py

@@ -2,8 +2,8 @@
 
 from driver import graphics
 
-class FontCache:
 
+class FontCache:
     FONT_PATHS = ["../assets/fonts/patched", "../submodules/matrix/fonts"]
 
     def __init__(self):