app.backbone.services.bot_service

 1from typing import List
 2from app.backbone.database.db_service import DbService
 3from app.backbone.entities.bot import Bot
 4from app.backbone.services.operation_result import OperationResult
 5
 6
 7class BotService:
 8    def __init__(self):
 9        self.db_service = DbService()
10    
11    def get_bots_by_strategy(self, strategy_id) -> List[Bot]:
12        with self.db_service.get_database() as db:
13            bots = self.db_service.get_many_by_filter(db, Bot, StrategyId=strategy_id)
14            return bots
15            
16    def get_bot(self, strategy_id, ticker_id, timeframe_id, risk) -> Bot:
17        with self.db_service.get_database() as db:
18            bot = self.db_service.get_by_filter(
19                db, 
20                Bot, 
21                StrategyId=strategy_id,
22                TickerId=ticker_id,
23                TimeframeId=timeframe_id,
24                Risk=risk
25            )
26            
27            return bot
28
29    def get_all_bots(self) -> List[Bot]:
30        with self.db_service.get_database() as db:
31            all_bots = self.db_service.get_all(db, Bot)
32            return all_bots
33            
34    def get_bot_by_id(self, bot_id) -> Bot:
35        with self.db_service.get_database() as db:
36            bot = self.db_service.get_by_id(db, Bot, id=bot_id)
37            return bot