cardo-python-utils 0.5.dev9__tar.gz → 0.5.dev10__tar.gz

{cardo_python_utils-0.5.dev9/cardo_python_utils.egg-info → cardo_python_utils-0.5.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cardo-python-utils
-Version: 0.5.dev9
+Version: 0.5.dev10
 Summary: Python library enhanced with a wide range of functions for different scenarios.
 Author-email: CardoAI <hello@cardoai.com>
 License: MIT

{cardo_python_utils-0.5.dev9 → cardo_python_utils-0.5.dev10/cardo_python_utils.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cardo-python-utils
-Version: 0.5.dev9
+Version: 0.5.dev10
 Summary: Python library enhanced with a wide range of functions for different scenarios.
 Author-email: CardoAI <hello@cardoai.com>
 License: MIT

{cardo_python_utils-0.5.dev9 → cardo_python_utils-0.5.dev10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "cardo-python-utils"
-version = "0.5.dev9"
+version = "0.5.dev10"
 description = "Python library enhanced with a wide range of functions for different scenarios."
 readme = "README.rst"
 requires-python = ">=3.8"

{cardo_python_utils-0.5.dev9 → cardo_python_utils-0.5.dev10}/python_utils/django/keycloak/service.py

@@ -4,10 +4,32 @@ from keycloak import KeycloakAdmin
 from keycloak import KeycloakOpenIDConnection
 from keycloak.exceptions import KeycloakGetError
 
+def _get_user_group_model():
+    """
+    Dynamically get the UserGroup model.
+
+    Attempts to use the model specified in settings.KEYCLOAK_USER_GROUP_MODEL.
+
+    Returns:
+        The UserGroup model class.
+
+    Raises:
+        LookupError: If the model cannot be found.
+    """
+    try:
+        model_string = getattr(settings, "KEYCLOAK_USER_GROUP_MODEL")
+    except AttributeError:
+        raise LookupError(
+            "Please set KEYCLOAK_USER_GROUP_MODEL in your Django settings "
+            "(e.g., 'myapp.UserGroup')."
+        )
+
+    return apps.get_model(model_string)
+
 
 class KeycloakService:
     def __init__(self):
-        self._user_group_model = self._get_user_group_model()
+        self._user_group_model = _get_user_group_model()
         self._keycloak_admin = self._get_keycloak_admin()
 
     def sync_user_groups(self, raise_exceptions: bool = False):
@@ -53,28 +75,6 @@ class KeycloakService:
         )
         return KeycloakAdmin(connection=keycloak_connection)
 
-    def _get_user_group_model(self):
-        """
-        Dynamically get the UserGroup model.
-
-        Attempts to use the model specified in settings.KEYCLOAK_USER_GROUP_MODEL.
-
-        Returns:
-            The UserGroup model class.
-
-        Raises:
-            LookupError: If the model cannot be found.
-        """
-        try:
-            model_string = getattr(settings, "KEYCLOAK_USER_GROUP_MODEL")
-        except AttributeError:
-            raise LookupError(
-                "Please set KEYCLOAK_USER_GROUP_MODEL in your Django settings "
-                "(e.g., 'myapp.UserGroup')."
-            )
-
-        return apps.get_model(model_string)
-
     def _process_group_recursively(
         self, group, existing_groups_by_id, reported_group_ids
     ):
@@ -98,3 +98,33 @@ class KeycloakService:
                 self._process_group_recursively(
                     subgroup, existing_groups_by_id, reported_group_ids
                 )
+
+
+class AuthServiceBase:
+    @staticmethod
+    def _get_all_level_paths(path: str) -> list[str]:
+        """
+        Given a group path, return all level paths up to the root.
+        E.g., for "/a/b/c", return ["/a/b/c", "/a/b", "/a"]
+        """
+        paths = []
+        while "/" in path:
+            paths.append(path)
+            path = "/".join(path.split("/")[:-1])
+        return paths
+
+    @classmethod
+    def _get_user_groups_from_paths(cls, group_paths: list[str]):
+        all_group_paths = set()
+        for path in group_paths:
+            all_group_paths.update(cls._get_all_level_paths(path))
+
+        UserGroup = _get_user_group_model()
+        user_groups = UserGroup.objects.filter(path__in=all_group_paths)
+
+        # If a group is missing/has been renamed in Keycloak, sync the groups
+        if user_groups.count() != len(all_group_paths):
+            KeycloakService().sync_user_groups()
+            user_groups = UserGroup.objects.filter(path__in=all_group_paths)
+
+        return user_groups