feat: Snake game functional

Author: DavidMarquezF

Controller.py

@@ -0,0 +1,38 @@
+import curses
+from Snake import  Directions
+P1 = {ord("w"): Directions.up, ord("s"):Directions.down, ord("d"):Directions.right, ord("a"):Directions.left}
+P2 = {curses.KEY_UP: Directions.up, curses.KEY_DOWN:Directions.down, curses.KEY_RIGHT:Directions.right, curses.KEY_LEFT:Directions.left}
+P3 = {ord("u"): Directions.up, ord("j"):Directions.down, ord("k"):Directions.right, ord("h"):Directions.left}
+P4 = {ord("8"): Directions.up, ord("5"):Directions.down, ord("6"):Directions.right, ord("4"):Directions.left} #Num Block must be off to use the numPad
+
+class Controller(object):
+    maxPlayers = 4
+    def __init__(self, players = 1):
+        if players > Controller.maxPlayers:
+            raise Exception("Too many players")
+        self.players = players
+
+    def changePlayerNum(self, players):
+        self.players = players
+    def keypressed(self, key):
+        if key in P1:
+            return [1, P1[key]]
+
+        if key in P2:
+            return  [2 if self.players > 1 else 1, P2[key]]
+
+        if key in P3:
+            if self.players > 2:
+                return [3, P3[key]]
+            elif self.players > 1 and self.players < 3:
+                return -1
+            else:
+                return  [1, P3[key]]
+        if key in P4:
+            if self.players > 3:
+                return [4, P4[key]]
+            else:
+                return -1
+
+
+        return -1

Helper.py

@@ -0,0 +1,60 @@
+import curses
+#-- Define the appearance of some interface elements
+hotkey_attr = curses.A_BOLD | curses.A_UNDERLINE
+menu_attr = curses.A_NORMAL
+
+heightStatScreen=7
+
+
+MINIMUM_HEIGHT=34
+MINIMUM_WIDTH=52
+
+def createMenu(screen, menus, startX, startY):
+    for menu in menus:
+        menu_name = menu[0]
+        menu_hotkey = menu_name[0]
+        menu_no_hot = menu_name[1:]
+        screen.addstr(startY, startX, menu_hotkey, hotkey_attr)
+        screen.addstr(startY, startX+1, menu_no_hot, menu_attr)
+        startY+=1
+        # Add key handlers for this hotkey
+        menu_handler(screen, (str.upper(menu_hotkey), menu[1]))
+        menu_handler(screen, (str.lower(menu_hotkey), menu[1]))
+
+    screen.refresh()
+
+#-- Magic key handler both loads and processes keys strokes
+def menu_handler(screen, key_assign=None, key_dict={}):
+    if key_assign:
+        key_dict[ord(key_assign[0])] = key_assign[1]
+    else:
+        c = screen.getch()
+        if c in (curses.KEY_END, ord('!')):
+            return 0
+        elif c not in key_dict.keys():
+            curses.beep()
+            return 1
+        else:
+            return key_dict[c]()
+
+def checkIfEnoughSpaceForGame(stdscr, gameScreen = None):
+
+    maxY, maxX = stdscr.getmaxyx()
+
+    if (maxX < MINIMUM_WIDTH or maxY < MINIMUM_HEIGHT):
+        return True
+
+    if gameScreen==None:
+        return False
+
+    return maxY <= (gameScreen.sizeY + heightStatScreen + 1) or maxX <= (gameScreen.sizeX)
+
+
+class NotEnoughSpace(Exception):
+    pass
+
+class ExitTotal(Exception):
+    pass
+
+class GoBackToMainScreen(Exception):
+    pass

KeyboardInput.py

@@ -0,0 +1,72 @@
+global isWindows
+isWindows = False
+
+try:
+    from win32api import STD_INPUT_HANDLE
+    from win32console import GetStdHandle, KEY_EVENT, ENABLE_ECHO_INPUT, ENABLE_LINE_INPUT, ENABLE_PROCESSED_INPUT
+    isWindows = True
+except ImportError as e:
+    import sys, select, termios
+
+class KeyPoller(object):
+    def __enter__(self):
+        global isWindows
+        if(isWindows):
+            self.readHandle = GetStdHandle(STD_INPUT_HANDLE)
+            self.readHandle.SetConsoleMode(ENABLE_LINE_INPUT|ENABLE_ECHO_INPUT|ENABLE_PROCESSED_INPUT)
+
+            self.currEventLength = 0
+            self.currKeysLength = 0
+
+            self.capturedChars = []
+        else:
+            #Save terminal settings
+            self.fd = sys.stdin.fileno()
+            self.newTerm = termios.tcgetattr(self.fd)
+            self.oldTerm = termios.tcgetattr(self.fd)
+
+            #New terminal setting unbuffered
+            self.newTerm[3] = (self.newTerm[3] & ~termios.ICANON & ~termios.ECHO)
+            termios.tcsetattr(self.fd, termios.TCSAFLUSH, self.newTerm)
+        return self
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        if isWindows:
+            pass
+        else:
+            termios.tcsetattr(self.fd, termios.TCSAFLUSH, self.oldTerm)
+    def poll(self):
+        if(isWindows):
+            if(not len(self.capturedChars) == 0):
+                return self.capturedChars.pop(0)
+            eventsPeek = self.readHandle.PeekConsoleInput(10000)
+            if len(eventsPeek) == 0:
+                return None
+
+            if not len(eventsPeek) == self.currEventLength:
+                for curEvent in eventsPeek[self.currEventLength]:
+                    if(curEvent.EventType == KEY_EVENT):
+                        if(ord(curEvent.Char) == 0 or not curEvent.KeyDown):
+                            pass
+                        else:
+                            curChar = str(curEvent.Char)
+                            self.capturedChars.append(curChar)
+                self.currEventLength = len(eventsPeek)
+            if(not len(self.capturedChars) == 0):
+                return self.capturedChars.pop(0)
+            else:
+                return None
+        else:
+            dr,dw,de = select.select([sys.stdin], [], [], 0)
+            if(not dr == []):
+                return sys.stdin.read(1)
+            return None
+
+if __name__ == "__main__":
+    #Simle use case
+    with KeyPoller() as keyPoler:
+        while True:
+            c = keyPoler.poll()
+            if(not c is None):
+                if c == "c":
+                    break
+                print(c)