fga-eps-mds/2022-2-Certifik8

import argparse
import time
from .modules.generator.certificado import Certificados
from .modules.handler.tabela import Tabela
from .path import path_inicial
from .modules.utils import verificar_xlsx
from .modules.selector.file_selection import FileSelection
from .modules.selector.folder_selection import FolderSelection
from .installer.installer import installer


def run():
    """
    Objetivo da classe: controlar o fluxo do programa.

        1. Exibição de menu inicial
        2. Seleção de tabelas .xlsx para gerar certificados
        3. Seleção da pasta destino para salvar os certificados
        4. Chama as funções de geração de certificados
    """

    parser = argparse.ArgumentParser(add_help=False)
    parser.add_argument("-h", "--help", help="Como usar", action="store_true")
    parser.add_argument("--install", help="Instalar dependências", action="store_true")
    args = parser.parse_args()

    if args.help:
        with open(
            file=path_inicial + "/constants/help.txt",
            encoding="utf-8",
        ) as text_help:
            msg_help = text_help.read()
        print(msg_help)
        exit()
    elif args.install:
        installer()
        exit()

    with open(
        file=path_inicial + "/constants/menu.txt",
        encoding="utf-8",
    ) as text:
        menu = text.read()

    print(menu)

    print()

    print("Selecione as tabela para gerar o certificado:")
    paths = FileSelection().run()

    for path in paths:
        print("\t" + path)

    print()

    time.sleep(3)

    print("Selecione a pasta para guardar os certificados:")

    path_destino = FolderSelection().run()
    print("\t" + path_destino)

    print()
    certificados = Certificados()
    tabela = Tabela()
    for path in paths:
        if verificar_xlsx(path):
            if tabela.set_data_frames(path):
                if tabela.verificar_tab_padrao(path_destino):
                    certificados.gerar_certificados(
                        path,
                        tabela.get_data_frame(),
                        tabela.get_data_frame_informacoes(),
                        path_destino,
                    )
        else:
            print(f"{path} - não é .xlsx, certificados não gerados!!!")
        time.sleep(1)
    return True