Update templates

Author: MDUYN

investing_algorithm_framework/core/configuration/setup/__init__.py

@@ -1 +1 @@
-from investing_algorithm_framework.core.configuration.setup.default_template_creators import DefaultBotProjectCreator
+from investing_algorithm_framework.core.configuration.setup.default_template_creators import DefaultProjectCreator

investing_algorithm_framework/core/configuration/setup/default_template_creators.py

@@ -6,16 +6,16 @@
 from investing_algorithm_framework.core.configuration.setup.template_creator import TemplateCreator
 
 
-class DefaultBotProjectCreator(TemplateCreator):
-    TEMPLATE_ROOT_DIR = 'templates/bot_project_directory'
-    BOT_PROJECT_NAME_PLACEHOLDER = '{{ bot_project_name }}'
-    BOT_PROJECT_TEMPLATE_DIR_NAME = 'bot_project_template'
+class DefaultProjectCreator(TemplateCreator):
+    TEMPLATE_ROOT_DIR = 'templates/algorithm_project_directory'
+    PROJECT_NAME_PLACEHOLDER = '{{ project_name }}'
+    PROJECT_TEMPLATE_DIR_NAME = 'algorithm_project_template'
 
     def configure(self) -> None:
         bot_dir = os.path.join(self._bot_project_directory, self._bot_name)
 
         if os.path.exists(bot_dir):
-            raise ImproperlyConfigured("Bot destination directory {} already exists".format(self._bot_name))
+            raise ImproperlyConfigured("Project destination directory {} already exists".format(self._bot_name))
 
     def create(self) -> None:
 
@@ -29,7 +29,7 @@ def create(self) -> None:
             path_rest = root[len(template_dir) + 1:]
 
             # Replace template investing_algorithm_framework directory with given investing_algorithm_framework name
-            path_rest = path_rest.replace(self.BOT_PROJECT_TEMPLATE_DIR_NAME, self._bot_name)
+            path_rest = path_rest.replace(self.PROJECT_TEMPLATE_DIR_NAME, self._bot_name)
 
             # Create the directories if they don't exist
             destination_dir = os.path.join(self._bot_project_directory, path_rest)
@@ -76,15 +76,15 @@ def create(self) -> None:
                     )
 
                 # Format placeholders in file if needed
-                if filename in ['manage.py-template', 'settings.py-template']:
+                if filename in ['manage.py-template', 'settings.py-template', 'context.py-template']:
 
                     # Read the file
                     with open(destination_path, 'r') as file:
 
                         file_data = file.read()
 
                     # Replace the placeholder with the investing_algorithm_framework name
-                    file_data = file_data.replace(self.BOT_PROJECT_NAME_PLACEHOLDER, self._bot_name)
+                    file_data = file_data.replace(self.PROJECT_NAME_PLACEHOLDER, self._bot_name)
 
                     # Write the file out again
                     with open(destination_path, 'w') as file:

investing_algorithm_framework/templates/bot_project_directory/bot_project_template/configuration/__init__.py-template renamed to investing_algorithm_framework/templates/algorithm_project_directory/algorithm_project_template/configuration/__init__.py-template

@@ -0,0 +1,23 @@
+from investing_algorithm_framework.core.context import Context
+from investing_algorithm_framework.core.states.templates.setup_state import SetupState
+from investing_algorithm_framework.core.states.templates.data_providing_state import DataProvidingState
+from investing_algorithm_framework.core.states.templates.strategy_state import StrategyState
+from investing_algorithm_framework.core.states.templates.ordering_state import OrderingState
+
+# Import custom components
+from {{ project_name }}.data_providers.data_providers import MyDataProvider
+from {{ project_name }}.strategies.strategies import MyStrategy
+from {{ project_name }}.order_executors.order_executors import MyOrderExecutor
+
+# Register Initial state
+context = Context()
+context.register_initial_state(SetupState)
+
+# Register all data providers
+DataProvidingState.register_data_providers([MyDataProvider()])
+
+# Register all strategies
+StrategyState.register_strategies([MyStrategy()])
+
+# Register all order executors
+OrderingState.register_order_executors([MyOrderExecutor()])

investing_algorithm_framework/templates/algorithm_project_directory/algorithm_project_template/configuration/context.py-template

@@ -1,9 +1,9 @@
 import os
 from pathlib import Path
 
-BOT_PROJECT_NAME = 'bot'
+PROJECT_NAME = '{{ project_name }}'
 
-BOT_CONTEXT_CONFIGURATION = 'bot.configuration.context'
+CONTEXT_CONFIGURATION = '{{ project_name }}.configuration.context'
 
 # Change this when not in development, feature or hot-fix branch
 DEBUG = int(os.environ.get('DEBUG', True))

investing_algorithm_framework/templates/bot_project_directory/bot_project_template/configuration/settings.py-template renamed to investing_algorithm_framework/templates/algorithm_project_directory/algorithm_project_template/configuration/settings.py-template

@@ -1,5 +1,5 @@
-from investing_bot_framework.core.data_providers import DataProvider
-from investing_bot_framework.core.utils import TimeUnit
+from investing_algorithm_framework.core.data_providers import DataProvider
+from investing_algorithm_framework.core.utils import TimeUnit
 
 """
 Define here all your data providers

investing_algorithm_framework/templates/bot_project_directory/bot_project_template/data_providers/__init__.py-template renamed to investing_algorithm_framework/templates/algorithm_project_directory/algorithm_project_template/data_providers/__init__.py-template

@@ -1,4 +1,4 @@
-from investing_bot_framework.core.order_executors import OrderExecutor
+from investing_algorithm_framework.core.order_executors import OrderExecutor
 
 
 class MyOrderExecutor(OrderExecutor):