Coverage for django_napse/core/models/modifications/connection.py: 77%

1from django.db import models 

2 

3from django_napse.utils.constants import MODIFICATION_STATUS 

4 

5from .modification import Modification 

6 

7 

8class ConnectionModification(Modification): 

9 connection_specific_arg = models.ForeignKey("ConnectionSpecificArgs", on_delete=models.CASCADE, related_name="modifications") 

10 

11 def apply(self): 

12 connection_specific_arg, self = self._apply() 

13 connection_specific_arg.save() 

14 self.save() 

15 

16 def _apply(self, **kwargs): 

17 self.connection_specific_arg.value = self.get_value(current_value=self.connection_specific_arg.get_value(), **kwargs) 

18 self.status = MODIFICATION_STATUS.APPLIED 

19 return self.connection_specific_arg, self