TeamPGM/pyrogram
2
0
Fork 0
mirror of https://github.com/TeamPGM/pyrogram.git synced 2024-11-16 04:35:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add suport for slug to join/leave folder

Browse Source
This commit is contained in:
KurimuzonAkuma 2024-04-05 12:11:31 +03:00
parent e31edb62b8
commit 9a993b918e
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
pyrogram/methods/chats/join_folder.py
View File

@ -49,12 +49,16 @@ class JoinFolder:
"""
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 @@ class JoinFolder:
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
],

8
pyrogram/methods/chats/leave_folder.py
View File

@ -55,12 +55,16 @@ class LeaveFolder:
"""
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
)
)