main.py

from enum import Enum

import csv_table.csv_table as cvt
import table_renderer.table_renderer as tr
import sys


class UserInput(Enum):
    FIRST_PAGE = 0
    PREVIOUS_PAGE = 1
    NEXT_PAGE = 2
    LAST_PATE = 3
    EXIT = 4
    NOT_PARSED = 5


def render_menue():
    print("F)irst page, P)revious page, N)ext page, L)ast page, E)xit")


def parse_user_input(user_input: str) -> UserInput:
    if user_input in ["F", "f", "First page", "first page"]:
        return UserInput.FIRST_PAGE
    if user_input in ["P", "p", "Previous page", "previous page"]:
        return UserInput.PREVIOUS_PAGE
    if user_input in ["N", "n", "Next page", "next page"]:
        return UserInput.NEXT_PAGE
    if user_input in ["E", "e", "Exit", "exit"]:
        return UserInput.EXIT
    else:
        return UserInput.NOT_PARSED


def render_page_count(table):
    current_page_number = table.get_current_page_index() + 1
    last_page_number = table.get_last_page_index() + 1
    print("Page {current} of {max}".format(current=current_page_number, max=last_page_number))
    pass


def main(path: str, page_length: int):
    table = cvt.load_csv(path)
    table.set_page_length(page_length)
    should_exit = False
    while not should_exit:
        render_table(table)
        render_page_count(table)
        use_input = parse_user_input(
            input("F)irst page, P)revious page, N)ext page, L)ast page, E)xit \n"))

        if use_input == UserInput.FIRST_PAGE:
            table.set_page_to_first()
        if use_input == UserInput.LAST_PATE:
            table.set_page_to_last()
        if use_input == UserInput.NEXT_PAGE:
            table.increment_page()
        if use_input == UserInput.PREVIOUS_PAGE:
            table.decrement_page()

        should_exit = use_input == UserInput.EXIT


def render_table(table):
    output = tr.create_table_markup_str(table.caption, table.get_body())
    print(output)


def read_data(path: str) -> str:
    f = open(path)
    lines = f.readlines()
    return "".join(lines)


def parse_program_input(argv: list[str]) -> (str, int):
    path = argv[0]
    size = int(argv[1]) if (len(argv)) > 1 else 3
    return path, size


if __name__ == '__main__':
    table_path, page_size = parse_program_input(sys.argv[1:])

    main(table_path, page_size)