Coverage for django_napse/core/models/bots/plugins/lbo.py: 94%

1from django_napse.core.models.bots.plugin import Plugin 

2from django_napse.core.models.connections.connection import ConnectionSpecificArgs 

3from django_napse.utils.constants import PLUGIN_CATEGORIES, SIDES 

4 

5 

6class LBOPlugin(Plugin): 

7 @classmethod 

8 def plugin_category(cls): 

9 return PLUGIN_CATEGORIES.POST_ORDER 

10 

11 def _apply(self, data: dict) -> dict: 

12 order = data["order"] 

13 if order["side"] == SIDES.BUY: 

14 order["ConnectionModifications"] += [ 

15 { 

16 "key": "lbo", 

17 "value": data["connection_data"][data["connection"]]["connection_specific_args"]["lbo"].get_value() + 1, 

18 "target_type": "int", 

19 "ignore_failed_order": False, 

20 "connection_specific_arg": data["connection_data"][data["connection"]]["connection_specific_args"]["lbo"], 

21 }, 

22 ] 

23 if order["side"] == SIDES.SELL: 

24 order["ConnectionModifications"] += [ 

25 { 

26 "key": "lbo", 

27 "value": 0, 

28 "target_type": "int", 

29 "ignore_failed_order": False, 

30 "connection_specific_arg": data["connection_data"][data["connection"]]["connection_specific_args"]["lbo"], 

31 }, 

32 ] 

33 return data 

34 

35 def _connect(self, connection): 

36 ConnectionSpecificArgs.objects.create(connection=connection, key="lbo", value="0", target_type="int")