utg-base 1.13.2__tar.gz → 1.14.0__tar.gz

{utg_base-1.13.2 → utg_base-1.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: utg-base
-Version: 1.13.2
+Version: 1.14.0
 Summary: UTG Base Package
 Author: Olimboy
 Author-email: shavkatov.olimboy@mail.ru

{utg_base-1.13.2 → utg_base-1.14.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "utg-base"
-version = "1.13.2"
+version = "1.14.0"
 description = "UTG Base Package"
 authors = ["Olimboy <shavkatov.olimboy@mail.ru>", "Rovshen <rovshenashirov1619@gmail.com>"]
 readme = "README.md"

utg_base-1.14.0/src/utg_base/permissions/decorators.py

@@ -0,0 +1,65 @@
+from functools import wraps
+from typing import Literal, Dict, Union, List
+
+from rest_framework.exceptions import PermissionDenied
+
+from .utils import has_perms as has_permissions
+
+
+def has_perm(*permission: str, operator: Literal["OR", "AND"] = "OR"):
+    """
+    Usage:
+        has_perm('PERM1')
+        has_perm('PERM1', 'PERM2')
+
+    :param permission: One or many permissions.
+    """
+    def decorator(view_func):
+        view_func._perms = permission
+        @wraps(view_func)
+        def _wrapped_view(self, request, *args, **kwargs):
+            if not has_permissions(user_id=request.user.id, perms=list(permission), operator=operator):
+                raise PermissionDenied
+            return view_func(self, request, *args, **kwargs)
+        return _wrapped_view
+    return decorator
+
+
+def has_class_perm(permissions_map: Dict[str, Union[str, List[str]]], operator: Literal["OR", "AND"] = "OR"):
+    """
+    Usage:
+        @has_class_perm({
+            'list': ['perm1', 'perm2'],
+            'create': 'perm3',
+            'update': ['perm4']
+        })
+    """
+    def decorator(cls):
+        for action, perms in permissions_map.items():
+            if isinstance(perms, str):
+                perms = [perms]
+
+            if hasattr(cls, action):
+                method = getattr(cls, action)
+                setattr(method, "_perms", perms)
+        original_initial = cls.initial
+
+        @wraps(original_initial)
+        def new_initial(self, request, *args, **kwargs):
+            action = getattr(self, 'action', None) or request.method.lower()
+
+            if action in permissions_map:
+                required_perms = permissions_map[action]
+
+                if isinstance(required_perms, str):
+                    required_perms = [required_perms]
+
+                if not has_permissions(user_id=request.user.id, perms=required_perms, operator=operator):
+                    raise PermissionDenied
+
+            return original_initial(self, request, *args, **kwargs)
+
+        cls.initial = new_initial
+        return cls
+
+    return decorator

{utg_base-1.13.2 → utg_base-1.14.0}/src/utg_base/references_api/utils.py

@@ -8,7 +8,7 @@ from drf_spectacular.utils import extend_schema, extend_schema_serializer
 from rest_framework import filters
 from rest_framework import viewsets, serializers
 
-from ..api.views import TranslatedListView
+from utg_base.permissions.decorators import has_class_perm
 
 serializer_classes = {}
 
@@ -82,7 +82,7 @@ def create_view_set(model: Model):
     fields_list = get_model_fields_list(model)
 
     @extend_schema(tags=[get_url_prefix(model)])
-    class ViewSet(viewsets.ModelViewSet, TranslatedListView):
+    class ViewSet(viewsets.ModelViewSet):
         http_method_names = get_api_meta_property(model, 'http_method_names') or ['get', 'patch']
         manager = model.objects.order_by(*get_ordering(model))
         filter_backends = [filters.SearchFilter]
@@ -98,6 +98,9 @@ def create_view_set(model: Model):
 
             return create_serializer(model)
 
+    if permissions := get_api_meta_property(model, 'permissions'):
+        has_class_perm(permissions)(ViewSet)
+
     return ViewSet
 
 

utg_base-1.13.2/src/utg_base/api/views.py

@@ -1,13 +0,0 @@
-
-from django.db.models import Manager
-from rest_framework import generics
-
-from .base import BaseAPIView
-
-
-class TranslatedListView(generics.ListAPIView, BaseAPIView):
-    manager: Manager
-
-    def get_queryset(self):
-        self.update_lang()
-        return self.manager.all()

utg_base-1.13.2/src/utg_base/permissions/decorators.py

@@ -1,25 +0,0 @@
-from functools import wraps
-from typing import Literal
-
-from rest_framework.exceptions import PermissionDenied
-
-from .utils import has_perms as has_permissions
-
-
-def has_perm(*permission: str, operator: Literal["OR", "AND"] = "OR"):
-    """
-    Usage:
-        has_perm('PERM1')
-        has_perm('PERM1', 'PERM2')
-
-    :param permission: One or many permissions.
-    """
-    def decorator(view_func):
-        view_func._perms = permission
-        @wraps(view_func)
-        def _wrapped_view(self, request, *args, **kwargs):
-            if not has_permissions(user_id=request.user.id, perms=list(permission), operator=operator):
-                raise PermissionDenied
-            return view_func(self, request, *args, **kwargs)
-        return _wrapped_view
-    return decorator

utg_base-1.13.2/src/utg_base/references_api/admin.py

@@ -1,3 +0,0 @@
-from django.contrib import admin
-
-# Register your models here.

utg_base-1.13.2/src/utg_base/references_api/models.py

@@ -1,3 +0,0 @@
-from django.db import models
-
-# Create your models here.