From 18011741c523fb2ea62133db1978c411ad818285 Mon Sep 17 00:00:00 2001
From: Mark Veidemanis <m@zm.is>
Date: Tue, 22 Feb 2022 19:21:29 +0000
Subject: [PATCH] Make ES optional and properly calculate open trade volume

---
 handler/transactions.py | 51 +++++++++++++++++++++++++++++------------
 1 file changed, 36 insertions(+), 15 deletions(-)

diff --git a/handler/transactions.py b/handler/transactions.py
index 7c693fd..f5baf00 100644
--- a/handler/transactions.py
+++ b/handler/transactions.py
@@ -31,13 +31,14 @@ class Transactions(object):
         Set the logger.
         """
         self.log = Logger("transactions")
-        self.es = Elasticsearch(
-            f"https://{settings.ES.Host}:9200",
-            verify_certs=False,
-            basic_auth=(settings.ES.Username, settings.ES.Pass),
-            # ssl_assert_fingerprint=("6b264fd2fd107d45652d8add1750a8a78f424542e13b056d0548173006260710"),
-            ca_certs="certs/ca.crt",
-        )
+        if settings.ES.Enabled == "1":
+            self.es = Elasticsearch(
+                f"https://{settings.ES.Host}:9200",
+                verify_certs=False,
+                basic_auth=(settings.ES.Username, settings.ES.Pass),
+                # ssl_assert_fingerprint=("6b264fd2fd107d45652d8add1750a8a78f424542e13b056d0548173006260710"),
+                ca_certs="certs/ca.crt",
+            )
 
     def run_checks_in_thread(self):
         """
@@ -55,9 +56,10 @@ class Transactions(object):
         """
         Set up the LoopingCalls to get the balance so we have data in ES.
         """
-        self.lc_es_checks = LoopingCall(self.run_checks_in_thread)
-        delay = int(settings.ES.RefreshSec)
-        self.lc_es_checks.start(delay)
+        if settings.ES.Enabled == "1":
+            self.lc_es_checks = LoopingCall(self.run_checks_in_thread)
+            delay = int(settings.ES.RefreshSec)
+            self.lc_es_checks.start(delay)
 
     # TODO: write tests then refactor, this is terribly complicated!
     def transaction(self, data):
@@ -536,10 +538,11 @@ class Transactions(object):
         return cast
 
     def write_to_es(self, msgtype, cast):
-        cast["type"] = msgtype
-        cast["ts"] = str(datetime.now().isoformat())
-        cast["xtype"] = "tx"
-        self.es.index(index=settings.ES.Index, document=cast)
+        if settings.ES.Enabled == "1":
+            cast["type"] = msgtype
+            cast["ts"] = str(datetime.now().isoformat())
+            cast["xtype"] = "tx"
+            self.es.index(index=settings.ES.Index, document=cast)
 
     def get_remaining(self):
         """
@@ -599,7 +602,25 @@ class Transactions(object):
         rates = self.agora.get_rates_all()
         cumul_usd = 0
         for contact_id, contact in dash.items():
-            amount = contact["data"]["amount"]
+            # We need created at in order to look up the historical prices
+            created_at = contact["data"]["created_at"]
+
+            # Reformat the date how CoinGecko likes
+            date_parsed = datetime.strptime(created_at, "%Y-%m-%dT%H:%M:%S.%fZ")
+            date_formatted = date_parsed.strftime("%d-%m-%Y")
+
+            # Get the historical rates for the right asset, extract the price
+            asset = contact["data"]["advertisement"]["asset"]
+            if asset == "XMR":
+                amount_crypto = contact["data"]["amount_xmr"]
+                history = self.agora.cg.get_coin_history_by_id(id="monero", date=date_formatted)
+                crypto_usd = float(history["market_data"]["current_price"]["usd"])
+            elif asset == "BTC":
+                amount_crypto = contact["data"]["amount_btc"]
+                history = self.agora.cg.get_coin_history_by_id(id="bitcoin", date=date_formatted)
+                crypto_usd = float(history["market_data"]["current_price"]["usd"])
+            # Convert crypto to fiat
+            amount = float(amount_crypto) * crypto_usd
             currency = contact["data"]["currency"]
             if not contact["data"]["is_selling"]:
                 continue