less exceptions raise
This commit is contained in:
@@ -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:
|
||||
|
@@ -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"
|
||||
}
|
||||
|
Reference in New Issue
Block a user