diff options
| -rw-r--r-- | bla.py | 13 |
1 files changed, 10 insertions, 3 deletions
@@ -35,17 +35,24 @@ def process_ledger(file_path): reader = csv.DictReader(file) for row in reader: currency = row["asset"] - # Use setdefault() for streamlined dictionary handling fifo_queues.setdefault(currency, FIFOQueue()) - if row["type"] == "trade" and row["amount"].startswith('-'): # Sale + # Handle deposits + if row["type"] == "deposit": + amount = float(row["amount"]) + price = 0 # Deposits typically have no associated cost basis + fifo_queues[currency].add(amount, price) + + # Handle trades - sales + elif row["type"] == "trade" and row["amount"].startswith('-'): amount = -float(row["amount"]) cost_basis = fifo_queues[currency].remove(amount) sale_proceeds = amount * float(row["fee"]) # Adjust fee as needed profit_or_loss = sale_proceeds - cost_basis report.append((row["time"], currency, profit_or_loss)) - elif row["type"] == "trade" and not row["amount"].startswith('-'): # Purchase + # Handle trades - purchases + elif row["type"] == "trade" and not row["amount"].startswith('-'): amount = float(row["amount"]) price = float(row["fee"]) # Adjust fee as needed fifo_queues[currency].add(amount, price) |