Подкласс

Часть раздела Спрайты и изображения путешествия по Pygame на Coddy. Урок 15 из 36.

Некоторые враги должны двигаться зигзагом. Они падают, как и остальные, а также скользят вбок, разворачиваясь у краёв окна. Если скопировать весь класс Enemy, чтобы изменить один метод, придётся исправлять каждую ошибку дважды. Подкласс наследует всё от родительского класса и меняет только то, что отличается:

class Zigzag(Enemy):
    def __init__(self, image, x, y):
        super().__init__(image, x, y)
        self.dx = 3

    def update(self):
        super().update()
        self.rect.x += self.dx
        if self.rect.left < 0 or self.rect.right > 640:
            self.dx = -self.dx

class Zigzag(Enemy) означает, что Zigzag — это Enemy. У него есть draw, хотя этот метод не нужно писать. Его собственный метод update заменяет родительский, а super().update() сначала запускает родительскую версию, поэтому падение и возвращение наверх не приходится писать дважды. super().__init__(...) делает то же самое для настройки.

Zigzag помещается в тот же список, что и остальные: цикл вызывает enemy.update(), и каждый объект выполняет свой вариант обновления.

Задание

Средне

Напишите class Zigzag(Enemy): его __init__ вызывает super().__init__(image, x, y) и задаёт self.dx = 3.
Его update вызывает super().update(), прибавляет self.dx к self.rect.x и меняет знак self.dx, когда Rect выступает за левый или правый край.
Сделайте жёлтого врага в enemies экземпляром Zigzag; красный и зелёный остаются Enemy.

Попробуйте сами

import pygame

pygame.init()
screen = pygame.display.set_mode((640, 480))
pygame.display.set_caption("Enemy Rain")
clock = pygame.time.Clock()

red = pygame.image.load("assets/images/enemy_red.png").convert_alpha()
green = pygame.image.load("assets/images/enemy_green.png").convert_alpha()
yellow = pygame.image.load("assets/images/enemy_yellow.png").convert_alpha()

class Enemy:
    def __init__(self, image, x, y):
        self.image = image
        self.rect = image.get_rect(center=(x, y))
        self.speed = 2

    def update(self):
        self.rect.y += self.speed
        if self.rect.top > 480:
            self.rect.bottom = 0

    def draw(self, screen):
        screen.blit(self.image, self.rect)


enemies = [Enemy(red, 160, 100), Enemy(green, 320, 100), Enemy(yellow, 480, 100)]

running = True
while running:
    for event in pygame.event.get():
        if event.type == pygame.QUIT:
            running = False

    for enemy in enemies:
        enemy.update()

    screen.fill((20, 22, 40))
    for enemy in enemies:
        enemy.draw(screen)
    pygame.display.flip()
    clock.tick(60)

pygame.quit()