Add suport for slug to join/leave folder

KurimuzonAkuma · KurimuzonAkuma · commit 9a993b918eec · 2024-04-05T12:11:31.000+03:00
diff --git a/pyrogram/methods/chats/join_folder.py b/pyrogram/methods/chats/join_folder.py
@@ -49,12 +49,16 @@ async def join_folder(
         """
         match = re.match(r"^(?:https?://)?(?:www\.)?(?:t(?:elegram)?\.(?:org|me|dog)/(?:addlist/|\+))([\w-]+)$", link)
 
-        if not match:
+        if match:
+            slug = match.group(1)
+        elif isinstance(link, str):
+            slug = link
+        else:
             raise ValueError("Invalid folder invite link")
 
         r = await self.invoke(
             raw.functions.chatlists.CheckChatlistInvite(
-                slug=match.group(1)
+                slug=slug
             )
         )
 
@@ -65,7 +69,7 @@ async def join_folder(
 
         await self.invoke(
             raw.functions.chatlists.JoinChatlistInvite(
-                slug=match.group(1),
+                slug=slug,
                 peers=[
                     await self.resolve_peer(utils.get_peer_id(id)) for id in peers
                 ],
diff --git a/pyrogram/methods/chats/leave_folder.py b/pyrogram/methods/chats/leave_folder.py
@@ -55,12 +55,16 @@ async def leave_folder(
         """
         match = re.match(r"^(?:https?://)?(?:www\.)?(?:t(?:elegram)?\.(?:org|me|dog)/(?:addlist/|\+))([\w-]+)$", link)
 
-        if not match:
+        if match:
+            slug = match.group(1)
+        elif isinstance(link, str):
+            slug = link
+        else:
             raise ValueError("Invalid folder invite link")
 
         r = await self.invoke(
             raw.functions.chatlists.CheckChatlistInvite(
-                slug=match.group(1)
+                slug=slug
             )
         )
 
