import random


class Snake():
    def __init__(self, settings):
        self.reset(settings)

    # 蛇的初始化
    def reset(self, settings):
        # 蛇头的坐标
        self.start_x = 5#random.randint(5, settings.cell_w - 6)
        self.start_y = 5#random.randint(5, settings.cell_h - 6)
        self.head_index = 0
        # 蛇的初始运动方向
        self.direction = 'right'
        # 蛇的初始坐标字典，初始蛇为蛇头位置及其左边的两个格子
        self.coords = [{'x': self.start_x, 'y': self.start_y},
                       {'x': self.start_x - 1, 'y': self.start_y},
                       {'x': self.start_x - 2, 'y': self.start_y}]
    # 更新蛇，每次蛇移动一步，因此相当于每次移动在蛇的坐标字典中再插入一个新蛇头
    # 按常识，每次移动后，不仅仅是蛇头向前移动一个，更重要的是蛇尾也要向前移，但是在该函数中先不处理蛇尾
    # 因为如果蛇吃了食物，此时蛇尾其实并没有移动

    def move_head(self):
        newHead = {}
        # 根据移动方向确定蛇头
        if self.direction == 'up':
            newHead = {'x': self.coords[self.head_index]['x'],
                    'y': self.coords[self.head_index]['y'] - 1}
        elif self.direction == 'down':
            newHead = {'x': self.coords[self.head_index]['x'],
                    'y': self.coords[self.head_index]['y'] + 1}
        elif self.direction == 'left':
            newHead = {'x': self.coords[self.head_index]['x'] - 1,
                    'y': self.coords[self.head_index]['y']}
        elif self.direction == 'right':
            newHead = {'x': self.coords[self.head_index]['x'] + 1,
                    'y': self.coords[self.head_index]['y']}
        self.coords.insert(0, newHead)

    def create_iterator(self,path):
        def automove_head():
            for newhead in path:
                self.coords.insert(0, newhead)
        return automove_head
    
    def automove_head(self,ai):
        ai.iterator()