Attempt to fix cleanup 2

code looks much cleaner too
This commit is contained in:
Jeremy "EndenDragon" Zhang 2017-06-24 12:50:55 -07:00 committed by GitHub
parent efcd2d9a83
commit 44af174287

View File

@ -467,22 +467,19 @@ def create_authenticated_user():
@api.route("/cleanup-db", methods=["DELETE"]) @api.route("/cleanup-db", methods=["DELETE"])
def cleanup_keyval_db(): def cleanup_keyval_db():
if request.form.get("secret", None) == config["app-secret"]: if request.form.get("secret", None) == config["app-secret"]:
q = KeyValueProperties.query.filter(KeyValueProperties.expiration < datetime.datetime.now()).all() db.session.query(KeyValueProperties).filter(KeyValueProperties.expiration < datetime.datetime.now()).delete()
for m in q: db.session.commit()
db.session.delete(m)
db.session.commit()
guilds = Guilds.query.all() guilds = Guilds.query.all()
for guild in guilds: for guild in guilds:
channelsjson = json.loads(guild.channels) try:
channelsjson = json.loads(guild.channels)
except:
continue
for channel in channelsjson: for channel in channelsjson:
chanid = channel["id"] chanid = channel["id"]
dbmsg = Messages.query.filter(Messages.channel_id == chanid).all() db.session.query(Messages).filter(Messages.channel_id == chanid).order_by(Messages.timestamp.desc()).offset(50).delete()
for idx, val in enumerate(dbmsg): db.session.commit()
if len(dbmsg) - idx > 50:
db.session.delete(val)
db.session.commit()
else:
continue
return ('', 204) return ('', 204)
abort(401) abort(401)