Pyrofork: Use session name as database name, add some parameters informations and some cleanup (#3)

Changes to be committed:
	modified:   pyrogram/client.py
	modified:   pyrogram/storage/mongo_storage.py

Signed-off-by: wulan17 <wulan17@nusantararom.org>

docs/source/topics/storage-engines.rst

@@ -73,11 +73,9 @@ mongodb config as ``dict`` to the ``mongodb`` parameter of the :obj:`~pyrogram.C
 
     # uri (``str``):
     #   mongodb database uri
-    # db_name (``str``, *optional*):
-    #   custom database name, default = pyrofork-session
     # remove_peers (``bool``, *optional*):
     #   remove peers collection on logout, default = False
-    async with Client("my_account", mongodb=dict(uri="mongodb://...", db_name="pyrofork-session", remove_peers=False)) as app:
+    async with Client("my_account", mongodb=dict(uri="mongodb://...", remove_peers=False)) as app:
         print(await app.get_me())
 
 This storage engine is backed by MongoDB, a session will be created and saved to mongodb database. Any subsequent client

pyrogram/client.py

@@ -121,7 +121,7 @@ class Client(Methods):
             Defaults to False.
 
         mongodb (``dict``, *optional*):
-            Mongodb config as dict, e.g.: *dict(uri="mongodb://...", db_name="pyrofork-session", remove_peers=False)*.
+            Mongodb config as dict, e.g.: *dict(uri="mongodb://...", remove_peers=False)*.
             Only applicable for new sessions.
 
         phone_number (``str``, *optional*):
@@ -255,7 +255,7 @@ class Client(Methods):
         elif self.in_memory:
             self.storage = MemoryStorage(self.name)
         elif self.mongodb:
-            self.storage = MongoStorage(self.mongodb)
+            self.storage = MongoStorage(self.name, **self.mongodb)
         else:
             self.storage = FileStorage(self.name, self.workdir)
 

pyrogram/storage/mongo_storage.py

@@ -11,23 +11,30 @@ from pyrogram.storage.sqlite_storage import get_input_peer
 
 class MongoStorage(Storage):
     """
-    config (``dict``)
-        Mongodb config as dict, e.g.: *dict(uri="mongodb://...", db_name="pyrofork-session", remove_peers=False)*.
-        Only applicable for new sessions.
+    Initializes a new session.
+
+    Parameters:
+        - name (`str`):
+            The session name used for database name.
+
+        - uri (`str`):
+            MongoDB Connection String URI.
+            For more information refer to https://www.mongodb.com/docs/manual/reference/connection-string
+
+        - remove_peers (`bool`, *optional*):
+            Flag to remove data in the peers collection. If set to True, 
+            the data related to peers will be removed everytime client log out. 
+            If set to False or None, the data will not be removed.
+
+    Example:
+        session = MongoStorage("my_session", uri="mongodb://...", remove_peers=True)
     """
     lock: asyncio.Lock
     USERNAME_TTL = 8 * 60 * 60
 
-    def __init__(self, config: dict):
-        super().__init__('')
-        db_name = "pyrofork-session"
-        db_uri = config["uri"]
-        remove_peers = False
-        if "db_name" in config:
-            db_name = config["db_name"]
-        if "remove_peers" in config:
-            remove_peers = config["remove_peers"]
-        database = AsyncIOMotorClient(db_uri)[db_name]
+    def __init__(self, name: str, uri: str, remove_peers: bool = False):
+        super().__init__(name=name)
+        database = AsyncIOMotorClient(uri)[name]
         self.lock = asyncio.Lock()
         self.database = database
         self._peer = database['peers']