From c76e9263ddb3462dd008c855c8648edbda18d50b Mon Sep 17 00:00:00 2001 From: wulan17 Date: Tue, 23 May 2023 20:48:58 +0700 Subject: [PATCH] Revert "PyroFork: storage: mongo_storage: Move into async_pymongo" This reverts commit 0280fc2572074793f49fcb37da296d3ba067ce0b. Signed-off-by: wulan17 --- pyrogram/storage/mongo_storage.py | 4 ++-- requirements.txt | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) 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