main.py

import textwrap
from abc import ABC, abstractclassmethod, abstractproperty
from datetime import datetime
import time


class Cliente:
    def __init__(self, endereco):
        self.endereco = endereco
        self.contas = []
    
    def realizar_transacao(self, conta, transacao):
        transacao.registrar(conta)
    
    def adicionar_conta(self, conta):
        self.contas.append(conta)


class PessoaFisica(Cliente):
    def __init__(self, nome, data_nascimento, cpf, endereco):
        super().__init__(endereco)
        self.nome = nome
        self.data_nascimento = data_nascimento
        self.cpf = cpf


class PessoaJuridica(Cliente):
    def __init__(self, razao_social, cnpj, endereco):
        super().__init__(endereco)
        self.razao_social = razao_social
        self.cnpj = cnpj


class Conta:
    def __init__(self, numero, cliente):
        self._saldo = 0
        self._numero = numero
        self._agencia = "0001"
        self._cliente = cliente
        self._historico = Historico()

    @classmethod
    def nova_conta(cls, cliente, numero):
        return cls(numero, cliente)

    @property
    def saldo(self):
        return self._saldo
    
    @property
    def numero(self):
        return self._numero
    
    @property
    def agencia(self):
        return self._agencia
    
    @property
    def cliente(self):
        return self._cliente
    
    @property
    def historico(self):
        return self._historico
    
    def sacar(self, valor):
        saldo = self.saldo

        if valor > saldo:
            print("Operação negada. Você não tem saldo suficiente em sua conta.")
        
        elif valor > 0:
            self._saldo -= valor
            print("""
                Saque autorizado!

                Aguarde a contagem das notas....

                Operação concluída. Retire seus notas da máquina. 
            """);
            return True
        
        else:
            print("Falha na operação. O valor informado é inválido.")
        
        return False

    def depositar(self, valor):
        if valor > 0:
            self._saldo += valor
            print("""
                Insira as notas no local indicado.

                Aguarde a contagem das notas....

                Operação concluída. Depósito realizado com sucesso. 
            """);
        else:
            print("Valor inválido. O valor mínimo de depósito é de R$1!")
            return False
        
        return True


class ContaCorrente(Conta):
    def __init__(self, numero, cliente, limite=500, limite_saque=3):
        super().__init__(numero, cliente)
        self._limite = limite
        self._limite_saque = limite_saque
        #Numa atualização futura será implementada uma regra para valdiar o tipo de conta para ter o valor limite de saque e o número de saques definidos mais dinamicamente

    def sacar(self, valor):
        numero_saques = len(
            [transacao for transacao in self.historico.transacoes if transacao["tipo"] == "Saque"]
        )

        if valor > self._limite:
            print("Operação negada! Valor do saque excede o limite de saque diário.")
        
        elif numero_saques >= self._limite_saque:
            print("Operação negada! Quantidade de saques diárias atingida.")
        
        else:
            return super().sacar(valor)
    
        return False

    def __str__(self):
        return f"""
            Agência:\t{self.agencia}
            C/C:\t\t{self.numero}
            Titular:\t{self.cliente.nome}
        """


class Historico:
    def __init__(self):
        self._transacoes = []
    
    @property
    def transacoes(self):
        return self._transacoes
    
    def adicionar_transacao(self, transacao):
        self._transacoes.append(
            {
                "tipo": transacao.__class__.__name__,
                "valor": transacao.valor,
                "data": datetime.now().strftime("%d/%m/%Y %H:%M:%S")
            }
        )


class Transacao(ABC):
    @property # type: ignore
    @abstractproperty
    def valor(self):
        pass

    @abstractclassmethod
    def registrar(self, conta):
        pass


class Saque(Transacao):
    def __init__(self, valor):
        self._valor = valor

    @property
    def valor(self):
        return self._valor
    
    def registrar(self, conta):
        sucesso_transacao = conta.sacar(self.valor)

        if sucesso_transacao:
            conta.historico.adicionar_transacao(self)


class Deposito(Transacao):
    def __init__(self, valor):
        self._valor = valor
    
    @property
    def valor(self):
        return self._valor
    
    def registrar(self, conta):
        sucesso_transacao = conta.depositar(self.valor)

        if sucesso_transacao:
            conta.historico.adicionar_transacao(self)


def menu():
    menu = """
    ====================PyBank====================

    ===========Seja Bem-Vindo ao Pybank===========

    Escolha uma das Opções do nosso sistema

    1 - SAQUE
    2 - DEPOSITO
    3 - EXTRATO
    4 - NOVO USUÁRIO
    5 - NOVA CONTA
    6 - LISTAR CONTAS
    0 - SAIR

    Informações Importantes:
        * Você pode realizar até 3 saques/dia
        * Seu limite por saque é de R$500,00

    ==============================================
    """;

    print(menu)

    opcao = int(input("Escolha a operação que deseja realizar: "));

    return opcao


