Improvements to purge queue

This commit is contained in:
REDNBLACK 2016-11-13 16:13:43 +03:00
parent c6cab34f21
commit eba9723780
1 changed files with 3 additions and 0 deletions

View File

@ -3,6 +3,7 @@ import logging
from datetime import datetime, timedelta
from telegram.ext import Job
from src.entity.chat import Chat
from src.entity.reply import Reply
from src.entity.job import Job as JobEntity
from src.config import config
@ -62,6 +63,8 @@ class ChatPurgeQueue:
chat = Chat.find(chat_id)
if chat is not None:
for pairs in chat.pairs().select('id').chunk(500):
Reply.where_in('pair_id', pairs.pluck('id').all()).delete()
chat.pairs().delete()
chat.delete()