utg-base 1.21.2__tar.gz → 1.22.0__tar.gz

{utg_base-1.21.2 → utg_base-1.22.0}/PKG-INFO

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

{utg_base-1.21.2 → utg_base-1.22.0}/pyproject.toml

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

{utg_base-1.21.2 → utg_base-1.22.0}/src/utg_base/permissions/decorators.py

@@ -1,8 +1,6 @@
 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
 
 
@@ -13,15 +11,20 @@ def has_perm(*permission: str, operator: Literal["OR", "AND"] = "OR"):
         has_perm('PERM1', 'PERM2')
 
     :param permission: One or many permissions.
+    :param operator: Any of (OR, AND)
     """
+
     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, request=request):
                 self.permission_denied(request)
             return view_func(self, request, *args, **kwargs)
+
         return _wrapped_view
+
     return decorator
 
 
@@ -34,6 +37,7 @@ def has_class_perm(permissions_map: Dict[str, Union[str, List[str]]], operator:
             'update': ['perm4']
         })
     """
+
     def decorator(cls):
         for action, perms in permissions_map.items():
             if isinstance(perms, str):

{utg_base-1.21.2 → utg_base-1.22.0}/src/utg_base/utils/data.py

@@ -1,8 +1,12 @@
+import datetime
+from collections import defaultdict
 from collections.abc import Iterable
 from copy import deepcopy
 
 from django.db.models import QuerySet
 
+from utg_base.utils.date import UDate, UDateTime
+
 
 def deep_map(data: dict | list, func_cond, func_map, in_place=True):
     if not in_place:
@@ -153,3 +157,35 @@ def compute_change_percent(current, previous, ndigits=2):
     if not current or previous in (None, 0):
         return None
     return round(((current - previous) / previous) * 100, ndigits)
+
+
+def list_to_dict(rows: Iterable[dict], key: str, key_func=None) -> dict:
+    if not key_func:
+        key_func = lambda x: x
+    return {row[key_func(key)]: row for row in rows}
+
+
+def generate_series(start: UDate | UDateTime, end: UDate | UDateTime, step: datetime.timedelta):
+    current = start
+    if step.total_seconds() == 0:
+        raise ValueError("step cannot be zero")
+
+    series = []
+    while current <= end:
+        series.append(current)
+        current += step
+
+    return series
+
+
+def fill_series(series: Iterable, data: dict, keys: str | list[str]):
+    result = defaultdict(list)
+
+    for term in series:
+        for key in keys:
+            result[key].append(data.get(term, {}).get(key, None))
+
+    data = []
+    for key in keys:
+        data.append(result[key])
+    return data