save project

snake.py

+import copy
+import random
+import os
+import msvcrt
+
+# the board class, used to put everything
+class board:
+
+    __points =[]
+
+    def __init__(self):
+        self.__points.clear()
+        for i in range(22):
+            line = []
+            if i == 0 or i == 21:
+                for j in range(22):
+                    line.append('#')
+            else:
+                line.append('#')
+                for j in range(20):
+                    line.append(' ')
+                line.append('#')
+            self.__points.append(line)
+
+    def getPoint(self, location):
+        return self.__points[location[0]][location[1]]
+
+    def clear(self):
+        self.__points.clear()
+        for i in range(22):
+            line = []
+            if i == 0 or i == 21:
+                for j in range(22):
+                    line.append('#')
+            else:
+                line.append('#')
+                for j in range(20):
+                    line.append(' ')
+                line.append('#')
+            self.__points.append(line)
+
+    def put_snake(self, snake_locations):
+        # clear the board
+        self.clear()
+
+        # put the snake points
+        for x in snake_locations:
+            self.__points[x[0]][x[1]] = 'o'
+
+        # the head
+        x = snake_locations[len(snake_locations) - 1]
+        self.__points[x[0]][x[1]] = 'O'
+
+    def put_food(self, food_location):
+        self.__points[food_location[0]][food_location[1]] = '*'
+
+    def show(self):
+        os.system("cls")
+        for i in range(22):
+            for j in range(22):
+                print(self.__points[i][j], end='')
+            print()
+
+# the snake class
+class snake:
+    __points = []
+
+    def __init__(self):
+        for i in range(1, 6):
+            self.__points.append([1, i])
+
+    def getPoints(self):
+        return self.__points
+
+    # move to the next position
+    # give the next head
+    def move(self, next_head):
+        self.__points.pop(0)
+        self.__points.append(next_head)
+
+    # eat the food
+    # give the next head
+    def eat(self, next_head):
+        self.__points.append(next_head)
+
+    # calc the next state
+    # and return the direction
+    def next_head(self, direction='default'):
+
+        # need to change the value, so copy it
+        head = copy.deepcopy(self.__points[len(self.__points) - 1])
+
+        # calc the "default" direction
+        if direction == 'default':
+            neck = self.__points[len(self.__points) - 2]
+            if neck[0] > head[0]:
+                direction = 'up'
+            elif neck[0] < head[0]:
+                direction = 'down'
+            elif neck[1] > head[1]:
+                direction = 'left'
+            elif neck[1] < head[1]:
+                direction = 'right'
+
+        if direction == 'up':
+            head[0] = head[0] - 1
+        elif direction == 'down':
+            head[0] = head[0] + 1
+        elif direction == 'left':
+            head[1] = head[1] - 1
+        elif direction == 'right':
+            head[1] = head[1] + 1
+        return head
+
+# the game
+class game:
+
+    board = board()
+    snake = snake()
+    food = []
+    count = 0
+
+    def __init__(self):
+        self.new_food()
+        self.board.clear()
+        self.board.put_snake(self.snake.getPoints())
+        self.board.put_food(self.food)
+
+    def new_food(self):
+        while 1:
+            line = random.randint(1, 20)
+            column = random.randint(1, 20)
+            if self.board.getPoint([column, line]) == ' ':
+                self.food = [column, line]
+                return
+
+    def show(self):
+        self.board.clear()
+        self.board.put_snake(self.snake.getPoints())
+        self.board.put_food(self.food)
+        self.board.show()
+
+
+    def run(self):
+        self.board.show()
+
+        # the 'w a s d' are the directions
+        operation_dict = {b'w': 'up', b'W': 'up', b's': 'down', b'S': 'down', b'a': 'left', b'A': 'left', b'd': 'right', b'D': 'right'}
+        op = msvcrt.getch()
+
+        while op != b'q':
+            if op not in operation_dict:
+                op = msvcrt.getch()
+            else:
+                new_head = self.snake.next_head(operation_dict[op])
+
+                # get the food
+                if self.board.getPoint(new_head) == '*':
+                    self.snake.eat(new_head)
+                    self.count = self.count + 1
+                    if self.count >= 15:
+                        self.show()
+                        print("Good Job")
+                        break
+                    else:
+                        self.new_food()
+                        self.show()
+
+                # 反向一Q日神仙
+                elif new_head == self.snake.getPoints()[len(self.snake.getPoints()) - 2]:
+                    pass
+
+                # rush the wall
+                elif self.board.getPoint(new_head) == '#' or self.board.getPoint(new_head) == 'o':
+                    print('GG')
+                    break
+
+                # normal move
+                else:
+                    self.snake.move(new_head)
+                    self.show()
+            op = msvcrt.getch()
+
+game().run()
\ No newline at end of file