Strankovani

src/decorators/endpoints.py

@@ -3,8 +3,18 @@ from fastapi import HTTPException
 from starlette.status import HTTP_401_UNAUTHORIZED
 
 
-def public_endpoint(func):
-    pass
+def allow_public(func):
+    @wraps(func)
+    async def decorator(*args, **kwargs):
+        if 'info' in kwargs:
+            user_id = kwargs['info'].context.user_id
+            public = kwargs.get('public')
+            if not user_id and not public:
+                raise HTTPException(HTTP_401_UNAUTHORIZED)
+
+            return await func(*args, **kwargs)
+
+    return decorator
 
 
 def authenticated_user_only(raise_when_unauthorized: bool = True, return_value_unauthorized=None):