helicopter.py

import sprites


class Helicopter(object):

    health = 3

    animation_number = 0
    animation_list = sprites.helicopter_list
    counter = 0
    current = sprites.helicopter_list[animation_number]

    crash_counter = 0

    damaged_counter = 0
    damaged = False

    wreck_start = False
    wrecked = False

    x = 0
    y = 0

    moving_up = False
    moving_left = False
    moving_down = False
    moving_right = False

    next_0 = True
    next_1 = False

    def __init__(self, x, y):
        self.x = x
        self.y = y

    def wreck(self):
        self.wreck_start = True
        self.crash_counter += 1
        if 5 <= self.crash_counter < 10:
            self.current = sprites.helicopter_crash_1
        if 10 <= self.crash_counter < 15:
            self.current = sprites.helicopter_crash_2
        if 15 <= self.crash_counter < 20:
            self.current = sprites.helicopter_crash_3
        if self.crash_counter >= 20:
            self.current = sprites.helicopter_crash_4
            self.crash_counter = 0
            self.wreck_start = False
            self.wrecked = True

    def blink_red(self):
        self.animation_list = sprites.damaged_helicopter_list
        self.damaged_counter += 1
        if self.damaged_counter >= 15:
            self.animation_list = sprites.helicopter_list
            self.damaged = False
            self.damaged_counter = 0

    def movement(self):

        speed = 10

        if not self.wreck_start:

            if (self.moving_up and self.moving_left) or (self.moving_down and self.moving_left):
                speed *= 0.707
            if (self.moving_up and self.moving_right) or (self.moving_down and self.moving_right):
                speed *= 0.707

            if self.moving_up:
                self.y -= speed
            if self.moving_left:
                self.x -= speed
            if self.moving_down:
                self.y += speed
            if self.moving_right:
                self.x += speed*2

            if self.x > 200:
                self.x -= speed*2
            elif self.x > 100:
                self.x -= speed/2

            if self.x < 0:
                self.x += speed
            elif self.x < 100:
                self.x += speed/2

        if self.y < 0:
            self.y = 0

        if self.y > 430:
            self.health = 0

    def animation(self):

        self.counter += 1

        if self.counter == 2:

            if self.next_0:
                self.current = self.animation_list[0]
                self.next_0 = False
                self.next_1 = True
            elif self.next_1:
                self.current = self.animation_list[1]
                self.next_1 = False
                self.next_0 = True

            self.counter = 0

    def player_init(self):
        self.animation()
        self.movement()
        if self.damaged:
            self.blink_red()