update

aggregator/client.py

@@ -1,10 +1,12 @@
 import os
+from typing import List
 
 from django.core.files.storage import default_storage
 from djconfig import config
-from pyrogram import Client, MessageHandler, DeletedMessagesHandler, \
-                     Message as PyrogramMessage, Messages as PyrogramMessages
-from pyrogram.api.errors import ChannelPrivate
+from pyrogram import Client
+from pyrogram.handlers import MessageHandler, DeletedMessagesHandler
+from pyrogram.types import Message as PyrogramMessage
+from pyrogram.errors import ChannelPrivate
 from pyrogram.session import Session
 
 from aggregator.models import AggregationSource, Message, Chat
@@ -30,8 +32,8 @@ def save_message(client, message: PyrogramMessage):
     Message.from_obj(message, client)
 
 
-def delete_messages(client, messages: PyrogramMessages):
-    for message in messages.messages:
+def delete_messages(client, messages: List[PyrogramMessage]):
+    for message in messages:
         Message.objects.filter(chat__chat_id=message.chat.id, message_id=message.message_id).update(deleted=True)