Coverage for django_napse/core/models/modifications/strategy.py: 65%

1from django_napse.utils.constants import MODIFICATION_STATUS 

2from django_napse.utils.usefull_functions import process_value_from_type 

3 

4from .modification import Modification 

5 

6 

7class StrategyModification(Modification): 

8 def apply(self): 

9 strategy = self.order.connection.bot.strategy.find() 

10 strategy, self = self._apply(strategy) 

11 strategy.save() 

12 self.save() 

13 

14 def _apply(self, **kwargs): 

15 strategy = kwargs.get("strategy") 

16 if not hasattr(strategy, f"variable_{self.key}"): 

17 error_msg: str = f"Strategy {strategy} must have attribute variable_{self.key}" 

18 raise ValueError(error_msg) 

19 setattr(strategy, f"variable_{self.key}", process_value_from_type(self.value, self.target_type)) 

20 self.status = MODIFICATION_STATUS.APPLIED 

21 return strategy, self