From 5afe8a78cc113e7e12fe9730cf23b27fafd46836 Mon Sep 17 00:00:00 2001
From: Adolfo Jimenez <adolfoj@protonmail.com>
Date: Wed, 3 Jan 2024 10:11:48 -0400
Subject: [PATCH] Add utilities for DB data loading

---
 parser.py | 79 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100644 parser.py

diff --git a/parser.py b/parser.py
new file mode 100644
index 0000000..b1cae91
--- /dev/null
+++ b/parser.py
@@ -0,0 +1,79 @@
+import csv
+
+from trades.models import Trade
+
+from assets.models import Currency, CurrencyPair
+
+
+def format_datetime(timestring):
+    """Converts the timestring to ISO format"""
+    date = timestring.split(" ")[0].split(".")
+    time = timestring.split(" ")[1]
+    return f"{date[0]}-{date[1]}-{date[2]} {time}"
+
+
+def load_trades_from_csv(file_path, user):
+    with open(file_path, "r") as f:
+        reader = csv.reader(f)
+        next(reader)  # Skip the header row.
+
+        operations = []
+        for row in reader:
+            try:
+                operation = Trade(
+                    user=user,
+                    ticket=int(row[0]),
+                    type="S" if row[2] == "sell" else "L",
+                    currency_pair=CurrencyPair.objects.get(symbol__iexact=row[4]),
+                    open_datetime=format_datetime(row[1]),
+                    close_datetime=format_datetime(row[8]),
+                    open_price=float(row[5]),
+                    stop_loss=float(row[6]),
+                    take_profit=float(row[7]),
+                    close_price=float(row[9]),
+                    volume=float(row[3]),
+                    pnl=float(row[10]),
+                )
+                operations.append(operation)
+            except:
+                continue
+        Trade.objects.bulk_create(operations)
+
+
+def load_currencies_from_csv(file_path):
+    with open(file_path, "r") as f:
+        reader = csv.reader(f)
+        next(reader)  # Skip the header row.
+
+        currencies = []
+        for row in reader:
+            try:
+                currency = Currency(
+                    symbol=row[0],
+                    name=row[1],
+                    description=row[2],
+                )
+                currencies.append(currency)
+            except:
+                continue
+        Currency.objects.bulk_create(currencies)
+
+
+def load_currency_pairs_from_csv(file_path):
+    with open(file_path, "r") as f:
+        reader = csv.reader(f)
+        next(reader)  # Skip the header row.
+
+        pairs = []
+        for row in reader:
+            try:
+                pair = CurrencyPair(
+                    symbol=row[0],
+                    base_currency=Currency.objects.get(symbol__iexact=row[1]),
+                    quote_currency=Currency.objects.get(symbol__iexact=row[2]),
+                    pip_decimal_position=int(row[3]),
+                )
+                pairs.append(pair)
+            except:
+                continue
+        CurrencyPair.objects.bulk_create(pairs)