can use 3min, 5min, 10min candle info for UpbitDataProvider

msaltnet · Aug 6, 2023 · 40657f8 · 40657f8
1 parent c3728b4
commit 40657f8
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/smtm/controller.py b/smtm/controller.py
@@ -79,7 +79,7 @@ def main(self):
             data_provider = BithumbDataProvider(currency=self.currency)
             trader = BithumbTrader(currency=self.currency, budget=self.budget)
         else:
-            data_provider = UpbitDataProvider(currency=self.currency)
+            data_provider = UpbitDataProvider(currency=self.currency, interval=Config.candle_interval)
             trader = UpbitTrader(currency=self.currency, budget=self.budget)
 
         self.operator.initialize(

diff --git a/smtm/jpt_controller.py b/smtm/jpt_controller.py
@@ -3,6 +3,7 @@
 Jupyter notebook에서 사용하기 좋게 만든 자동 거래 시스템 컨트롤 모듈
 """
 from IPython.display import Image, display
+from .config import Config
 from .log_manager import LogManager
 from .analyzer import Analyzer
 from .upbit_trader import UpbitTrader
@@ -51,7 +52,7 @@ def initialize(self, interval=10, strategy=0, budget=50000, is_bithumb=False):
                 currency=self.market, commission_ratio=0.0005, budget=self.budget
             )
         else:
-            data_provider = UpbitDataProvider(currency=self.market)
+            data_provider = UpbitDataProvider(currency=self.market, interval=Config.candle_interval)
             trader = UpbitTrader(currency=self.market, commission_ratio=0.0005, budget=self.budget)
 
         self.operator.initialize(

diff --git a/smtm/telegram_controller.py b/smtm/telegram_controller.py
@@ -275,7 +275,9 @@ def _on_start_step3(self, command):
         not_ok = True
         if command.upper() in ["1. UPBIT", "1", "UPBIT"]:
             if self.currency in self.UPBIT_CURRENCY:
-                self.data_provider = UpbitDataProvider(currency=self.currency)
+                self.data_provider = UpbitDataProvider(
+                    currency=self.currency, interval=Config.candle_interval
+                )
                 if self.is_demo:
                     self.trader = DemoTrader(budget=self.budget, currency=self.currency)
                 else: