Separate ledger read functions

1 files changed, 22 insertions, 0 deletions

diff --git a/kraken.py b/kraken.py
new file mode 100644
index 0000000..c420d89
--- /dev/null
+++ b/kraken.py
@@ -0,0 +1,22 @@
+import csv
+from decimal import Decimal
+from typing import List
+from ledger_action import LedgerAction
+
+
+def parse_row(row: dict[str,str]) -> LedgerAction:
+    date = row["time"].split(" ")[0]
+    return LedgerAction(
+        type=row["type"],
+        asset=row["asset"],
+        amount=Decimal(row["amount"]),
+        fee=Decimal(row.get("fee", "0")),
+        refid=row.get("refid", ""),
+        date=date,
+    )
+
+
+def read_ledger(csv_path: str) -> List[LedgerAction]:
+    with open(csv_path, "r") as file:
+        reader = csv.DictReader(file)
+        return list(map(parse_row, reader))