Coverage for django_napse/core/models/modifications/architecture.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 ArchitectureModification(Modification): 

8 def apply(self): 

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

10 architectur, self = self._apply(architecture) 

11 architecture.save() 

12 self.save() 

13 

14 def _apply(self, **kwargs): 

15 architecture = kwargs.get("architecture") 

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

17 error_msg: str = f"Architecture {architecture} must have attribute {self.key}" 

18 raise ValueError(error_msg) 

19 

20 setattr(architecture, f"variable_{self.key}", process_value_from_type(self.value, self.target_type)) 

21 self.status = MODIFICATION_STATUS.APPLIED 

22 return architecture, self