Further improve the session-switching mechanism for inline bots (pyrogram#739)

HasibulKabir · delivrance · web-flow · commit e68da74e89f1 · 2021-08-29T10:49:47.000+02:00
* Ability to run multiple bot.

Global session creates a problem for other bot. As that session was generated for another bot can't be used by this bot.

* Use the existing media session lock

Co-authored-by: Dan &lt;14043624+delivrance@users.noreply.github.com&gt;
diff --git a/pyrogram/methods/messages/inline_session.py b/pyrogram/methods/messages/inline_session.py
@@ -16,29 +16,22 @@
 #  You should have received a copy of the GNU Lesser General Public License
 #  along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
-from asyncio import Lock
-
 import pyrogram
 from pyrogram import raw
 from pyrogram.errors import AuthBytesInvalid
 from pyrogram.session import Session
 from pyrogram.session.auth import Auth
 
-lock = Lock()
-sessions = {}
-
 
 async def get_session(client: "pyrogram.Client", dc_id: int):
     if dc_id == await client.storage.dc_id():
         return client
 
-    async with lock:
-        global sessions
-
-        if sessions.get(dc_id):
-            return sessions[dc_id]
+    async with client.media_sessions_lock:
+        if client.media_sessions.get(dc_id):
+            return client.media_sessions[dc_id]
 
-        session = sessions[dc_id] = Session(
+        session = client.media_sessions[dc_id] = Session(
             client, dc_id,
             await Auth(client, dc_id, False).create(),
             False, is_media=True
