Ruff format

personal_data/__main__.py

@@ -17,7 +17,9 @@ def parse_arguments():
     )
     parser.add_argument('--cookiejar', action='store_true')
     parser.add_argument('--email', action='store_true', dest='send_email_notification')
-    parser.add_argument('--loud-sound', action='store_true', dest='trigger_loud_and_annoying_sound')
+    parser.add_argument(
+        '--loud-sound', action='store_true', dest='trigger_loud_and_annoying_sound',
+    )
     return parser.parse_args()
 
 

personal_data/fetchers/defi_kucoin.py

@@ -1,10 +1,8 @@
-import abc
 import dataclasses
 import datetime
 import logging
 from collections.abc import Iterator, Mapping
 
-import fin_depo
 import kucoin.client
 from frozendict import frozendict
 
@@ -22,6 +20,7 @@ client = kucoin.client.Client(
     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)

personal_data/fetchers/defi_partisia_blockchain.py

@@ -1,18 +1,10 @@
 import dataclasses
-import datetime
-import email.utils
-import json
 import logging
 from collections.abc import Iterator, Mapping
-from decimal import Decimal
 
-import requests
-from frozendict import frozendict
-
-from personal_data.data import DeduplicateMode, Scraper
 import fin_depo
 
-from .. import secrets
+from personal_data.data import DeduplicateMode, Scraper
 
 logger = logging.getLogger(__name__)
 

personal_data/fetchers/fin_depo.py

@@ -1,11 +1,9 @@
 import abc
 import dataclasses
-import datetime
 import logging
 from collections.abc import Iterator, Mapping
 
 import fin_depo
-import kucoin.client
 from frozendict import frozendict
 
 from personal_data.data import DeduplicateMode, Scraper
@@ -60,6 +58,7 @@ class KucoinAccountBalances(FinanceDepoScraper):
             kucoin_pass=secrets.KUCOIN_PASS,
         )
 
+
 @dataclasses.dataclass(frozen=True)
 class MpcBalance(FinanceDepoScraper):
     dataset_name = 'defi_mpc_balance'
@@ -67,6 +66,5 @@ class MpcBalance(FinanceDepoScraper):
     def get_depo_fetcher(self):
         return fin_depo.defi_partisia_blockchain.PartisiaBlockchainAccountDepoFetcher(
             self.session,
-                blockchain_address = secrets.PBC_ACCOUNT_ADDRESS
+            blockchain_address=secrets.PBC_ACCOUNT_ADDRESS,
         )
-

personal_data/main.py

@@ -28,6 +28,7 @@ except ImportError:
 
 import personal_data.data
 import personal_data.fetchers
+
 from . import notification
 
 CSV_DIALECT = 'one_true_dialect'
@@ -235,7 +236,7 @@ def main(
         logger.warning('No cookiejar is used')
 
     if len(notification_types) == 0:
-        logger.info('Email not enabled: Notifications will not be sent!')
+        logger.info('No notifications enabled: Notifications will not be sent!')
 
     for scraper_cls in available_scrapers():
         session = get_session(cookiejar, with_cfscrape=scraper_cls.requires_cfscrape())
@@ -264,9 +265,8 @@ def main(
         logger.info('Scraper done: %s', scraper.dataset_name)
 
         if status['extended']:
-            notification.send_notifications(session,
+            notification.send_notifications(
-                                            scraper_cls.__name__,
+                session, scraper_cls.__name__, status['dicts'][-1], notification_types,
-                                            status['dicts'][-1],
+            )
-                                            notification_types)
 
         del scraper, session

personal_data/notification.py

@@ -1,31 +1,19 @@
-import csv
+import enum
-import datetime
-import decimal
-import inspect
-import io
 import logging
-from collections.abc import Iterable, Mapping, Sequence
-from decimal import Decimal
 
 import requests
-import requests_cache
 from frozendict import frozendict
 
-import personal_data.data
-import personal_data.fetchers
-
 from . import mailgun
 
-import enum
-
 logger = logging.getLogger(__name__)
 
+
 class NotificationType(enum.Enum):
     EMAIL = 1
     LOUD_SOUND = 2
 
 
-
 def send_email_notification(
     session: requests.Session,
     scraper_name: str,
@@ -44,6 +32,7 @@ def play_loud_and_annoying_sound(
 ) -> None:
     pass
 
+
 NOTIFICATION_TYPE_TO_NOTIFIER = {
     NotificationType.EMAIL: send_email_notification,
     NotificationType.LOUD_SOUND: play_loud_and_annoying_sound,
@@ -57,5 +46,6 @@ def send_notifications(
     notification_types: set[NotificationType],
 ) -> None:
     for notification_type in notification_types:
-        NOTIFICATION_TYPE_TO_NOTIFIER[notification_type](session, scraper_name, latest_dict)
+        NOTIFICATION_TYPE_TO_NOTIFIER[notification_type](
-
+            session, scraper_name, latest_dict,
+        )

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')