Put states in dedicated folder

Author: sadays

investing_bot_framework/core/states/templates/__init__.py

@@ -6,7 +6,7 @@
 from investing_bot_framework.core.events import Observer
 from investing_bot_framework.core.context.bot_context import BotContext
 from investing_bot_framework.core.exceptions import OperationalException
-from investing_bot_framework.core.context.states import BotState
+from investing_bot_framework.core.states import BotState
 from investing_bot_framework.core.executors import ExecutionScheduler
 from investing_bot_framework.core.workers import Worker
 from investing_bot_framework.core.data_providers import DataProvider
@@ -78,7 +78,7 @@ class DataProvidingState(BotState, Observer):
 
     registered_data_providers: List = None
 
-    from investing_bot_framework.core.context.states.strategy_state import StrategyState
+    from investing_bot_framework.core.states.templates.strategy_state import StrategyState
     transition_state_class = StrategyState
 
     data_provider_scheduler: DataProviderScheduler = None
@@ -88,6 +88,9 @@ def __init__(self, context: BotContext) -> None:
         self._updated = False
         self.data_provider_executor = None
 
+        if self.registered_data_providers is None or len(self.registered_data_providers) < 1:
+            raise OperationalException("Data providing state has not any data providers configured")
+
     def _schedule_data_providers(self) -> List[DataProvider]:
 
         if not DataProvidingState.data_provider_scheduler:
@@ -122,9 +125,6 @@ def _start_data_providers(self, data_providers: List[DataProvider]) -> None:
 
     def run(self) -> None:
 
-        if self.registered_data_providers is None:
-            raise OperationalException("Data providing state has not any data providers configured")
-
         # Schedule the data_providers providers
         planned_data_providers = self._schedule_data_providers()
 

…e/context/states/data_providing_state.py …states/templates/data_providing_state.pyinvesting_bot_framework/core/context/states/data_providing_state.py renamed to investing_bot_framework/core/states/templates/data_providing_state.py investing_bot_framework/core/context/states/data_providing_state.py renamed to investing_bot_framework/core/states/templates/data_providing_state.py

@@ -1,15 +1,19 @@
+import logging
+
 from investing_bot_framework.core.exceptions import ImproperlyConfigured
-from investing_bot_framework.core.context.states import BotState
+from investing_bot_framework.core.states import BotState
+
+logger = logging.getLogger(__name__)
 
 
 class SetupState(BotState):
 
-    from investing_bot_framework.core.context.states.data_providing_state import DataProvidingState
+    from investing_bot_framework.core.states.templates.data_providing_state import DataProvidingState
     transition_state_class = DataProvidingState
 
     def __init__(self, context):
         super(SetupState, self).__init__(context)
-        
+
     def run(self) -> None:
         """
         Running the setup state.
@@ -27,10 +31,4 @@ def run(self) -> None:
                 "manage.py file"
             )
 
-    def stop(self) -> None:
-        # Stopping all services
-        pass
 
-    def reconfigure(self) -> None:
-        # Clean up and reconfigure all the services
-        pass

…ework/core/context/states/setup_state.py …ork/core/states/templates/setup_state.pyinvesting_bot_framework/core/context/states/setup_state.py renamed to investing_bot_framework/core/states/templates/setup_state.py investing_bot_framework/core/context/states/setup_state.py renamed to investing_bot_framework/core/states/templates/setup_state.py

@@ -2,7 +2,7 @@
 import time
 from typing import List
 
-from investing_bot_framework.core.context.states import BotState
+from investing_bot_framework.core.states import BotState
 from investing_bot_framework.core.executors import ExecutionScheduler
 from investing_bot_framework.core.exceptions import OperationalException
 from investing_bot_framework.core.workers import Worker
@@ -66,9 +66,6 @@ def configured(self) -> bool:
 
 class StrategyState(BotState, Observer):
 
-    from investing_bot_framework.core.context.states.data_providing_state import DataProvidingState
-    transition_state_class = DataProvidingState
-
     registered_strategies: List = None
     strategy_scheduler: StrategyScheduler = None
 
@@ -126,11 +123,16 @@ def run(self) -> None:
 
         # Collect all strategies from the strategies providers
         for strategies in self.strategy_executor.registered_strategies:
-            logger.info("Data provider: {} finished running".format(strategies.get_id()))
+            logger.info("Strategy: {} finished running".format(strategies.get_id()))
 
     def update(self, observable, **kwargs) -> None:
         self._updated = True
 
     @staticmethod
     def register_strategies(strategies: List) -> None:
         StrategyState.registered_strategies = strategies
+
+    def get_transition_state_class(self):
+        from investing_bot_framework.core.states.templates.ordering_state import OrderingState
+        return OrderingState
+