telegram_bots/aggregator/tasks.py

18 lines
467 B
Python
Raw Normal View History

2019-01-27 21:30:40 +00:00
from celery_once import QueueOnce
from aggregator.client import get_client
from config.celery import app
from .client import collect_new_messages as _collect_new_messages
@app.task(base=QueueOnce, once={'keys': ['chat_id'], 'graceful': True})
def collect_new_messages(chat_id):
2019-01-27 21:32:34 +00:00
from .models import Chat
2019-01-27 21:30:40 +00:00
chat = Chat.objects.get(pk=chat_id)
2019-01-27 22:07:57 +00:00
client = get_client(takeout=True)
client.start()
_collect_new_messages(client, chat)
client.stop()