decor.py

import pygame

from config import *


class Ground:

    def __init__(self):
        pass

    def draw(self, surf: pygame.Surface):
        pygame.draw.line(surf, (0, 255, 0), (0, WORLD_DIM[1] - 1), (WORLD_DIM[0] - 1, WORLD_DIM[1] - 1))


class Barricade:

    def __init__(self, center):
        self.sprite = pygame.image.load(SPRITE_PATH + BARRICADE_SPRITE_NAME)
        self.rect = self.sprite.get_rect(center=center)
        self.mask = pygame.mask.from_threshold(self.sprite, (0, 0, 0, 0), (1, 1, 1, 255))
        self.mask.invert()

    def draw(self, surf: pygame.Surface):
        surf.blit(self.sprite, self.rect)

class Barricades:

    def __init__(self):
        self.barricade_list = [Barricade(b_pos) for b_pos in BARRICADE_POSITIONS]

    def draw(self, surf: pygame.Surface):
        for b in self.barricade_list:
            b.draw(surf)

    def __iter__(self):
        return self.barricade_list.__iter__()

    def __next__(self):
        return next(self.__iter__())