Vetsi sada zmen

src/decorators/endpoints.py

@@ -7,13 +7,19 @@ def public_endpoint(func):
     pass
 
 
-def authenticated_user_only(func):
-    @wraps(func)
-    async def decorator(*args, **kwargs):
-        if 'info' in kwargs:
-            if not kwargs['info'].context.user_id:
-                raise HTTPException(HTTP_401_UNAUTHORIZED, "Not authorized")
+def authenticated_user_only(raise_when_unauthorized: bool = True, return_value_unauthorized=None):
+    def wrapper(func):
+        @wraps(func)
+        async def decorator(*args, **kwargs):
+            if 'info' in kwargs:
+                if not kwargs['info'].context.user_id:
+                    if raise_when_unauthorized:
+                        raise HTTPException(HTTP_401_UNAUTHORIZED, "Not authorized")
+                    else:
+                        return return_value_unauthorized
 
-        return await func(*args, **kwargs)
+            return await func(*args, **kwargs)
 
-    return decorator
+        return decorator
+
+    return wrapper