uaveiro-leci/1ano/1semestre/fp/avaliacao/apx2/shop.py

# Pode correr o programa sem argumentos:
#   python3 shop
# ou passando outros ficheiros de produtos como argumento:
#   python3 shop produtos1.txt ...

def loadDataBase(fname, produtos):
    """Lê dados do ficheiro fname e atualiza/acrescenta a informação num
    dicionário de produtos com o formato {código: (nome, secção, preço, iva), ...}.
    """
    with open(fname, 'r') as f:                               # Abre o ficheiro em modo de leitura
        productsFileContent = f.read()                        # Cria uma string com o conteudo do ficheiro

    for product in productsFileContent.split('\n')[1:]:       # Divide a string 'productsFileContent' numa lista com cada produto
        productComponents = product.split(';')                # Divide as componentes do produto (código, nome, secção, preço, iva)

        if len(productComponents) != 5:
            continue

        produtos.update({productComponents[0]: (              # Atualiza a entrada de uma determinada chave (código)
            productComponents[1],                             # Nome
            productComponents[2],                             # Secção
            float(productComponents[3]),                      # Preço
            float(productComponents[4].strip('%')) / 100      # IVA
        )})


def registaCompra(produtos):
    """Lê códigos de produtos (ou códigos e quantidades),
    mostra nome, quantidade e preço final de cada um,
    e devolve dicionário com {codigo: quantidade, ...}
    """
    compra = {"totals": [0, 0, 0]}     # Inicia o dicionário da compra com os totais da mesma: [total bruto, total iva, total liquido]
    userInput = input('Code? ')
    while userInput != "":
        try:
            code, amount = userInput.split(' ')              # Divide o input do utilizador no código e na quantia
        except ValueError:
            code, amount = userInput, 1                      # No caso de não ser introduzida quantia, então ela fica 1

        # Caso a segunda parcela da entrada não seja um número, é pedido ao utilizador para introduzir o código de novo
        try:
            amount = int(amount)
        except ValueError:
            userInput = input('Code? ')
            continue

        if code in produtos:
            if code not in compra:                                        # Se o produto ainda não estiver na lista é adicionado à mesma
                compra[code] = 0

            compra[code] += amount                                        # Adiciona ao dicionário da compra a quantidade comprada do produto
            noIvaPrice = produtos[code][2] * amount                       # Obtém o preço (sem iva) do determinado produto
            compra["totals"][0] += noIvaPrice                             # Adiciona o preço sem iva ao total bruto
            compra["totals"][1] += noIvaPrice * produtos[code][3]         # Adiciona o valor do iva ao total iva
            print(f"{produtos[code][0]}  {amount}  {noIvaPrice * (1+produtos[code][3]):.2f}")

        userInput = input('Code? ')

    compra["totals"][2] += compra["totals"][0] + compra["totals"][1]      # Calcula o total liquido da compra
    return compra    # Devolve a lista


def fatura(produtos, compra):
    """Imprime a fatura de uma dada compra."""

    # Obtém a lista de secções presentes na compra (por ordem alfabética)
    sections = sorted(list({section for section in [product[1] for code, product in produtos.items() if code in compra]}))

    # Itera as secções para apresentar os produtos ordenados por secção
    for section in sections:
        print(section)     # Mostra a secção

        # Obtém a lista de codigos usados na compra presentes na secção atual
        sectionProductsCodes = sorted([code for code in produtos if produtos[code][1] == section and code in compra])

        # Itera os códigos para apresentar as informações de cada produto
        for code in sectionProductsCodes:
            print(f"{compra[code]:>4} {produtos[code][0]:<31}({int(produtos[code][3]*100):>2}%){round(compra[code]*produtos[code][2]*(1+produtos[code][3]), 2):>11}")

    # Apresenta os totais da compra
    print(f"""{'Total Bruto:':>41}{round(compra["totals"][0], 2):>11}
{'Total IVA:':>41}{round(compra["totals"][1], 2):>11}
{'Total Liquido:':>41}{round(compra["totals"][2], 2):>11}""")


def main(args):
    # produtos guarda a informação da base de dados numa forma conveniente.
    produtos = {'p1': ('Ketchup.', 'Mercearia Salgado', 1.59, 0.23)}
    # Carregar base de dados principal
    loadDataBase("produtos.txt", produtos)
    # Juntar bases de dados opcionais (Não altere)
    for arg in args:
        loadDataBase(arg, produtos)
    
    # Use este código para mostrar o menu e ler a opção repetidamente
    MENU = "(C)ompra (F)atura (S)air ? "
    compras = {}
    repetir = True
    while repetir:
        # Utilizador introduz a opção com uma letra minúscula ou maiúscula
        op = input(MENU).upper()
        # Processar opção
        if op == "C":
            # Esta opção regista os produtos de uma compra
            compras[len(compras)+1] = registaCompra(produtos)

        elif op == "F":
            # Esta opção apresenta a fatura de uma compra
            try:
                numCompra = int(input("Numero compra? "))
                fatura(produtos, compras[numCompra])
            except:
                continue

        elif op == "S":
            repetir = False

    print("Obrigado!")


# Não altere este código / Do not change this code
import sys
if __name__ == "__main__":
    main(sys.argv[1:])