1 files changed, 35 insertions, 31 deletions
diff --git a/__main__.py b/__main__.py
index e823054..98872c0 100755
--- a/__main__.py
+++ b/__main__.py
@@ -1,5 +1,7 @@
 #!/usr/bin/env python3
 
+""" Entry point to play the game """
+
 import argparse
 
 import pygame
@@ -11,47 +13,49 @@ from cameltris.screens.Pause import Pause as PauseScreen
 from cameltris.misc import Pause, UnPause
 
 
-PARSER = argparse.ArgumentParser()
+def main(args: dict):
+    """ Main function to be called with option arguments already parsed """
+    pygame.init()
 
-PARSER.add_argument("--starting-level", type=int, choices=list(range(1, 30)), default=1)
-PARSER.add_argument("--joystick", type=int, dest="joystick_id", metavar="Joystick ID")
+    screen = pygame.display.set_mode((801, 1000))
 
-ARGS = vars(PARSER.parse_args())
+    controller: Controller
+    if args["joystick_id"] is not None:
+        joystick = pygame.joystick.Joystick(args["joystick_id"])
+        joystick.init()
+        controller = JoystickController(joystick)
+    else:
+        controller = KeyboardController(pygame.key)
 
-print(ARGS)
+    # Just one player
+    players = [Player(controller, args["starting_level"])]
 
-pygame.init()
+    for player in players:
+        player.refresh_piece_preview_canvas()
 
-screen = pygame.display.set_mode((801, 1000))
+    clock = pygame.time.Clock()
 
-controller: Controller
-if ARGS["joystick_id"] is not None:
-    joystick = pygame.joystick.Joystick(ARGS["joystick_id"])
-    joystick.init()
-    controller = JoystickController(joystick)
-else:
-    controller = KeyboardController(pygame.key)
+    current_screen: Screen
+    current_screen = ingame_screen = InGameScreen(players, screen)
 
-# Just one player
-players = [Player(controller, ARGS["starting_level"])]
+    while True:
+        try:
+            current_screen.oneframe()
+        except Pause:
+            current_screen = PauseScreen(screen)
+        except UnPause:
+            current_screen = ingame_screen
 
-for player in players:
-    player.refresh_piece_preview_canvas()
+        current_screen.refresh()
+        pygame.display.update()
 
-clock = pygame.time.Clock()
+        clock.tick(60)
 
-current_screen: Screen
-current_screen = ingame_screen = InGameScreen(players, screen)
 
-while True:
-    try:
-        current_screen.oneframe()
-    except Pause:
-        current_screen = PauseScreen(screen)
-    except UnPause:
-        current_screen = ingame_screen
+if __name__ == "__main__":
+    PARSER = argparse.ArgumentParser()
 
-    current_screen.refresh()
-    pygame.display.update()
+    PARSER.add_argument("--starting-level", type=int, choices=list(range(1, 30)), default=1)
+    PARSER.add_argument("--joystick", type=int, dest="joystick_id", metavar="Joystick ID")
 
-    clock.tick(60)
+    main(vars(PARSER.parse_args()))