mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-05-12 16:59:50 +00:00
Code Issues Projects Releases Wiki Activity

accept_invite now works on some invite URLs.

Browse Source
This commit is contained in:
Rapptz 2015-09-19 18:41:14 -04:00
parent 6c54b24a6f
commit 9559f02f95
1 changed files with 17 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
discord/client.py
View File

@ -762,13 +762,27 @@ class Client(object):
log.debug(request_logging_format.format(response=response, name='create_invite')) log.debug(request_logging_format.format(response=response, name='create_invite'))
def accept_invite(self, invite): def accept_invite(self, invite):
"""Accepts an :class:`Invite`. """Accepts an :class:`Invite` or a URL to an invite.
:param invite: The :class:`Invite` to accept. The URL must be a discord.gg URL. e.g. "http://discord.gg/codehere"
:param invite: The :class:`Invite` or URL to an invite to accept.
:returns: True if the invite was successfully accepted, False otherwise. :returns: True if the invite was successfully accepted, False otherwise.
""" """
url = '{0}/invite/{1.id}'.format(endpoints.API_BASE, invite) destination = None
if isinstance(invite, Invite):
destination = invite.id
else:
rx = r'(?:https?\:\/\/)?discord\.gg\/(.+)'
m = re.match(rx, invite)
if m:
destination = m.group(1)
if destination is None:
return False
url = '{0}/invite/{1}'.format(endpoints.API_BASE, destination)
response = requests.post(url, headers=self.headers) response = requests.post(url, headers=self.headers)
log.debug(request_logging_format.format(response=response, name='accept_invite')) log.debug(request_logging_format.format(response=response, name='accept_invite'))
return response.status_code in (200, 201) return response.status_code in (200, 201)