add_scheme.py

from functions import *


def save_scheme(flashcard_id: int, filename: str):
    """
    ## `save_scheme` method

    ===============================================

    #### Description

    It saves the scheme.
    """
    exit_message = "EXIT_SUCCESS"

    if exists_filename(filename):
        cursor.execute(
            "UPDATE flashcards SET filename_scheme=? WHERE id=?",
            (filename, flashcard_id),
        )
        con.commit()
    else:
        sg.popup_error("Path Wrong", keep_on_top=True, modal=True)
        exit_message = "PATH_WRONG"

    return exit_message


def update_scheme(flashcard_id: int) -> (str, str):
    layout = [
        [sg.Text("Please, select a file", size=(20, 1))],
        [
            sg.InputText(key="filename_scheme"),
            sg.FileBrowse(file_types=(("Image Portable Network Graphics", "*.png"),)),
        ],
        [sg.Column([[sg.Button("Save", key="save_scheme")]], justification="right")],
    ]

    window = sg.Window(
        layout=layout, title="File selector", keep_on_top=True, modal=True
    )

    filename: str = ""
    exit_message = "EXIT_SUCCESS"

    while True:
        event, values = window.read()

        if event in [sg.WIN_CLOSED, "Exit"]:
            filename = ""
            exit_message = "EXIT_WINDOW"
            break

        if event == "save_scheme":
            if "EXIT_SUCCESS" == save_scheme(
                flashcard_id=flashcard_id, filename=values["filename_scheme"]
            ):
                break

    window.close()

    return filename, exit_message