summaryrefslogtreecommitdiff
path: root/bla.py
diff options
context:
space:
mode:
Diffstat (limited to 'bla.py')
-rw-r--r--bla.py16
1 files changed, 9 insertions, 7 deletions
diff --git a/bla.py b/bla.py
index 6c2a4c2..d3ddcdf 100644
--- a/bla.py
+++ b/bla.py
@@ -2,6 +2,7 @@ import csv
from collections import defaultdict
from datetime import datetime
+from trade import Trade
from trade_queue import FIFOQueue
@@ -9,15 +10,16 @@ def generate_report(sale_entries, proceeds, crypto_asset, date_sold):
report = []
sell_date = datetime.strptime(date_sold, "%Y-%m-%d").strftime("%d.%m.%Y")
- for amount, cost, buy_date in sale_entries:
- buy_date_formatted = datetime.strptime(buy_date, "%Y-%m-%d").strftime("%d.%m.%Y")
- holding_period = (datetime.strptime(date_sold, "%Y-%m-%d") - datetime.strptime(buy_date, "%Y-%m-%d")).days
+ trade: Trade
+ for trade in sale_entries:
+ buy_date_formatted = datetime.strptime(trade.date, "%Y-%m-%d").strftime("%d.%m.%Y")
+ holding_period = (datetime.strptime(date_sold, "%Y-%m-%d") - datetime.strptime(trade.date, "%Y-%m-%d")).days
short_or_long = "Short" if holding_period < 365 else "Long"
- cost_basis = cost
- gain_or_loss = proceeds - cost_basis
+ total_cost = trade.total_cost
+ gain_or_loss = proceeds - total_cost
report.append({
- "Amount": f"{amount:.8f}",
+ "Amount": f"{trade.amount:.8f}",
"Currency": crypto_asset,
"Date Sold": sell_date,
"Date Acquired": buy_date_formatted,
@@ -25,7 +27,7 @@ def generate_report(sale_entries, proceeds, crypto_asset, date_sold):
"Buy/Input at": "Kraken",
"Sell/Output at": "Kraken",
"Proceeds": f"{proceeds:.2f}",
- "Cost Basis": f"{cost_basis:.2f}",
+ "Cost Basis": f"{total_cost:.2f}",
"Gain/Loss": f"{gain_or_loss:.2f}",
})