level.py

from pathlib import Path
from typing import List

from pygame import Rect

from interfaces import MapObject
from enemies import Common, Fast, RapidFire, Armored
from landscape import Brick, Steel, Grass, Water, Ice


class MapCreator:
    block_dict = {'B': Brick,
                  'S': Steel,
                  'W': Water,
                  'G': Grass,
                  'I': Ice}

    def __init__(self, level_number: int):
        levels_path = str(Path(Path.cwd(), 'Levels'))
        self.level_path = levels_path + f'\\{level_number}.txt'

    def create_map(self, tile_size=16) -> list:
        lvl_map: List[MapObject] = []
        with open(self.level_path, 'r') as lvl_data:
            for i, line in enumerate(lvl_data.readlines()):
                for j, symbol in enumerate(line):
                    pos = (j * tile_size, i * tile_size)
                    rect = Rect(pos, (tile_size, tile_size))
                    if symbol == '\n' or symbol == 'H' or symbol == 'E':
                        continue
                    map_object = self.block_dict[symbol](rect)
                    lvl_map.append(map_object)
        return lvl_map


class EnemyQueueCreator:
    enemies_queue = [18 * [Common]]

    def generate_queue(self, level_num: int):
        return self.enemies_queue[level_num - 1]


class Level:
    def __init__(self, level_number: int):
        self.map = MapCreator(level_number).create_map()