import pygame
import random


class Zombie(pygame.sprite.Sprite):
    def __init__(self, name):
        super(Zombie, self).__init__()
        self.image = pygame.image.load('material/images/Zombie_0.png').convert_alpha()
        self.images = [pygame.image.load('material/images/Zombie_{}.png'.format(i)).convert_alpha() for i in
                       range(0, 22)]
        self.dieimages = [pygame.image.load('material/images/ZombieDie_{}.png'.format(i)).convert_alpha() for i in
                          range(0, 10)]
        self.attackimages = [pygame.image.load('material/images/ZombieAttack_{}.png'.format(i)).convert_alpha() for i in
                          range(0, 21)]
        self.rect = self.images[0].get_rect()
        self.rect.top = 25 + random.randrange(0, 4) * 125
        # print(self.rect.top)
        self.rect.left = 1000
        self.speed = 2
        self.name = name
        self.energy = 6
        self.dietimes = 0
        self.isMeetWallNut = False #僵尸是否遇到了坚果
        self.isAlive = True

    def update(self, *args):
        if self.energy > 0:
            if self.isMeetWallNut:
                self.image = self.attackimages[args[0] % len(self.attackimages)]
            else:
                self.image = self.images[args[0] % len(self.images)]

            if self.rect.left > 250 and not self.isMeetWallNut:
                self.rect.left -= self.speed
        else:
            if self.dietimes > 9:
                if self.dietimes > 24:
                    self.kill()
                else:
                    self.isAlive = False
                    self.dietimes += 1
            else:
                self.image = self.dieimages[self.dietimes]
                self.dietimes += 1