testing on identify

webapp/titanembeds/blueprints/gateway/gateway.py

@@ -8,13 +8,11 @@ import json
 from titanembeds.decorators import timeit
 
 class Gateway(Namespace):
-    @timeit
     def teardown_db_session(self):
         time.sleep(0)
         db.session.commit()
         db.session.remove()
 
-    @timeit
     def on_connect(self):
         emit('hello')
 
@@ -52,7 +50,6 @@ class Gateway(Namespace):
         emit("identified")
         self.teardown_db_session()
 
-    @timeit
     def on_disconnect(self):
         time.sleep(0)
         if "user_keys" not in session:
@@ -85,7 +82,6 @@ class Gateway(Namespace):
                         discord_api.delete_webhook(webhook["id"], webhook["token"])
         self.teardown_db_session()
 
-    @timeit
     def on_heartbeat(self, data):
         time.sleep(0)
         if "socket_guild_id" not in session:
@@ -118,7 +114,6 @@ class Gateway(Namespace):
                 return
         self.teardown_db_session()
 
-    @timeit
     def on_channel_list(self, data):
         time.sleep(0)
         if "socket_guild_id" not in session:
@@ -140,7 +135,6 @@ class Gateway(Namespace):
         emit("channel_list", channels)
         self.teardown_db_session()
 
-    @timeit
     def on_current_user_info(self, data):
         time.sleep(0)
         if "socket_guild_id" not in session:
@@ -159,7 +153,6 @@ class Gateway(Namespace):
             emit("current_user_info", usr)
         self.teardown_db_session()
 
-    @timeit
     def get_user_color(self, guild_id, user_id):
         color = None
         member = redisqueue.get_guild_member(guild_id, user_id)
@@ -188,7 +181,6 @@ class Gateway(Namespace):
                     color = "0" + color
         return color
 
-    @timeit
     def on_lookup_user_info(self, data):
         time.sleep(0)
         if "socket_guild_id" not in session:

webapp/titanembeds/redisqueue.py

@@ -1,6 +1,7 @@
 from titanembeds.utils import redis_store
 import json
 import time
+from titanembeds.decorators import timeit
 
 class RedisQueue:
     def __init__(self):
@@ -93,6 +94,7 @@ class RedisQueue:
         sorted_msgs = sorted(msgs, key=lambda k: k['id'], reverse=True) 
         return sorted_msgs[:50] # only return last 50 messages in cache please
 
+    @timeit
     def get_guild_member(self, guild_id, user_id):
         key = "/guilds/{}/members/{}".format(guild_id, user_id)
         q = self.get(key, "get_guild_member", {"guild_id": guild_id, "user_id": user_id})

webapp/titanembeds/utils.py

@@ -13,6 +13,7 @@ import string
 import hashlib
 import time
 import json
+from titanembeds.decorators import timeit
 
 redis_store = FlaskRedis(charset="utf-8", decode_responses=True)
 
@@ -79,6 +80,7 @@ def check_guild_existance(guild_id):
     else:
         return True
 
+@timeit
 def guild_accepts_visitors(guild_id):
     dbGuild = Guilds.query.filter_by(guild_id=guild_id).first()
     return dbGuild.visitor_view
@@ -191,6 +193,7 @@ def get_online_embed_user_keys(guild_id="*", user_type=None):
             usrs[utype].append(client_key)
     return usrs
 
+@timeit
 def check_user_in_guild(guild_id):
     if user_unauthenticated():
         return guild_id in session.get("user_keys", {})
@@ -207,6 +210,7 @@ def get_member_roles(guild_id, user_id):
         role_converted.append(str(role))
     return role_converted
 
+@timeit
 def get_guild_channels(guild_id, force_everyone=False, forced_role=0):
     if user_unauthenticated() or force_everyone:
         member_roles = [guild_id] #equivilant to @everyone role
@@ -315,6 +319,7 @@ def get_channel_permission(channel, guild_id, guild_owner, guild_roles, member_r
         result["attach_files"] = False
     return result
     
+@timeit
 def get_forced_role(guild_id):
     dbguild = db.session.query(Guilds).filter(Guilds.guild_id == guild_id).first()
     if not session.get("unauthenticated", True):