utg-base 1.1.0__tar.gz → 1.1.1__tar.gz

{utg_base-1.1.0 → utg_base-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: utg-base
-Version: 1.1.0
+Version: 1.1.1
 Summary: UTG Base Package
 Author: Rovshen
 Author-email: rovshenashirov1619@gmail.com

{utg_base-1.1.0 → utg_base-1.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "utg-base"
-version = "1.1.0"
+version = "1.1.1"
 description = "UTG Base Package"
 authors = ["Rovshen <rovshenashirov1619@gmail.com>"]
 readme = "README.md"

utg_base-1.1.1/src/utg_base/api/base.py

@@ -0,0 +1,75 @@
+from django.utils import timezone
+from django.utils.translation import activate, get_language
+from rest_framework.request import Request
+from rest_framework.views import APIView
+
+from ..models import JWTUser
+
+
+AVAILABLE_LANGUAGES = ['uz', 'ru', 'crl']
+
+
+class BaseRequest(Request):
+
+    @property
+    def user(self) -> JWTUser:
+        return super(BaseRequest, self).user()
+
+
+class BaseAPIView(APIView):
+    user_id_assign_fields_on_create = []
+    user_id_assign_fields_on_update = []
+    user_id_assign_fields_on_delete = []
+    deleted_at_field = None
+
+    def __init__(self, **kwargs):
+        # Check user_id_assign_fields_on_create
+        assert isinstance(self.user_id_assign_fields_on_create, list), (
+            'Expected iterable user_id_assign_fields_on_create field'
+        )
+
+        # Check user_id_assign_fields_on_update
+        assert isinstance(self.user_id_assign_fields_on_update, list), (
+            'Expected iterable user_id_assign_fields_on_update field'
+        )
+
+        # Check user_id_assign_fields_on_delete
+        assert isinstance(self.user_id_assign_fields_on_delete, list), (
+            'Expected iterable user_id_assign_fields_on_delete field'
+        )
+
+        # Check deleted_at_field
+        assert isinstance(self.deleted_at_field, (str, type(None))), (
+            'Expected str | None deleted_at_field field'
+        )
+        super().__init__(**kwargs)
+        self.request: BaseRequest | None = None
+
+    def perform_create(self, serializer):
+        kws = {}
+        for attr in getattr(self, 'user_id_assign_fields_on_create', []):
+            kws[attr] = self.request.user.id
+        serializer.save(**kws)
+
+    def perform_update(self, serializer):
+        kws = {}
+        for attr in getattr(self, 'user_id_assign_fields_on_update', []):
+            kws[attr] = self.request.user.id
+        serializer.save(**kws)
+
+    def perform_destroy(self, instance):
+        for attr in getattr(self, 'user_id_assign_fields_on_delete', []):
+            setattr(instance, attr, self.request.user.id)
+        if self.deleted_at_field and hasattr(instance, self.deleted_at_field):
+            setattr(instance, self.deleted_at_field, timezone.now())
+        instance.save()
+
+    def update_lang(self):
+        activate(self.get_language())
+
+    def get_language(self):
+        if lang_from_header := self.request.headers.get('accept-language'):
+            if lang_from_header and lang_from_header not in AVAILABLE_LANGUAGES:
+                lang_from_header = 'ru'
+
+        return lang_from_header or get_language()

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

@@ -0,0 +1,13 @@
+
+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.1.1/src/utg_base/models/__init__.py

@@ -0,0 +1 @@
+from .jwt_user import JWTUser

utg_base-1.1.0/src/utg_base/models/__init__.py

@@ -1,2 +0,0 @@
-from energy_base.models.jwt_user import JWTUser
-from energy_base.models.deviation_reason_type import DeviationReasonType