Initial structure for discordbot addition

webapp/titanembeds/utils.py

@@ -0,0 +1,77 @@
+from beaker.cache import CacheManager
+from beaker.util import parse_cache_config_options
+from titanembeds.database import db, Guilds, KeyValueProperties
+from flask import request, session
+from flask_limiter import Limiter
+from config import config
+import random
+import string
+import hashlib
+
+cache_opts = {
+    'cache.type': 'ext:database',
+    'cache.lock_dir': 'tmp/cachelock',
+    'cache.url': config["database-uri"],
+    'cache.sa.pool_recycle': 250,
+}
+cache = CacheManager(**parse_cache_config_options(cache_opts))
+
+from titanembeds.discordrest import DiscordREST
+
+discord_api = DiscordREST(config['bot-token'])
+
+def get_client_ipaddr():
+    if "X-Real-IP" in request.headers: # pythonanywhere specific
+        ip = request.headers['X-Real-IP']
+    else: # general
+        ip = request.remote_addr
+    return hashlib.sha512(config['app-secret'] + ip).hexdigest()[:15]
+
+def generate_session_key():
+    sess = session.get("sessionunique", None)
+    if not sess:
+        rand_str = lambda n: ''.join([random.choice(string.lowercase) for i in xrange(n)])
+        session['sessionunique'] = rand_str(25)
+        sess = session['sessionunique']
+    return sess #Totally unique
+
+def make_cache_key(*args, **kwargs):
+    path = request.path
+    args = str(hash(frozenset(request.args.items())))
+    ip = get_client_ipaddr()
+    sess = generate_session_key()
+    return (path + args + sess + ip).encode('utf-8')
+
+def make_guilds_cache_key():
+    sess = generate_session_key()
+    ip = get_client_ipaddr()
+    return (sess + ip + "user_guilds").encode('utf-8')
+
+def make_guildchannels_cache_key():
+    guild_id = request.values.get('guild_id', "0")
+    sess = generate_session_key()
+    ip = get_client_ipaddr()
+    return (sess + ip + guild_id + "user_guild_channels").encode('utf-8')
+
+def channel_ratelimit_key(): # Generate a bucket with given channel & unique session key
+    sess = generate_session_key()
+    channel_id = request.values.get('channel_id', "0")
+    return (sess + channel_id).encode('utf-8')
+
+def guild_ratelimit_key():
+    ip = get_client_ipaddr()
+    guild_id = request.values.get('guild_id', "0")
+    return (ip + guild_id).encode('utf-8')
+
+def check_guild_existance(guild_id):
+    dbGuild = Guilds.query.filter_by(guild_id=guild_id).first()
+    if not dbGuild:
+        return False
+    guild = discord_api.get_guild(guild_id)
+    return guild['code'] == 200
+
+def guild_query_unauth_users_bool(guild_id):
+    dbGuild = db.session.query(Guilds).filter(Guilds.guild_id==guild_id).first()
+    return dbGuild.unauth_users
+
+rate_limiter = Limiter(key_func=get_client_ipaddr) # Default limit by ip address