managed: allow for matching on multiple interfaces

This commit is contained in:
Jens Langhammer 2021-02-05 13:18:44 +01:00
parent 397dfc29f1
commit fadf746234
1 changed files with 10 additions and 9 deletions

View File

@ -12,12 +12,12 @@ class EnsureOp:
"""Ensure operation, executed as part of an ObjectManager run""" """Ensure operation, executed as part of an ObjectManager run"""
_obj: Type[ManagedModel] _obj: Type[ManagedModel]
_match_field: str _match_fields: list[str]
_kwargs: dict _kwargs: dict
def __init__(self, obj: Type[ManagedModel], match_field: str, **kwargs) -> None: def __init__(self, obj: Type[ManagedModel], *match_fields: str, **kwargs) -> None:
self._obj = obj self._obj = obj
self._match_field = match_field self._match_fields = match_fields
self._kwargs = kwargs self._kwargs = kwargs
def run(self): def run(self):
@ -29,14 +29,15 @@ class EnsureExists(EnsureOp):
"""Ensure object exists, with kwargs as given values""" """Ensure object exists, with kwargs as given values"""
def run(self): def run(self):
matcher_value = self._kwargs.get(self._match_field, None) update_kwargs = {
"managed": True,
"defaults": self._kwargs,
}
for field in self._match_fields:
update_kwargs[field] = self._kwargs.get(field, None)
self._kwargs.setdefault("managed", True) self._kwargs.setdefault("managed", True)
self._obj.objects.update_or_create( self._obj.objects.update_or_create(
**{ **update_kwargs
self._match_field: matcher_value,
"managed": True,
"defaults": self._kwargs,
}
) )