Games loading now in games

Author: PaulBrownMagic

README.md

@@ -19,6 +19,7 @@ Currently the project is developing the user interface and menu system before be
 - [ ] Single constants for config variables and game names (tidy up!)
 - [ ] Implement Logging
 - [ ] Set of unit tests for inputs, displays
+- [ ] Known bugs: Pacman pass-through ghosts. Purple rain: 2 in one pixel (-2 lives)
 
 \<LONG-TERM TODO\>:
 - [ ] Move menu screen to game class?

games/__init__.py

@@ -1 +1,13 @@
 from games.assets.sprite import Sprite, Co_ordinates
+from games import snake, purple_rain, pacman
+
+
+def load_game(name):
+    if name == "Snake":
+        return snake.Game()
+    elif name == "Purple Rain":
+        return purple_rain.Game()
+    elif name == "Pacman":
+        return pacman.Game()
+    else:
+        raise KeyError("Unknown Game Selected")

main.py

@@ -2,7 +2,7 @@
 import animations
 from clock import Clock
 from displays import load_display
-from games import pacman, purple_rain, snake
+from games import load_game
 from inputs import load_controller
 from menu import Menu
 
@@ -66,15 +66,11 @@ def menu_loop(self):
         while not self.menu.selected:
             self.frame()
         # Select chosen game
-        if self.menu.selected == "Snake":
-            self.game = snake.Game()
-        elif self.menu.selected == "Purple Rain":
-            self.game = purple_rain.Game()
-        elif self.menu.selected == "Pacman":
-            self.game = pacman.Game()
-        elif self.menu.selected == "Exit":
+        if self.menu.selected == "Exit":
             self.state = "Exit"
             return
+        else:
+            self.game = load_game(self.menu.selected)
         # Load new game into view
         self.menu.selected = None
         self.view = self.game
@@ -93,8 +89,7 @@ def program_loop(self):
         arcade = Arcade()
         arcade.program_loop()
     except Exception as e:
-        print("\nERROR: {}\n\n".format(e))
-        print(e.__traceback__)
+        print("\nERROR: {}\n\n".format(e.with_traceback(e)))
     finally:
         arcade.controller.cleanup()
         arcade.display.clear()