Listen for clicks, detect which display was clicked

squirrelbattle/enums.py

@@ -16,6 +16,7 @@ class DisplayActions(Enum):
     """
     REFRESH = auto()
     UPDATE = auto()
+    MOUSE = auto()
 
 
 class GameMode(Enum):
@@ -32,6 +33,7 @@ class KeyValues(Enum):
     """
     Key values options used in the game
     """
+    MOUSE = auto()
     UP = auto()
     DOWN = auto()
     LEFT = auto()
@@ -48,8 +50,10 @@ class KeyValues(Enum):
         """
         Translate the raw string key into an enum value that we can use.
         """
-        if key in (settings.KEY_DOWN_SECONDARY,
-                   settings.KEY_DOWN_PRIMARY):
+        if key == "KEY_MOUSE":
+            return KeyValues.MOUSE
+        elif key in (settings.KEY_DOWN_SECONDARY,
+                     settings.KEY_DOWN_PRIMARY):
             return KeyValues.DOWN
         elif key in (settings.KEY_LEFT_PRIMARY,
                      settings.KEY_LEFT_SECONDARY):