add alert_callback to strategy

msaltnet · Apr 28, 2024 · 5263cdb · 5263cdb
1 parent 3cd4328
commit 5263cdb
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 6 deletions.
diff --git a/smtm/strategy/strategy.py b/smtm/strategy/strategy.py
@@ -12,14 +12,15 @@ class Strategy(metaclass=ABCMeta):
 
     @abstractmethod
     def initialize(
-        self, budget, min_price=100, add_spot_callback=None, add_line_callback=None
+        self, budget, min_price=100, add_spot_callback=None, add_line_callback=None, alert_callback=None
     ):
         """예산을 설정하고 초기화한다
 
         budget: 예산
         min_price: 최소 거래 금액, 거래소의 최소 거래 금액
         add_spot_callback(date_time, value): 그래프에 그려질 spot을 추가하는 콜백 함수
         add_line_callback(date_time, value): 그래프에 그려질 line을 추가하는 콜백 함수
+        alert_callback(msg): 알림을 전달하는 콜백 함수 e.g. Operator나 Controller에 전달
         """
 
     @abstractmethod

diff --git a/smtm/strategy/strategy_bnh.py b/smtm/strategy/strategy_bnh.py
@@ -194,7 +194,7 @@ def get_request(self):
             return None
 
     def initialize(
-        self, budget, min_price=5000, add_spot_callback=None, add_line_callback=None
+        self, budget, min_price=5000, add_spot_callback=None, add_line_callback=None, alert_callback=None
     ):
         """예산과 최소 거래 가능 금액을 설정한다"""
         if self.is_intialized:

diff --git a/smtm/strategy/strategy_rsi.py b/smtm/strategy/strategy_rsi.py
@@ -39,7 +39,7 @@ def __init__(self):
         self.position = None
 
     def initialize(
-        self, budget, min_price=100, add_spot_callback=None, add_line_callback=None
+        self, budget, min_price=100, add_spot_callback=None, add_line_callback=None, alert_callback=None
     ):
         """예산을 설정하고 초기화한다
 

diff --git a/smtm/strategy/strategy_sma_0.py b/smtm/strategy/strategy_sma_0.py
@@ -358,7 +358,7 @@ def __create_sell(self):
         }
 
     def initialize(
-        self, budget, min_price=5000, add_spot_callback=None, add_line_callback=None
+        self, budget, min_price=5000, add_spot_callback=None, add_line_callback=None, alert_callback=None
     ):
         """
         예산과 최소 거래 가능 금액을 설정한다

diff --git a/smtm/strategy/strategy_sma_dual_ml.py b/smtm/strategy/strategy_sma_dual_ml.py
@@ -87,7 +87,7 @@ def __init__(self):
         self.last_upper = None
 
     def initialize(
-        self, budget, min_price=5000, add_spot_callback=None, add_line_callback=None
+        self, budget, min_price=5000, add_spot_callback=None, add_line_callback=None, alert_callback=None
     ):
         """
         예산과 최소 거래 가능 금액을 설정한다

diff --git a/smtm/strategy/strategy_sma_ml.py b/smtm/strategy/strategy_sma_ml.py
@@ -80,7 +80,7 @@ def __init__(self):
         self.last_upper = None
 
     def initialize(
-        self, budget, min_price=5000, add_spot_callback=None, add_line_callback=None
+        self, budget, min_price=5000, add_spot_callback=None, add_line_callback=None, alert_callback=None
     ):
         """
         예산과 최소 거래 가능 금액을 설정한다