Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2025-06-17 03:45:25 +02:00
Code Issues Packages Projects Releases Wiki Activity

Populate msgs on server join and remove on leave

Browse Source
This commit is contained in:
Jeremy Zhang
2017-05-08 20:19:45 -07:00
parent 4bb559420f
commit a621fa8b0a
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
discordbot/titanembeds/database/__init__.py
View File

@ -188,6 +188,9 @@ class DatabaseInterface(object):
disguild = discord.utils.get(guilds, id=guild.guild_id)
if not disguild:
changed = True
dbmsgs = session.query(Messages).filter(Messages.guild_id == guild.guild_id).all()
for msg in dbmsgs:
session.delete(msg)
session.delete(guild)
if changed:
session.commit()
@ -197,6 +200,9 @@ class DatabaseInterface(object):
with self.get_session() as session:
gui = session.query(Guilds).filter(Guilds.guild_id == guild.id).first()
if gui:
dbmsgs = session.query(Messages).filter(Messages.guild_id == guild.id).all()
for msg in dbmsgs:
session.delete(msg)
session.delete(gui)
session.commit()