add get_symbol_candles into real time evaluator

Drakkar-Software · Apr 7, 2020 · 5b3bc8e · 5b3bc8e
1 parent 6dfc148
commit 5b3bc8e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 2 deletions.
diff --git a/octobot_evaluators/constants.py b/octobot_evaluators/constants.py
@@ -32,8 +32,8 @@
 
 EVALUATOR_CLASS_TYPE_MRO_INDEX = -4
 
-MATRIX_CHANNEL = "Matrix"
-MATRIX_CHANNELS = "MatrixChannels"
+MATRIX_CHANNEL: str = "Matrix"
+MATRIX_CHANNELS: str = "MatrixChannels"
 
 evaluator_class_str_to_matrix_type_dict = {
     "TAEvaluator": EvaluatorMatrixTypes.TA,

diff --git a/octobot_evaluators/evaluator/realtime_evaluator.py b/octobot_evaluators/evaluator/realtime_evaluator.py
@@ -27,3 +27,12 @@ def __init__(self):
     def load_config(self):
         self.set_default_config()
         self.specific_config = {**self.specific_config, **get_tentacle_config(self.__class__)}
+
+    def get_symbol_candles(self, exchange_name: str, exchange_id: str, symbol: str, time_frame):
+        try:
+            from octobot_trading.api.symbol_data import get_symbol_candles_manager
+            return get_symbol_candles_manager(self.get_exchange_symbol_data(exchange_name, exchange_id, symbol),
+                                              time_frame)
+        except ImportError:
+            self.logger.error(f"Can't get candles manager: requires OctoBot-Trading package installed")
+
diff --git a/tests/channels/test_matrix.py b/tests/channels/test_matrix.py
@@ -18,6 +18,7 @@
 
 from octobot_evaluators.api.evaluators import create_matrix
 from octobot_evaluators.api.initialization import create_matrix_channels
+from octobot_evaluators.constants import MATRIX_CHANNEL
 from octobot_evaluators.data.matrix import get_tentacle_path
 from octobot_evaluators.matrices.matrices import Matrices