Adequanto Python-CRUD para duas tabelas (Temas e Histórias)

python_crud_temas_historias.py

import json

# CRUD in dict table (create, read, update, delete, save, load)
# table => {<id_1>: {<field_1>: <value_1>, <field_2>: <value_2>, ...}, ...}
# fields => {<field_1>: <type_1>, <field_2>: <type_2>, ...}
# field name `id´ is reserved


def crud_create(table, fields, register=None,
                fields_error="Not all fields in register."):
    ''' id is autonum '''
    id_pk = (len(table) and max(table)) + 1
    if not register:
        register = {}
        for field, cast in fields.items():
            while True:
                try:
                    register[field] = cast(input(f"{field.capitalize()}: "))
                    break
                except Exception:
                    print(f"😔 [{cast.__name__}]")
    if not all(f in register for f in fields):
        print(f"***** {fields_error} *****")
        return False
    table[id_pk] = register
    return True


def crud_read(table, search=None, empty="List is empty", just_verify=False):
    ''' search => {field: search} '''
    if search:
        field, value = list(search.items())[0]
        if field == 'id':
            if isinstance(value, str) and value.isdigit():
                value = int(value)
            result_set = ((value, table[value]),) if value in table else None
        else:
            result_set = tuple(filter(
                lambda r: str(value).lower() in str(r[1][field]).lower(),
                table.items()))
    else:
        result_set = table.items()
    if just_verify:
        return bool(result_set)
    if not result_set:
        print(f"{'-':->50}")
        print(f"***** {empty} *****")
        print(f"{'=':=>50}")
        return False
    for id_pk, register in result_set:
        print(f"{'-':->50}")
        print(f"Id: {id_pk}")
        for field, value in register.items():
            print(f"{field.capitalize()}: {value}")
    print(f"{'=':=>50}")
    return True


def crud_update(table, id_pk, register=None, fileds=None,
                not_found="not found.",
                fields_error="Not all fields in register."):
    if id_pk not in table:
        print(f"***** {id_pk} {not_found} *****")
        return False
    if not register:
        register = table[id_pk]
        for field, value in register.items():
            cast = type(value)
            while True:
                try:
                    new = input(f"{field.capitalize()} [{value}]: ")
                    if new:
                        register[field] = cast(new)
                    break
                except Exception:
                    print(f"😔 [{cast.__name__}]")
        return True
    if not all(f in register for f in fields):
        print(f"***** {fields_error} *****")
        return False
    table[id_pk] = register
    return True


def crud_delete(table, id_pk, not_found="not found"):
    if id_pk not in table:
        print(f"***** {id_pk} {not_found} *****")
        return False
    del table[id_pk]
    return True


def crud_save(table, file_name="dict_crud.json"):
    try:
        with open(file_name, 'w') as f:
            json.dump(table, f, indent=4)
    except Exception:
        return False
    return True


def crud_load(table, file_name="dict_crud.json"):
    if True:  # try:
        table.clear()
        with open(file_name, 'r') as f:
            table.update({int(k): v for k, v in json.load(f).items()})
    else:  # except Exception:
        return False
    return True


# Teste

if __name__ == '__main__':

    # Variáveis globais para teste
    tab_temas = {}
    tab_historias = {}
    campos_tema = {'tema': str}
    campos_historia = {'tema_id': int, 'título': str, 'história': str}
    opcoes = ('Menu',
              'Cadastrar Tema', 'Listar Temas', 'Alterar Tema', 'Apagar Tema',
              'Cadastrar História', 'Listar Histórias', 'Alterar História',
              'Apagar História', 'Carregar', 'Salvar', 'Sair')

    # Funções para teste do CRUD com menu

    def listar_temas():
        crud_read(tab_temas, empty="Listagem vazia!")

    def listar_histórias():
        crud_read(tab_temas, empty="Listagem vazia!")
        crud_read(tab_historias, empty="Listagem vazia!")

    def cadastrar_tema():
        crud_create(tab_temas, campos_tema)

    def cadastrar_história():
        listar_temas()
        crud_create(tab_historias, campos_historia)

    def alterar_tema():
        listar_temas()
        try:
            id_pk = int(input("Qual deseja alterar? "))
            if crud_update(tab_temas, id_pk, not_found="não encontrado!"):
                print("****** Alterado com sucesso! *****")
        except Exception:
            print("*** Erro! ***")

    def alterar_história():
        listar_histórias()
        try:
            id_pk = int(input("Qual deseja alterar? "))
            if crud_update(tab_historias, id_pk, not_found="não encontrado!"):
                print("****** Alterado com sucesso! *****")
        except Exception:
            print("*** Erro! ***")

    def apagar_tema():
        listar_temas()
        try:
            id_pk = int(input("Qual deseja deletar? "))
            if crud_delete(tab_temas, id_pk, not_found="não encontrado!"):
                print("****** Excluído com sucesso! *****")
        except Exception:
            print("*** Erro! ***")

    def apagar_história():
        listar_histórias()
        try:
            id_pk = int(input("Qual deseja deletar? "))
            if crud_delete(tab_historias, id_pk, not_found="não encontrado!"):
                print("****** Excluído com sucesso! *****")
        except Exception:
            print("*** Erro! ***")

    def carregar():
        crud_load(tab_temas, 'tab_temas.json')
        crud_load(tab_historias, 'tab_historias.json')
        listar_histórias()

    def salvar():
        if crud_save(tab_temas, 'tab_temas.json'):
            print("***** Temas salvos com sucesso! *****")
        else:
            print("***** Erro ao salvar Temas! *****")
        if crud_save(tab_historias, 'tab_historias.json'):
            print("***** Histórias salvas com sucesso! *****")
        else:
            print("***** Erro ao salvar Histórias! *****")

    def main():
        menu()
        while True:
            opcao = menu(False, True)
            if opcao == len(opcoes) - 1:
                break
            globals()[opcoes[opcao].lower().replace(' ', '_')]()

    def menu(mostar=True, perguntar=False):
        largura = max(map(len, opcoes))
        if mostar:
            print(f"|{'-':->{largura + 4}s}|")
            print('\n'.join(f"|{n:2}. {opcao:{largura}s}|"
                            for n, opcao in enumerate(opcoes)))
        while perguntar:
            print(f"|{'-':->{largura + 4}s}|")
            try:
                opcao = int(input("| Opção: "))
                assert 0 <= opcao < len(opcoes)
                return opcao
            except Exception:
                print(f"|{' Inválida':{largura + 4}s}|")
                menu()

    main()