#! /usr/bin/env python # pyright: basic import json import os import pandas as pd import parser import crud HEADER = """=== Terramotos ===""" MENU ="""[1] Criar a base de dados [2] Atualizar uma entrada [3] Apagar uma entrada [4] Visualizar uma entrada [Q] Sair """ def guardar_df(df: pd.DataFrame, fname: str) -> bool: with open(fname, "w") as fp: fname = f"{fname}.txt" try: fp.write(df.to_string()) except ValueError: return False return True def guardar_json(df: pd.DataFrame, fname: str) -> bool: with open(fname , "w") as fp: try: json.dump(df.to_json(), fp) except: return False return True def guardar_csv(df: pd.DataFrame, fname: str): with open(fname, "w") as fp: try: df.to_csv(fp, index=False) except ValueError: return False return True def main(): isRunning = True db = parser.parse("dados.txt") retInfo = None while isRunning: os.system("cls") print(HEADER + "\n" + MENU) usrIn = input("Opção: ").lower() match usrIn: case "1": os.system("cls") print(HEADER + "\nCRIAR") fname = input("Nome do ficheiro com os dados. (Branco para dados.txt)") if fname == "": fname = "dados.txt" if _file_exists(fname): db = parser.parse(fname) else: retInfo = "Nenhum ficheiro encontrado!" pass case "2": if db is None: continue pass case "3": if db is None: continue pass case "4": if db is not None: ids = crud.read_ids(db) print(ids) input() else: retInfo = "Base de dados não encontrada!" case "q": isRunning = False continue case _: pass if retInfo: print(retInfo) retInfo = None input("Clique Enter para continuar") def _file_exists(name: str) -> bool: currFiles = os.listdir(os.getcwd()) if name in currFiles: return True return False if __name__ == '__main__': main()