Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2025-07-04 20:25:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Disconnect if bad websocket requests instead of flasks abort

Browse Source
This commit is contained in:
Jeremy Zhang
2018-02-16 19:37:12 +00:00
parent 9e53f16eb2
commit a02b33a3cb
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
webapp/titanembeds/oauth.py
View File

@ -3,6 +3,7 @@ import json
from requests_oauthlib import OAuth2Session
from flask import session, abort, url_for
from titanembeds.utils import redis_store, make_user_cache_key
from flask_socketio import disconnect
authorize_url = "https://discordapp.com/api/oauth2/authorize"
token_url = "https://discordapp.com/api/oauth2/token"
@ -49,6 +50,9 @@ def get_user_guilds():
return cache.decode("utf-8")
req = discordrest_from_user("/users/@me/guilds")
if req.status_code != 200:
if getattr(request, sid):
disconnect()
return
abort(req.status_code)
req = json.dumps(req.json())
redis_store.set("OAUTH/USERGUILDS/"+str(make_user_cache_key()), req, 250)