Moznost nahravat fotky copilotem

src/decorators/endpoints.py

@@ -3,6 +3,10 @@ from fastapi import HTTPException
 from starlette.status import HTTP_401_UNAUTHORIZED
 
 
+def raise_unauthorized():
+    raise HTTPException(HTTP_401_UNAUTHORIZED, "Not authorized")
+
+
 def allow_public(func):
     @wraps(func)
     async def decorator(*args, **kwargs):
@@ -10,7 +14,7 @@ def allow_public(func):
             user_id = kwargs['info'].context.user_id
             public = kwargs.get('public')
             if not user_id and not public:
-                raise HTTPException(HTTP_401_UNAUTHORIZED)
+                raise_unauthorized()
 
             return await func(*args, **kwargs)
 
@@ -24,7 +28,7 @@ def authenticated_user_only(raise_when_unauthorized: bool = True, return_value_u
             if 'info' in kwargs:
                 if not kwargs['info'].context.user_id:
                     if raise_when_unauthorized:
-                        raise HTTPException(HTTP_401_UNAUTHORIZED, "Not authorized")
+                        raise_unauthorized()
                     else:
                         return return_value_unauthorized
             return await func(*args, **kwargs)