Browse Source

Add Client.get_invite to turn a URL to an Invite object.

pull/42/merge
Rapptz 9 years ago
parent
commit
ea80812fdd
  1. 23
      discord/client.py
  2. 10
      discord/invite.py

23
discord/client.py

@ -1180,6 +1180,29 @@ class Client(object):
data['channel'] = utils.find(lambda ch: ch.id == channel_id, data['server'].channels)
return Invite(**data)
def get_invite(self, url):
"""Returns a :class:`Invite` object from the discord.gg invite URL or ID.
.. note::
If the server attribute of the returned invite is ``None`` then that means
that you have not joined the server.
"""
destination = self._resolve_invite(url)
rurl = '{0}/invite/{1}'.format(endpoints.API_BASE, destination)
response = requests.get(rurl, headers=self.headers)
log.debug(request_logging_format.format(response=response))
_verify_successful_response(response)
data = response.json()
server = self.connection._get_server(data['guild']['id'])
data['server'] = server
ch_id = data['channel']['id']
channels = getattr(server, 'channels', [])
data['channel'] = utils.find(lambda c: c.id == ch_id, channels)
return Invite(**data)
def accept_invite(self, invite):
"""Accepts an :class:`Invite`, URL or ID to an invite.

10
discord/invite.py

@ -30,6 +30,9 @@ from .utils import parse_time
class Invite(object):
"""Represents a Discord :class:`Server` or :class:`Channel` invite.
Depending on the way this object was created, some of the attributes can
have a value of ``None``.
Instance attributes:
.. attribute:: max_age
@ -49,7 +52,8 @@ class Invite(object):
A datetime object denoting the time the invite was created.
.. attribute:: temporary
A boolean indicating that the invite grants temporary membership. If True, members who joined via this invite will be kicked upon disconnect.
A boolean indicating that the invite grants temporary membership.
If True, members who joined via this invite will be kicked upon disconnect.
.. attribute:: uses
How many times the invite has been used.
@ -77,7 +81,9 @@ class Invite(object):
self.uses = kwargs.get('uses')
self.max_uses = kwargs.get('max_uses')
self.xkcd = kwargs.get('xkcdpass')
self.inviter = User(**kwargs.get('inviter', {}))
inviter_data = kwargs.get('inviter')
self.inviter = None if inviter_data is None else User(**inviter_data)
self.channel = kwargs.get('channel')
@property

Loading…
Cancel
Save