curses_tools.py

from pathlib import Path, PurePath

from game_constants import KeyCode


def read_controls(canvas):
    """Read keys pressed and returns tuple witl controls state."""
    canvas.nodelay(True)
    rows_direction = columns_direction = 0
    space_pressed = False

    while True:
        pressed_key_code = canvas.getch()

        if pressed_key_code == -1:
            # https://docs.python.org/3/library/curses.html#curses.window.getch
            break

        if pressed_key_code == KeyCode.UP.value:
            rows_direction = -1

        if pressed_key_code == KeyCode.DOWN.value:
            rows_direction = 1

        if pressed_key_code == KeyCode.RIGHT.value:
            columns_direction = 1

        if pressed_key_code == KeyCode.LEFT.value:
            columns_direction = -1

        if pressed_key_code == KeyCode.SPACE.value:
            space_pressed = True

    return rows_direction, columns_direction, space_pressed


def draw_frame(canvas, start_row, start_column, text, negative=False):
    """Draw multiline text fragment on canvas, erase text instead of drawing if negative=True is specified."""

    rows_number, columns_number = canvas.getmaxyx()

    for row, line in enumerate(text.splitlines(), round(start_row)):
        if row < 0:
            continue

        if row >= rows_number:
            break

        for column, symbol in enumerate(line, round(start_column)):
            if column < 0:
                continue

            if column >= columns_number:
                break

            if symbol == ' ':
                continue

            # Check that current position it is not in a lower right corner of the window
            # Curses will raise exception in that case. Don`t ask why…
            # https://docs.python.org/3/library/curses.html#curses.window.addch
            if row == rows_number - 1 and column == columns_number - 1:
                continue

            symbol = symbol if not negative else ' '
            canvas.addch(row, column, symbol)


def get_frame_size(text):
    """Calculate size of multiline text fragment, return pair — number of rows and colums."""

    lines = text.splitlines()
    rows = len(lines)
    columns = max([len(line) for line in lines])
    return rows, columns


def get_frames(path: Path) -> list:
    """
    Read content from all files in directory.
    :return: frames_content: content of frames, that reads from txt file.
    """
    abs_dir_path = Path(path).absolute()
    frames = Path.iterdir(abs_dir_path)
    frames_content = []

    for frame in frames:
        with open(PurePath.joinpath(abs_dir_path, frame), 'r') as spaceship:
            frames_content.append(spaceship.read())

    return frames_content