drf-iam 0.0.3__py3-none-any.whl → 0.2.0__py3-none-any.whl

drf_iam/utils/load_viewset_permissions.py

@@ -0,0 +1,71 @@
+import inspect
+
+from django.template.base import kwarg_re
+from django.urls.resolvers import URLPattern, URLResolver
+from rest_framework.viewsets import ViewSetMixin
+from rest_framework.decorators import action
+
+from django.urls import get_resolver
+from ..models import Policy
+
+
+def is_viewset(cls):
+    return isinstance(cls, type) and issubclass(cls, ViewSetMixin)
+
+
+def get_viewset_actions(viewset_cls):
+    actions = set()
+
+    # Default ViewSet methods
+    default_actions = {'list', 'retrieve', 'create', 'update', 'partial_update', 'destroy'}
+    for action_name in default_actions:
+        if hasattr(viewset_cls, action_name):
+            actions.add(action_name)
+
+    # Custom @action methods
+    for name, method in inspect.getmembers(viewset_cls, predicate=inspect.isfunction):
+        if hasattr(method, 'mapping'):
+            for http_method in method.mapping:
+                viewset_function = getattr(viewset_cls, name)
+                if viewset_function.get("kwargs"):
+                    actions.add(f"{name}")
+
+    return actions
+
+
+def extract_viewsets_from_urlpatterns(urlpatterns, prefix=''):
+    for pattern in urlpatterns:
+        if isinstance(pattern, URLResolver):
+            # Recurse into included URLConfs
+            yield from extract_viewsets_from_urlpatterns(pattern.url_patterns, prefix + str(pattern.pattern))
+        elif isinstance(pattern, URLPattern):
+            callback = pattern.callback
+            cls = getattr(callback, 'cls', None)
+            if is_viewset(cls):
+                yield {
+                    'prefix': prefix,
+                    'pattern': pattern.pattern,
+                    'viewset': cls,
+                    'callback': callback
+                }
+
+
+def load_permissions_from_urls(**kwargs):
+    urlpatterns = get_resolver().url_patterns
+    for entry in extract_viewsets_from_urlpatterns(urlpatterns):
+        viewset_cls = entry['viewset']
+        basename = getattr(viewset_cls,"iam_policy_name") or viewset_cls.__name__.lower().replace('viewset', '')
+        exclude_from_permissions = getattr(viewset_cls, "drf_iam_exclude_from_permissions", False)
+        if exclude_from_permissions:
+            continue
+        actions = get_viewset_actions(viewset_cls)
+
+        for action in actions:
+            full_action = f"{basename}:{action}"
+            Policy.objects.get_or_create(
+                action=full_action,
+                resource_type=basename,
+                defaults={
+                    'description': f"Permission for {full_action} on {basename}"
+                }
+            )

{drf_iam-0.0.3.dist-info → drf_iam-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: drf-iam
-Version: 0.0.3
+Version: 0.2.0
 Summary: IAM-style roles and permissions for Django Rest Framework
 Home-page: https://github.com/tushar1328/drf-iam.git
 Author: Tushar Patel

{drf_iam-0.0.3.dist-info → drf_iam-0.2.0.dist-info}/RECORD

@@ -6,7 +6,9 @@ drf_iam/permissions.py,sha256=kUhB7V48sb4v3uZI2lWv40hvI85nE61rL-amh6OY8mg,958
 drf_iam/tests.py,sha256=mrbGGRNg5jwbTJtWWa7zSKdDyeB4vmgZCRc2nk6VY-g,60
 drf_iam/migrations/0001_initial.py,sha256=y_4jXnr7gjU4UXxVrgVrStTSFu3h1ZrmjEZDL4FtZa4,3086
 drf_iam/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-drf_iam-0.0.3.dist-info/METADATA,sha256=TX_Bn6q0VhHzdnDd-LgsjfbyhZe4T1cZ7zLkUWpeSyc,2763
-drf_iam-0.0.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-drf_iam-0.0.3.dist-info/top_level.txt,sha256=daz6AaQ9e_cfCjLk2aRoLb_PCOoFofYUX4DU85VwHSM,8
-drf_iam-0.0.3.dist-info/RECORD,,
+drf_iam/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+drf_iam/utils/load_viewset_permissions.py,sha256=xWV9-8qwuOYysofFcKHxhbvh4u-fOH-Wy37ysvx2XTw,2574
+drf_iam-0.2.0.dist-info/METADATA,sha256=u1HeciXJ3Ndr1G3OQylQjH51ycaYniTFClHKC80xX1U,2763
+drf_iam-0.2.0.dist-info/WHEEL,sha256=ck4Vq1_RXyvS4Jt6SI0Vz6fyVs4GWg7AINwpsaGEgPE,91
+drf_iam-0.2.0.dist-info/top_level.txt,sha256=daz6AaQ9e_cfCjLk2aRoLb_PCOoFofYUX4DU85VwHSM,8
+drf_iam-0.2.0.dist-info/RECORD,,

{drf_iam-0.0.3.dist-info → drf_iam-0.2.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (78.1.0)
+Generator: setuptools (80.0.0)
 Root-Is-Purelib: true
 Tag: py3-none-any