Resolução dos exercicios do item 2_4

lucas-jsvd · lucas-jsvd · commit a5cfa047d8d1 · 2020-06-18T15:26:21.000-03:00
diff --git a/Work/pcost.py b/Work/pcost.py
@@ -9,15 +9,16 @@ def portfolio_cost(filename):
     compras_totais = 0.0
     with open(filename, "rt") as arquivo:
         csvfile = csv.reader(arquivo)
-        next(csvfile)
-        for linha in csvfile:
+        headers = next(csvfile)
+        for num_linha, linha in enumerate(csvfile, start=1):
+            record = dict(zip(headers, linha))
             try:
-                quant_acao = int(linha[1])
-                preco_acao = float(linha[2])
+                quant_acao = int(record["shares"])
+                preco_acao = float(record["price"])
+                valor_total_acao = quant_acao * preco_acao
+                compras_totais += valor_total_acao
             except ValueError:
-                print("Existem valores vazios ou passados em um formato incorreto.")
-            valor_total_acao = quant_acao * preco_acao
-            compras_totais += valor_total_acao
+                print(f'Row {num_linha}: Couldn\'t convert: {linha}')
     return compras_totais
 
 
diff --git a/Work/report.py b/Work/report.py
@@ -9,9 +9,9 @@ def read_portfolio(filename):
     portfolio = []
     with open(filename, 'rt') as f:
         rows = csv.reader(f)
-        next(rows)
-        for row in rows:
-            holding = dict(nome=row[0], acoes=int(row[1]), preco=float(row[2]))
+        cabecalho = next(rows)
+        for number_row, row in enumerate(rows, start=1):
+            holding = dict(zip(cabecalho, row))
             portfolio.append(holding)
     return portfolio
 
@@ -34,25 +34,25 @@ def ganhos_perda(portfolio, acoes):
     diferencia_total = 0.0
     print("Nome\tPreco Compra\tQuantidade\tTotal da Compra\t\tPreco de Venda\t\tTotal de Venda\t\tGanho/Perda")
     for acao in portfolio:
-        valor_acao_compra = acao["acoes"] * acao["preco"]
-        valor_acao_atual = acao["acoes"] * acoes[acao["nome"]]
+        valor_acao_compra = acao["shares"] * acao["price"]
+        valor_acao_atual = acao["shares"] * float(acoes[acao["name"]])
         diferencia = valor_acao_atual - valor_acao_compra
         diferencia_total += diferencia
-        print(f'{acao["nome"]:<4s} {acao["preco"]:^16.2f} {acao["acoes"]:^16d} {valor_acao_compra:^18.2f} {acoes[acao["nome"]]:^26.2f} {valor_acao_atual:^21.2f} {diferencia:^21.2f}')
+        print(f'{acao["name"]:<4s} {acao["price"]:^16.2f} {acao["shares"]:^16d} {valor_acao_compra:^18.2f} {acoes[acao["name"]]:^26.2f} {valor_acao_atual:^21.2f} {diferencia:^21.2f}')
     print(f"Total de ganhos/perdas: {diferencia_total:>97.2f}")
 
 
 def make_report(portfolio, acoes):
     lista_acoes = []
     for linha in portfolio:
-        diferencia_preco = acoes[linha["nome"]] - linha["preco"]
-        lista_acoes.append((linha["nome"], linha["acoes"], acoes[linha["nome"]], diferencia_preco))
+        diferencia_preco = acoes[linha["name"]] - float(linha["price"])
+        lista_acoes.append((linha["name"], linha["shares"], acoes[linha["name"]], diferencia_preco))
     return lista_acoes
 
 
-report = make_report(read_portfolio("Work\\Data\\portfolio.csv"), read_prices("Work\\Data\\prices.csv"))
+report = make_report(read_portfolio("Work\\Data\\portfoliodate.csv"), read_prices("Work\\Data\\prices.csv"))
 cabecalho = ("Name", "Shares", "Price", "Change")
 print(f'{cabecalho[0]:>10s} {cabecalho[1]:>10s} {cabecalho[2]:>10s} {cabecalho[3]:>10s}')
 print("---------- " * 4)
 for name, shares, price, change in report:
-    print(f'{name:>10s} {shares:>10d} {price:>10.2f} {change:>10.2f}')
+    print(f'{name:>10s} {int(shares):>10d} {price:>10.2f} {change:>10.2f}')
