From fadf746234981b3aecca5c49aab919fb39de58b4 Mon Sep 17 00:00:00 2001
From: Jens Langhammer <jens.langhammer@beryju.org>
Date: Fri, 5 Feb 2021 13:18:44 +0100
Subject: [PATCH] managed: allow for matching on multiple interfaces

---
 authentik/managed/manager.py | 19 ++++++++++---------
 1 file changed, 10 insertions(+), 9 deletions(-)

diff --git a/authentik/managed/manager.py b/authentik/managed/manager.py
index 259259979..654b3772d 100644
--- a/authentik/managed/manager.py
+++ b/authentik/managed/manager.py
@@ -12,12 +12,12 @@ class EnsureOp:
     """Ensure operation, executed as part of an ObjectManager run"""
 
     _obj: Type[ManagedModel]
-    _match_field: str
+    _match_fields: list[str]
     _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._match_field = match_field
+        self._match_fields = match_fields
         self._kwargs = kwargs
 
     def run(self):
@@ -29,14 +29,15 @@ class EnsureExists(EnsureOp):
     """Ensure object exists, with kwargs as given values"""
 
     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._obj.objects.update_or_create(
-            **{
-                self._match_field: matcher_value,
-                "managed": True,
-                "defaults": self._kwargs,
-            }
+            **update_kwargs
         )