From ff6e5110a439d1466dee9d5907d74b39491c247c Mon Sep 17 00:00:00 2001
From: Jon Michael Aanes <jonjmaa@gmail.com>
Date: Sun, 2 Jun 2024 18:00:55 +0200
Subject: [PATCH] Kraken use fin_depo

---
 personal_data/fetchers/defi_kraken.py | 43 -------------------------
 personal_data/fetchers/defi_kucoin.py | 45 ++++++++++++++++++++-------
 personal_data/main.py                 | 11 ++++++-
 3 files changed, 43 insertions(+), 56 deletions(-)
 delete mode 100644 personal_data/fetchers/defi_kraken.py

diff --git a/personal_data/fetchers/defi_kraken.py b/personal_data/fetchers/defi_kraken.py
deleted file mode 100644
index d3e1f0a..0000000
--- a/personal_data/fetchers/defi_kraken.py
+++ /dev/null
@@ -1,43 +0,0 @@
-import dataclasses
-import datetime
-import logging
-from collections.abc import Iterator, Mapping
-from decimal import Decimal
-
-import krakenex
-from frozendict import frozendict
-
-from personal_data.data import DeduplicateMode, Scraper
-
-from .. import secrets
-
-logger = logging.getLogger(__name__)
-
-# TODO: Move these into secrets!
-client = krakenex.API(
-    secrets.KRAKEN_KEY,
-    secrets.KRAKEN_SECRET,
-)
-
-
-@dataclasses.dataclass(frozen=True)
-class KrakenAccountBalances(Scraper):
-    dataset_name = 'defi_kraken_balance'
-    deduplicate_mode = DeduplicateMode.ONLY_LATEST
-    deduplicate_ignore_columns = ['account.update_time']
-
-    def scrape(self) -> Iterator[Mapping[str, object]]:
-        data_point = {
-            'account.update_time': datetime.datetime.now(tz=datetime.UTC),
-        }
-
-        result = client.query_private('Balance')
-
-        for account, balance_str in result['result'].items():
-            key = f'balance.{account}'
-            balance = Decimal(balance_str)
-            balance = balance if balance != 0 else Decimal(0)
-            data_point[key] = data_point.get(key, Decimal(0)) + balance
-            del key, balance
-
-        yield frozendict(data_point)
diff --git a/personal_data/fetchers/defi_kucoin.py b/personal_data/fetchers/defi_kucoin.py
index 15f7fe1..eb9d5c3 100644
--- a/personal_data/fetchers/defi_kucoin.py
+++ b/personal_data/fetchers/defi_kucoin.py
@@ -3,10 +3,11 @@ import datetime
 import logging
 from collections.abc import Iterator, Mapping
 from decimal import Decimal
+import abc
 
 import kucoin.client
 from frozendict import frozendict
-from fin_depo.defi_kucoin import KucoinDepoFetcher
+import fin_depo
 
 from personal_data.data import DeduplicateMode, Scraper
 
@@ -23,19 +24,14 @@ client = kucoin.client.Client(
 )
 
 @dataclasses.dataclass(frozen=True)
-class KucoinAccountBalances(Scraper):
-    dataset_name = 'defi_kucoin_balance'
-    deduplicate_mode = DeduplicateMode.ONLY_LATEST
-    deduplicate_ignore_columns = ['account.update_time']
+class FinanceDepoScraper(Scraper):
+
+    @abc.abstractmethod
+    def get_depo_fetcher(self):
+        pass
 
     def scrape(self) -> Iterator[Mapping[str, object]]:
-        fetcher = KucoinDepoFetcher(
-            kucoin_key = secrets.KUCOIN_KEY,
-            kucoin_secret = secrets.KUCOIN_SECRET,
-            kucoin_pass = secrets.KUCOIN_PASS,
-        )
-
-        depo = fetcher.get_depo()
+        depo = self.get_depo_fetcher().get_depo()
 
         data_point = {
             'account.update_time': depo.updated_time,
@@ -50,6 +46,31 @@ class KucoinAccountBalances(Scraper):
         yield frozendict(data_point)
 
 
+@dataclasses.dataclass(frozen=True)
+class KrakenAccountBalances(FinanceDepoScraper):
+    dataset_name = 'defi_kraken_balance'
+    deduplicate_mode = DeduplicateMode.ONLY_LATEST
+    deduplicate_ignore_columns = ['account.update_time']
+
+    def get_depo_fetcher(self):
+        return fin_depo.defi_kraken.KrakenDepoFetcher(
+            kraken_key = secrets.KRAKEN_KEY,
+            kraken_secret = secrets.KRAKEN_SECRET,
+        )
+
+@dataclasses.dataclass(frozen=True)
+class KucoinAccountBalances(FinanceDepoScraper):
+    dataset_name = 'defi_kucoin_balance'
+    deduplicate_mode = DeduplicateMode.ONLY_LATEST
+    deduplicate_ignore_columns = ['account.update_time']
+
+    def get_depo_fetcher(self):
+        return fin_depo.defi_kucoin.KucoinDepoFetcher(
+            kucoin_key = secrets.KUCOIN_KEY,
+            kucoin_secret = secrets.KUCOIN_SECRET,
+            kucoin_pass = secrets.KUCOIN_PASS,
+        )
+
 def addresses_to_data_points(addresses: list[dict[str,str]]) -> frozendict:
     data_point = {}
     data_point['account.update_time'] = datetime.datetime.now(tz=datetime.UTC)
diff --git a/personal_data/main.py b/personal_data/main.py
index 7d4733b..566ec4a 100644
--- a/personal_data/main.py
+++ b/personal_data/main.py
@@ -1,5 +1,6 @@
 import csv
 import datetime
+import inspect
 import decimal
 import io
 import logging
@@ -217,7 +218,15 @@ def send_notification(
 
 
 def available_scrapers() -> list[type[personal_data.data.Scraper]]:
-    return personal_data.data.Scraper.__subclasses__()
+    subclasses = []
+    class_queue = [personal_data.data.Scraper]
+    while class_queue:
+        clazz = class_queue.pop()
+        if inspect.isabstract(clazz):
+            class_queue.extend(clazz.__subclasses__())
+        else:
+            subclasses.append(clazz)
+    return subclasses
 
 
 def available_scraper_names() -> list[str]: