From 0280fc2572074793f49fcb37da296d3ba067ce0b Mon Sep 17 00:00:00 2001 From: wulan17 Date: Fri, 19 May 2023 20:57:41 +0700 Subject: [PATCH] PyroFork: storage: mongo_storage: Move into async_pymongo 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 10841a27..a9ab4d1c 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 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 index d0b2161a..fa90d917 100644 --- 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.2 pyaes==1.6.1 pymediainfo==6.0.1 pymongo==4.3.3