PyroFork: storage: mongo_storage: Move into async_pymongo

Signed-off-by: wulan17 <wulan17@nusantararom.org>
Mayuri-Chan · May 19, 2023 · 155a4a8 · 155a4a8
1 parent 2bf362c
commit 155a4a8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyrogram/storage/mongo_storage.py b/pyrogram/storage/mongo_storage.py
@@ -3,7 +3,7 @@
 import time
 from typing import List, Tuple, Any
 
-from motor.motor_asyncio import AsyncIOMotorClient
+from async_pymongo import AsyncClient
 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 = AsyncIOMotorClient(uri)[name]
+        database = AsyncClient(uri)[name]
         self.lock = asyncio.Lock()
         self.database = database
         self._peer = database['peers']

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
 aiosqlite>=0.17.0,<0.19.0
-motor==3.1.2
+async_pymongo==0.1.1
 pyaes==1.6.1
 pymediainfo==6.0.1
 pymongo==4.3.3