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

Cache authors during fetching messages and insert latest authorship

Browse Source
This commit is contained in:
Jeremy Zhang 2018-01-25 23:26:47 +00:00
parent a2f378da57
commit 8c3e3c256b
1 changed files with 17 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
webapp/titanembeds/database/messages.py
View File

@ -37,6 +37,7 @@ def get_channel_messages(guild_id, channel_id, after_snowflake=None):
q = db.session.query(Messages).filter(Messages.channel_id == channel_id).filter(Messages.message_id > after_snowflake).order_by(Messages.timestamp.desc()).limit(50) q = db.session.query(Messages).filter(Messages.channel_id == channel_id).filter(Messages.message_id > after_snowflake).order_by(Messages.timestamp.desc()).limit(50)
msgs = [] msgs = []
snowflakes = [] snowflakes = []
guild_members = {}
for x in q: for x in q:
if x.message_id in snowflakes: if x.message_id in snowflakes:
continue continue
@ -55,14 +56,28 @@ def get_channel_messages(guild_id, channel_id, after_snowflake=None):
"mentions": json.loads(x.mentions), "mentions": json.loads(x.mentions),
"embeds": json.loads(embeds), "embeds": json.loads(embeds),
} }
member = get_guild_member(guild_id, message["author"]["id"]) if message["author"]["id"] not in guild_members:
member = get_guild_member(guild_id, message["author"]["id"])
guild_members[message["author"]["id"]] = member
else:
member = guild_members[message["author"]["id"]]
message["author"]["nickname"] = None message["author"]["nickname"] = None
if member: if member:
message["author"]["nickname"] = member.nickname message["author"]["nickname"] = member.nickname
message["author"]["avatar"] = member.avatar
message["author"]["discriminator"] = member.discriminator
message["author"]["username"] = member.username
for mention in message["mentions"]: for mention in message["mentions"]:
author = get_guild_member(guild_id, mention["id"]) if mention["id"] not in guild_members:
author = get_guild_member(guild_id, mention["id"])
guild_members[mention["id"]] = author
else:
author = guild_members[mention["id"]]
mention["nickname"] = None mention["nickname"] = None
if author: if author:
mention["nickname"] = author.nickname mention["nickname"] = author.nickname
mention["avatar"] = author.avatar
mention["username"] = author.username
mention["discriminator"] = author.discriminator
msgs.append(message) msgs.append(message)
return msgs return msgs