drf-iam 0.0.3__tar.gz → 0.2.0__tar.gz

{drf_iam-0.0.3 → drf_iam-0.2.0}/PKG-INFO

@@ -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.2.0/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 → drf_iam-0.2.0}/drf_iam.egg-info/PKG-INFO

@@ -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 → drf_iam-0.2.0}/drf_iam.egg-info/SOURCES.txt

@@ -13,4 +13,6 @@ drf_iam.egg-info/dependency_links.txt
 drf_iam.egg-info/requires.txt
 drf_iam.egg-info/top_level.txt
 drf_iam/migrations/0001_initial.py
-drf_iam/migrations/__init__.py
+drf_iam/migrations/__init__.py
+drf_iam/utils/__init__.py
+drf_iam/utils/load_viewset_permissions.py

{drf_iam-0.0.3 → drf_iam-0.2.0}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='drf-iam',
-    version='0.0.3',
+    version='0.2.0',
     packages=find_packages(),  # Automatically finds all sub-packages
     include_package_data=True,  # Include non-Python files from MANIFEST.in
     install_requires=[