a1batross
/
imaginaryfriend
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/service/chat_purge_queue.py
View File

@ -3,6 +3,7 @@ import logging
from datetime import datetime, timedelta from datetime import datetime, timedelta
from telegram.ext import Job from telegram.ext import Job
from src.entity.chat import Chat from src.entity.chat import Chat
from src.entity.reply import Reply
from src.entity.job import Job as JobEntity from src.entity.job import Job as JobEntity
from src.config import config from src.config import config
@ -62,6 +63,8 @@ class ChatPurgeQueue:
chat = Chat.find(chat_id) chat = Chat.find(chat_id)
if chat is not None: 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.pairs().delete()
chat.delete() chat.delete()