less exceptions raise

This commit is contained in:
2023-01-10 11:15:28 +03:00
parent 4e0582e513
commit 3ec349eda1
5 changed files with 46 additions and 16 deletions

View File

@@ -1,4 +1,4 @@
from base.exceptions import Unauthorized
# from base.exceptions import Unauthorized
from auth.tokenstorage import SessionToken
from base.redis import redis
@@ -17,7 +17,10 @@ async def on_connect(req, params):
return
token = params.get('token')
if not token:
raise Unauthorized("Please login")
# raise Unauthorized("Please login")
return {
"error": "Please login first"
}
else:
payload = await SessionToken.verify(token)
if payload and payload.user_id:

View File

@@ -2,7 +2,7 @@ from sse_starlette.sse import EventSourceResponse
from starlette.requests import Request
from graphql.type import GraphQLResolveInfo
from resolvers.inbox.messages import message_generator
from base.exceptions import Unauthorized
# from base.exceptions import Unauthorized
# https://github.com/enisdenjo/graphql-sse/blob/master/PROTOCOL.md
@@ -16,4 +16,7 @@ async def sse_messages(request: Request):
event_generator = await message_generator(None, info)
return EventSourceResponse(event_generator)
else:
raise Unauthorized("Please login")
# raise Unauthorized("Please login")
return {
"error": "Please login first"
}