bakatrouble/telegram_bots
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
bakatrouble 2019-01-31 00:26:06 +03:00
parent 0334a04260
commit 2f1076b75b
2 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
aggregator/models.py
View File

@ -48,6 +48,8 @@ class Chat(models.Model):
obj.photo.save(os.path.basename(path), f, save=True) obj.photo.save(os.path.basename(path), f, save=True)
obj.photo_id = chat.photo.small_file_id obj.photo_id = chat.photo.small_file_id
obj.save() obj.save()
if created:
transaction.on_commit(lambda: collect_new_messages.delay(obj.pk))
return obj return obj
def __str__(self): def __str__(self):

6
aggregator/signals.py
View File

@ -2,7 +2,6 @@ from pyrogram import Error
from .client import get_client from .client import get_client
from .models import Chat, AggregationSource from .models import Chat, AggregationSource
from .tasks import collect_new_messages
def aggregation_source_deleted(sender, instance: AggregationSource, **kwargs): def aggregation_source_deleted(sender, instance: AggregationSource, **kwargs):
@ -10,11 +9,6 @@ def aggregation_source_deleted(sender, instance: AggregationSource, **kwargs):
Chat.objects.filter(chat_id=instance.chat_id).delete() Chat.objects.filter(chat_id=instance.chat_id).delete()
def chat_created(sender, instance: Chat, created, **kwargs):
if created:
collect_new_messages.delay(instance.pk)
def chat_deleted(sender, instance: Chat, **kwargs): def chat_deleted(sender, instance: Chat, **kwargs):
with get_client() as client: with get_client() as client:
try: try: