Coverage for src/finbot/strategy/strategies.py: 100%

1import random 

2 

3from finbot.market.models import Bar 

4from finbot.strategy.base import Strategy 

5from finbot.strategy.signal import Signal, SignalDirection 

6 

7 

8class RandomStrategy(Strategy): 

9 def on_bar(self, bar: Bar) -> Signal | None: 

10 n = random.randint(1, 100) 

11 if n == 42: 

12 return Signal(symbol=bar.symbol, direction=SignalDirection.LONG) 

13 if n == 69: 

14 return Signal(symbol=bar.symbol, direction=SignalDirection.SHORT) 

15 

16 return None 

17 

18 

19class BuyOnceStrategy(Strategy): 

20 def __init__(self) -> None: 

21 self._signaled_symbols: set[str] = set() 

22 

23 def on_bar(self, bar: Bar) -> Signal | None: 

24 if bar.symbol in self._signaled_symbols: 

25 return None 

26 

27 self._signaled_symbols.add(bar.symbol) 

28 

29 return Signal(symbol=bar.symbol, direction=SignalDirection.LONG)