Yadciel/Titan
1
0
Fork 0
mirror of https://github.com/TitanEmbeds/Titan.git synced 2024-11-14 18:11:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Better cleanup db msgs?

Browse Source
This commit is contained in:
Jeremy Zhang 2017-09-22 00:53:59 +00:00
parent 956e466db1
commit c0798bd5b1
2 changed files with 14 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
discordbot/cleanup_db_messages.py
View File

@ -80,9 +80,8 @@ class Titan(discord.Client):
continue
for channel in channelsjson:
chanid = channel["id"]
msgs = session.query(Messages).filter(Messages.channel_id == chanid).order_by(Messages.timestamp.desc()).offset(50).all()
for msg in msgs:
session.delete(msg)
keep_these = session.query(Messages.id).filter(Messages.channel_id == chanid).order_by(Messages.timestamp.desc()).limit(50)
session.query(Messages).filter(Messages.channel_id == chanid, ~Messages.id.in_(keep_these)).delete(synchronize_session=False)
session.commit()
print("done!")
await self.logout()

24
webapp/titanembeds/blueprints/api/api.py
View File

@ -436,18 +436,16 @@ def cleanup_keyval_db():
db.session.query(KeyValueProperties).filter(KeyValueProperties.expiration < datetime.datetime.now()).delete()
db.session.commit()
# guilds = Guilds.query.all()
# for guild in guilds:
# try:
# channelsjson = json.loads(guild.channels)
# except:
# continue
# for channel in channelsjson:
# chanid = channel["id"]
# msgs = db.session.query(Messages).filter(Messages.channel_id == chanid).order_by(Messages.timestamp.desc()).offset(50).all()
# for msg in msgs:
# db.session.delete(msg)
# db.session.commit()
guilds = Guilds.query.all()
for guild in guilds:
try:
channelsjson = json.loads(guild.channels)
except:
continue
for channel in channelsjson:
chanid = channel["id"]
keep_these = db.session.query(Messages.id).filter(Messages.channel_id == chanid).order_by(Messages.timestamp.desc()).limit(50)
db.session.query(Messages).filter(Messages.channel_id == chanid, ~Messages.id.in_(keep_these)).delete(synchronize_session=False)
db.session.commit()
return ('', 204)
abort(401)