request is not defined

webapp/titanembeds/.~c9_invoke_ZmFTgg.py

@@ -0,0 +1,108 @@
+from config import config
+import json
+from requests_oauthlib import OAuth2Session
+from flask import session, abort, url_for, request
+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"
+avatar_base_url = "https://cdn.discordapp.com/avatars/"
+guild_icon_url = "https://cdn.discordapp.com/icons/"
+
+def update_user_token(discord_token):
+    session['user_keys'] = discord_token
+
+def make_authenticated_session(token=None, state=None, scope=None):
+    return OAuth2Session(
+        client_id=config['client-id'],
+        token=token,
+        state=state,
+        scope=scope,
+        redirect_uri=url_for("user.callback", _external=True),
+        auto_refresh_kwargs={
+            'client_id': config['client-id'],
+            'client_secret': config['client-secret'],
+        },
+        auto_refresh_url=token_url,
+        token_updater=update_user_token,
+    )
+
+def discordrest_from_user(endpoint):
+    token = session['user_keys']
+    discord = make_authenticated_session(token=token)
+    req = discord.get("https://discordapp.com/api/v6{}".format(endpoint))
+    return req
+
+def get_current_authenticated_user():
+    req = discordrest_from_user("/users/@me")
+    if req.status_code != 200:
+        abort(req.status_code)
+    user = req.json()
+    return user
+
+def user_has_permission(permission, index):
+    return bool((int(permission) >> index) & 1)
+
+def get_user_guilds():
+    cache = redis_store.get("OAUTH/USERGUILDS/"+str(make_user_cache_key()))
+    if cache:
+        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)
+    return req
+
+def get_user_managed_servers():
+    guilds = json.loads(get_user_guilds())
+    filtered = []
+    for guild in guilds:
+        permission = guild['permissions'] # Manage Server, Ban Members, Kick Members
+        if guild['owner'] or user_has_permission(permission, 5) or user_has_permission(permission, 2) or user_has_permission(permission, 1):
+            filtered.append(guild)
+    filtered = sorted(filtered, key=lambda guild: guild['name'])
+    return filtered
+
+def get_user_managed_servers_safe():
+    guilds = get_user_managed_servers()
+    if guilds:
+        return guilds
+    return []
+
+def get_user_managed_servers_id():
+    guilds = get_user_managed_servers_safe()
+    ids=[]
+    for guild in guilds:
+        ids.append(guild['id'])
+    return ids
+
+def check_user_can_administrate_guild(guild_id):
+    guilds = get_user_managed_servers_id()
+    return guild_id in guilds
+
+def check_user_permission(guild_id, id):
+    guilds = get_user_managed_servers_safe()
+    for guild in guilds:
+        if guild['id'] == guild_id:
+            return user_has_permission(guild['permissions'], id) or guild['owner']
+    return False
+
+def generate_avatar_url(id, av, discrim="0000"):
+    if av:
+        return avatar_base_url + str(id) + '/' + str(av) + '.jpg'
+    else:
+        default_av = [0, 1, 2, 3, 4]
+        discrim = int(discrim)
+        return "https://cdn.discordapp.com/embed/avatars/{}.png".format(default_av[int(discrim) % len(default_av)])
+
+def generate_guild_icon_url(id, hash):
+    return guild_icon_url + str(id) + "/" + str(hash) + ".jpg"
+
+def generate_bot_invite_url(guild_id):
+    url = "https://discordapp.com/oauth2/authorize?&client_id={}&scope=bot&permissions={}&guild_id={}".format(config['client-id'], '641195117', guild_id)
+    return url

webapp/titanembeds/oauth.py

@@ -1,7 +1,7 @@
 from config import config
 import json
 from requests_oauthlib import OAuth2Session
-from flask import session, abort, url_for
+from flask import session, abort, url_for, request
 from titanembeds.utils import redis_store, make_user_cache_key
 from flask_socketio import disconnect