def sacar(clientes):
    print("SAQUE \n");
    cpf = input("Digite seu CPF: ")
    cliente = filtrar_clientes(cpf, clientes)

    if not cliente:
        print("Cliente não encontrado. Vefique o CPF digitado e tente novamente.")
        return
    
    valor = int(input("Digite o valor a ser sacado: "))
    transacao = Saque(valor)

    conta = recurperar_conta_cliente(cliente)
    if not conta:
        return
    
    cliente.realizar_transacao(conta, transacao)


def recurperar_conta_cliente(cliente):
    if not cliente.contas:
        print("Falha na operação! Cliente informado não possui nenhuma conta registrada.")
        return
    # FIXME:
    return cliente.contas[0]


def depositar(clientes):
    print('DEPÓSITO \n');

    cpf = input("Digite seu CPF: ")
    cliente = filtrar_clientes(cpf, clientes)
    
    if not cliente:
        print("Cliente não encontrado. Vefique o CPF digitado e tente novamente.")
        return
    
    valor = int(input("Digite o valor a ser depositado: "))
    transacao = Deposito(valor)

    conta = recurperar_conta_cliente(cliente)
    if not conta:
        return

    cliente.realizar_transacao(conta, transacao)


def extrato(clientes):
    print("EXTRATO \n");
    cpf = input("Digite seu CPF: ")
    cliente = filtrar_clientes(cpf, clientes)
    
    if not cliente:
        print("Cliente não encontrado. Vefique o CPF digitado e tente novamente.")
        return
    
    conta = recurperar_conta_cliente(cliente)
    if not conta:
        return
    
    print("""
        EXTRATO DA CONTA.

        Aguarde a exibição das informações em tela....

        Operação concluída. Confira abaixo o seu extrato. 
    """);
    transacoes = conta.historico.transacoes

    extrato = ""
    if not transacoes:
        extrato = "Não foram realizados movementações nos últimos 30 dias."
    else: 
        for transacao in transacoes:
            extrato += f"\n{transacao['tipo']}: \n\tR$ {transacao['valor']:.2f}"
        
    print(extrato)
    print(f"\nSaldo:\n\t R$ {conta.saldo:.2f}")
    print("="*50)


def cria_usuario(clientes):
    print("Criação de usuário")
    cpf = input("Digite seu CPF (somente números): ")
    cliente = filtrar_clientes(cpf, clientes)

    if cliente:
        print("Já existe um usuário com esse CPF cadastrado em nosso sistema.")
        return

    nome = input("Digite seu nome completo: ").upper()
    data_nascimento = input("Digite sua data de nascimento (dd-mm-aaaa): ")
    endereco = input("Digite seu endereço completo: ")

    cliente = PessoaFisica(nome=nome, data_nascimento=data_nascimento, cpf=cpf, endereco=endereco)

    clientes.append(cliente)

    print("Usuário criado com sucesso! Bem Vindo ao PyBank.")


def filtrar_clientes(cpf, clientes):
    clientes_filtrados = [cliente for cliente in clientes if cliente.cpf == cpf]
    return clientes_filtrados[0] if clientes_filtrados else None


def criar_conta(numero_conta, clientes, contas):
    cpf = input("Digite seu CPF (somente números): ")
    cliente = filtrar_clientes(cpf, clientes)

    if not cliente:
        print("Falha na operação! Cliente não encontrada.")
        return
    
    conta = ContaCorrente.nova_conta(cliente=cliente, numero=numero_conta)
    contas.append(conta)
    cliente.contas.append(conta)

    print("Conta aberta com sucesso!")


def listar_contas(contas):
    for conta in contas:
        print("=" * 100)
        print(textwrap.dedent(str(conta)))


def main():
    clientes = []
    contas = []

    while True:
        opcao = menu();

        match opcao:
            case 1:
                sacar(clientes)                

            case 2:
                depositar(clientes)

            case 3:
                extrato(clientes)

            case 4:
                cria_usuario(clientes)
            
            case 5:
                numero_conta = len(contas) + 1
                criar_conta(numero_conta, clientes, contas)
            
            case 6:
                listar_contas(contas)

            case 0:
                break

            case _:
                print("Opção inválida. \n\n")
                time.sleep(2)

    print("\n\nSistema sendo encerrado. Volte sempre!\n")


main()