Skip to content

Commit 0bc6d4e

Browse files
committed
Add Strategy template
1 parent fa4a13c commit 0bc6d4e

File tree

3 files changed

+24
-4
lines changed

3 files changed

+24
-4
lines changed
Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,18 @@
11
from investing_bot_framework.core.context import BotContext
22
from investing_bot_framework.core.context.states.setup_state import SetupState
3-
from investing_bot_framework.core.context.states.data_provider_state import DataProviderState
3+
from investing_bot_framework.core.context.states.data_providing_state import DataProvidingState
4+
from investing_bot_framework.core.context.states.strategy_state import StrategyState
45

6+
# Import custom components
57
from bot.data_providers.data_providers import MyDataProvider
8+
from bot.strategies.strategies import MyStrategy
69

710
# Register Initial state
811
context = BotContext()
912
context.register_initial_state(SetupState)
1013

1114
# Register all data providers
12-
DataProviderState.register_data_providers([MyDataProvider()])
15+
DataProvidingState.register_data_providers([MyDataProvider()])
1316

17+
# Register all strategies
18+
StrategyState.register_strategies([MyStrategy()])

investing_bot_framework/templates/bot_project_directory/bot_project_template/data_providers/data_providers.py-template

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,13 @@ from investing_bot_framework.core.data_providers import DataProvider
22
from investing_bot_framework.core.utils import TimeUnit
33

44
"""
5-
Define here all you data providers, e.g. Rest API client
5+
Define here all your data providers
66
"""
77

88

99
class MyDataProvider(DataProvider):
1010
time_unit = TimeUnit.SECOND
11-
time_interval = 1
11+
time_interval = 10
1212
id = 'my_data_provider'
1313

1414
def provide_data(self) -> None:
Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,15 @@
1+
from investing_bot_framework.core.strategies import Strategy
2+
from investing_bot_framework.core.utils import TimeUnit
3+
4+
"""
5+
Define here all your strategies
6+
"""
7+
8+
9+
class MyStrategy(Strategy):
10+
time_unit = TimeUnit.SECOND
11+
time_interval = 10
12+
id = 'my_strategy'
13+
14+
def apply_strategy(self) -> None:
15+
pass

0 commit comments

Comments
 (0)