yasirarism/pyrofork
1
0
Fork 0
mirror of https://github.com/Mayuri-Chan/pyrofork.git synced 2026-01-03 14:04:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

PyroFork: storage: mongo_storage: Move into async_pymongo

Browse source 
Signed-off-by: wulan17 <wulan17@nusantararom.org>
This commit is contained in:
wulan17 2023-05-19 20:57:41 +07:00
parent 89feb4a4d8
commit 0280fc2572
No known key found for this signature in database
GPG key ID: 318CD6CD3A6AC0A5
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
pyrogram/storage/mongo_storage.py
View file

@ -3,7 +3,7 @@ import inspect
import time import time
from typing import List, Tuple, Any from typing import List, Tuple, Any
from motor.motor_asyncio import AsyncIOMotorClient from async_pymongo import AsyncClient
from pymongo import UpdateOne from pymongo import UpdateOne
from pyrogram.storage.storage import Storage from pyrogram.storage.storage import Storage
from pyrogram.storage.sqlite_storage import get_input_peer 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): def __init__(self, name: str, uri: str, remove_peers: bool = False):
super().__init__(name=name) super().__init__(name=name)
database = AsyncIOMotorClient(uri)[name] database = AsyncClient(uri)[name]
self.lock = asyncio.Lock() self.lock = asyncio.Lock()
self.database = database self.database = database
self._peer = database['peers'] self._peer = database['peers']

2
requirements.txt
View file

@ -1,5 +1,5 @@
aiosqlite>=0.17.0,<0.19.0 aiosqlite>=0.17.0,<0.19.0
motor==3.1.2 async_pymongo==0.1.2
pyaes==1.6.1 pyaes==1.6.1
pymediainfo==6.0.1 pymediainfo==6.0.1
pymongo==4.3.3 pymongo==4.3.3