Ruff

personal_data/fetchers/defi_kucoin.py

@@ -1,13 +1,12 @@
+import abc
 import dataclasses
 import datetime
 import logging
 from collections.abc import Iterator, Mapping
-from decimal import Decimal
-import abc
 
+import fin_depo
 import kucoin.client
 from frozendict import frozendict
-import fin_depo
 
 from personal_data.data import DeduplicateMode, Scraper
 
@@ -23,9 +22,9 @@ client = kucoin.client.Client(
     secrets.KUCOIN_PASS,
 )
 
+
 @dataclasses.dataclass(frozen=True)
 class FinanceDepoScraper(Scraper):
-
     @abc.abstractmethod
     def get_depo_fetcher(self):
         pass
@@ -38,7 +37,7 @@ class FinanceDepoScraper(Scraper):
         }
 
         for asset in depo.assets():
-            key = f"balance.{asset}"
+            key = f'balance.{asset}'
             data_point[key] = depo.get_amount_of_asset(asset)
 
         print(data_point)
@@ -54,10 +53,11 @@ class KrakenAccountBalances(FinanceDepoScraper):
 
     def get_depo_fetcher(self):
         return fin_depo.defi_kraken.KrakenDepoFetcher(
-            kraken_key = secrets.KRAKEN_KEY,
-            kraken_secret = secrets.KRAKEN_SECRET,
+            kraken_key=secrets.KRAKEN_KEY,
+            kraken_secret=secrets.KRAKEN_SECRET,
         )
 
+
 @dataclasses.dataclass(frozen=True)
 class KucoinAccountBalances(FinanceDepoScraper):
     dataset_name = 'defi_kucoin_balance'
@@ -66,12 +66,13 @@ class KucoinAccountBalances(FinanceDepoScraper):
 
     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,
+            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:
+
+def addresses_to_data_points(addresses: list[dict[str, str]]) -> frozendict:
     data_point = {}
     data_point['account.update_time'] = datetime.datetime.now(tz=datetime.UTC)
     data_point['account.num_deposit_addresses'] = len(addresses)

personal_data/main.py

@@ -1,7 +1,7 @@
 import csv
 import datetime
-import inspect
 import decimal
+import inspect
 import io
 import logging
 from collections.abc import Iterable, Mapping, Sequence

setup.py

@@ -15,6 +15,7 @@ PACKAGE_NAME = 'personal_data'
 with open('README.md') as f:
     readme = f.read()
 
+
 def parse_version_file(text: str) -> str:
     match = re.match(r'^__version__\s*=\s*(["\'])([\d\.]+)\1$', text)
     if match is None:
@@ -22,9 +23,11 @@ def parse_version_file(text: str) -> str:
         raise Exception(msg)
     return match.group(2)
 
+
 with open(PACKAGE_NAME + '/_version.py') as f:
     version = parse_version_file(f.read())
 
+
 def parse_requirements(text: str) -> list[str]:
     return text.strip().split('\n')