secrets and api keys for all

r0fls · Nov 11, 2024 · 48e1b69 · 48e1b69
1 parent 2005ab8
commit 48e1b69
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/brokers/base_broker.py b/brokers/base_broker.py
@@ -20,6 +20,9 @@ def __init__(
             broker_name,
             engine,
             prevent_day_trading=False):
+        # TODO: remove api_key and secret_key from base broker
+        self.api_key = api_key
+        self.secret_key = secret_key
         self.broker_name = broker_name.lower()
         self.db_manager = DBManager(engine)
         # Use AsyncSession

diff --git a/brokers/kraken_broker.py b/brokers/kraken_broker.py
@@ -11,6 +11,8 @@
 class KrakenBroker(BaseBroker):
     def __init__(self, api_key, secret_key, engine, base_url="https://api.kraken.com", base_currency="ZUSD", **kwargs):
         super().__init__(api_key, secret_key, 'Kraken', engine=engine, **kwargs)
+        self.api_key = api_key
+        self.secret_key = secret_key
         self.base_currency = base_currency
         self.secret_key = secret_key
         self.base_url = base_url