# Write By Guobao
# 2017/4//7
#
# 贪吃蛇
# 用#做边界，*做食物，o做身体和头部
# python 3.6.1

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()