diff --git a/personal_data/fetchers/defi_kucoin.py b/personal_data/fetchers/defi_kucoin.py index 240fa5e..5bcfc84 100644 --- a/personal_data/fetchers/defi_kucoin.py +++ b/personal_data/fetchers/defi_kucoin.py @@ -12,13 +12,15 @@ from .. import secrets logger = logging.getLogger(__name__) - -# TODO: Move these into secrets! -client = kucoin.client.Client( - secrets.KUCOIN_KEY, - secrets.KUCOIN_SECRET, - secrets.KUCOIN_PASS, -) +def get_client(): + assert secrets.KUCOIN_KEY, 'Missing secret: KUCOIN_KEY' + assert secrets.KUCOIN_SECRET, 'Missing secret: KUCOIN_SECRET' + assert secrets.KUCOIN_PASS, 'Missing secret: KUCOIN_PASS' + return kucoin.client.Client( + secrets.KUCOIN_KEY, + secrets.KUCOIN_SECRET, + secrets.KUCOIN_PASS, + ) def addresses_to_data_points(addresses: list[dict[str, str]]) -> frozendict: @@ -40,5 +42,5 @@ class KucoinDepositAddresses(Scraper): deduplicate_ignore_columns = ['account.update_time'] def scrape(self) -> Iterator[Mapping[str, object]]: - addresses = client.get_deposit_address('MPC') + addresses = get_client().get_deposit_address('MPC') yield addresses_to_data_points(addresses)