diff --git a/pyrogram/storage/mongo_storage.py b/pyrogram/storage/mongo_storage.py index a9ab4d1c..10841a27 100644 --- a/pyrogram/storage/mongo_storage.py +++ b/pyrogram/storage/mongo_storage.py @@ -3,7 +3,7 @@ import inspect import time from typing import List, Tuple, Any -from async_pymongo import AsyncClient +from motor.motor_asyncio import AsyncIOMotorClient from pymongo import UpdateOne from pyrogram.storage.storage import Storage from pyrogram.storage.sqlite_storage import get_input_peer @@ -34,7 +34,7 @@ class MongoStorage(Storage): def __init__(self, name: str, uri: str, remove_peers: bool = False): super().__init__(name=name) - database = AsyncClient(uri)[name] + database = AsyncIOMotorClient(uri)[name] self.lock = asyncio.Lock() self.database = database self._peer = database['peers'] diff --git a/requirements.txt b/requirements.txt index fa90d917..d0b2161a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,5 +1,5 @@ aiosqlite>=0.17.0,<0.19.0 -async_pymongo==0.1.2 +motor==3.1.2 pyaes==1.6.1 pymediainfo==6.0.1 pymongo==4.3.3