utg-base 1.9.1__tar.gz → 1.10.0__tar.gz

{utg_base-1.9.1 → utg_base-1.10.0}/PKG-INFO

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

{utg_base-1.9.1 → utg_base-1.10.0}/pyproject.toml

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

{utg_base-1.9.1 → utg_base-1.10.0}/src/utg_base/env.py

@@ -6,9 +6,9 @@ from django.conf import settings
 
 dotenv.load_dotenv(settings.BASE_DIR / '.env')
 
-assert os.environ.get('VAULT_URL') is not None, ("VAULT_URL not defined, please set VAULT_URL on .env file")
-assert os.environ.get('VAULT_TOKEN') is not None, ("VAULT_TOKEN not defined, please set VAULT_TOKEN on .env file")
-assert os.environ.get('VAULT_PATH') is not None, ("VAULT_PATH not defined, please set VAULT_PATH on .env file")
+assert os.environ.get('VAULT_URL') is not None, "VAULT_URL not defined, please set VAULT_URL on .env file"
+assert os.environ.get('VAULT_TOKEN') is not None, "VAULT_TOKEN not defined, please set VAULT_TOKEN on .env file"
+assert os.environ.get('VAULT_PATH') is not None, "VAULT_PATH not defined, please set VAULT_PATH on .env file"
 
 client = hvac.Client(
     url=os.environ.get("VAULT_URL"),

utg_base-1.10.0/src/utg_base/utils/permission.py

@@ -0,0 +1,24 @@
+from typing import Literal
+
+from django_redis import get_redis_connection
+from redis.client import Redis
+
+
+def generate_perm_cache_key(user_id: str):
+    return f"user:{user_id}:permissions"
+
+
+def has_perm(user_id, perm):
+    redis_conn: Redis = get_redis_connection("shared")
+
+    return bool(redis_conn.sismember(f'user:{user_id}:permissions', perm))
+
+
+def has_perms(user_id: str, perms: list[str], operator: Literal["OR", "AND"] = "OR"):
+    redis_conn: Redis = get_redis_connection("shared")
+
+    result = redis_conn.smismember(f'user:{user_id}:permissions', perms)
+
+    if operator == "OR":
+        return any(result)
+    return all(result)