TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-09-28 11:47:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow Custom Storage Engines

Browse Source
This commit is contained in:
KurimuzonAkuma 2023-11-24 01:31:51 +03:00
parent 24b46743a2
commit eeb1a6cf42
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pyrogram/client.py
View File

@ -49,7 +49,7 @@ from pyrogram.errors import (
from pyrogram.handlers.handler import Handler
from pyrogram.methods import Methods
from pyrogram.session import Auth, Session
from pyrogram.storage import FileStorage, MemoryStorage
from pyrogram.storage import Storage, FileStorage, MemoryStorage
from pyrogram.types import User, TermsOfService
from pyrogram.utils import ainput
from .dispatcher import Dispatcher
@ -177,6 +177,10 @@ class Client(Methods):
Set the maximum amount of concurrent transmissions (uploads & downloads).
A value that is too high may result in network related issues.
Defaults to 1.
storage_engine (:obj:`~pyrogram.storage.Storage`, *optional*):
Pass an instance of your own implementation of session storage engine.
Useful when you want to store your session in databases like Mongo, Redis, etc.
"""
APP_VERSION = f"Pyrogram {__version__}"
@ -225,7 +229,8 @@ class Client(Methods):
takeout: bool = None,
sleep_threshold: int = Session.SLEEP_THRESHOLD,
hide_password: bool = False,
max_concurrent_transmissions: int = MAX_CONCURRENT_TRANSMISSIONS
max_concurrent_transmissions: int = MAX_CONCURRENT_TRANSMISSIONS,
storage_engine: Storage = None
):
super().__init__()
@ -261,6 +266,8 @@ class Client(Methods):
self.storage = MemoryStorage(self.name, self.session_string)
elif self.in_memory:
self.storage = MemoryStorage(self.name)
elif isinstance(storage_engine, Storage):
self.storage = storage_engine
else:
self.storage = FileStorage(self.name, self.workdir)