Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2024-11-15 02:21:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

Made makeshift caching for get user guilds func

Browse Source
This commit is contained in:
Jeremy Zhang 2017-05-08 21:38:47 -07:00
parent 972063ad18
commit ed0131fb10
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
webapp/titanembeds/oauth.py
View File

@ -1,6 +1,9 @@
from config import config from config import config
import json
from requests_oauthlib import OAuth2Session from requests_oauthlib import OAuth2Session
from flask import session, abort, url_for from flask import session, abort, url_for
from titanembeds.database import get_keyvalproperty, set_keyvalproperty
from titanembeds.utils import make_cache_key
authorize_url = "https://discordapp.com/api/oauth2/authorize" authorize_url = "https://discordapp.com/api/oauth2/authorize"
token_url = "https://discordapp.com/api/oauth2/token" token_url = "https://discordapp.com/api/oauth2/token"
@ -42,14 +45,18 @@ def user_has_permission(permission, index):
return bool((int(permission) >> index) & 1) return bool((int(permission) >> index) & 1)
def get_user_guilds(): def get_user_guilds():
cache = get_keyvalproperty("OAUTH/USERGUILDS/"+make_cache_key())
if cache:
return cache
req = discordrest_from_user("/users/@me/guilds") req = discordrest_from_user("/users/@me/guilds")
if req.status_code != 200:
abort(req.status_code)
req = json.dumps(req.json())
set_keyvalproperty("OAUTH/USERGUILDS/"+make_cache_key(), req, 100)
return req return req
def get_user_managed_servers(): def get_user_managed_servers():
guilds = get_user_guilds() guilds = json.loads(get_user_guilds())
if guilds.status_code != 200:
abort(guilds.status_code)
guilds = guilds.json()
filtered = [] filtered = []
for guild in guilds: for guild in guilds:
permission = guild['permissions'] # Manage Server, Ban Members, Kick Members permission = guild['permissions'] # Manage Server, Ban Members, Kick Members