wall.py

import pygame


class Wall(pygame.sprite.Sprite):

    def __init__(self, screen, health = 4):
        super(Wall, self).__init__()
        self.screen = screen
        self.screen_rect = screen.get_rect()
        self.image = pygame.image.load(f"icons/wall_{health}.png")
        self.rect = self.image.get_rect()
        self.rect.centerx = 100
        self.health = health

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

    def update(self):
        self.health -= 1
        if self.health >= 0:
            self.image = pygame.image.load(f"icons/wall_{self.health}.png")
            self.screen.blit(self.image, self.rect)
        else:
            self.kill()