aboutsummaryrefslogtreecommitdiff
path: root/discord/client.py
diff options
context:
space:
mode:
authorRapptz <[email protected]>2016-06-12 20:32:59 -0400
committerRapptz <[email protected]>2016-06-12 20:33:09 -0400
commit1fba1b06faca31d07c9296b2badabfe22f173001 (patch)
tree858fc299aa7299f1013da117599992f6049fc42d /discord/client.py
parentChange HTTPException to only take a single parameter. (diff)
downloaddiscord.py-1fba1b06faca31d07c9296b2badabfe22f173001.tar.xz
discord.py-1fba1b06faca31d07c9296b2badabfe22f173001.zip
Rewrite HTTP handling significantly.
This should have a more uniform approach to rate limit handling. Instead of queueing every request, wait until we receive a 429 and then block the requesting bucket until we're done being rate limited. This should reduce the number of 429s done by the API significantly (about 66% avg). This also consistently checks for 502 retries across all requests.
Diffstat (limited to 'discord/client.py')
-rw-r--r--discord/client.py545
1 files changed, 98 insertions, 447 deletions
diff --git a/discord/client.py b/discord/client.py
index 7f3118f0..8221afe1 100644
--- a/discord/client.py
+++ b/discord/client.py
@@ -42,6 +42,7 @@ from .enums import ChannelType, ServerRegion
from .voice_client import VoiceClient
from .iterators import LogsFromIterator
from .gateway import *
+from .http import HTTPClient
import asyncio
import aiohttp
@@ -52,7 +53,6 @@ import sys, re
import tempfile, os, hashlib
import itertools
import datetime
-from random import randint as random_integer
from collections import namedtuple
PY35 = sys.version_info >= (3, 5)
@@ -136,16 +136,8 @@ class Client:
self.connection = ConnectionState(self.dispatch, self.request_offline_members, max_messages, loop=self.loop)
- # Blame Jake for this
- user_agent = 'DiscordBot (https://github.com/Rapptz/discord.py {0}) Python/{1[0]}.{1[1]} aiohttp/{2}'
-
- self.headers = {
- 'content-type': 'application/json',
- 'user-agent': user_agent.format(library_version, sys.version_info, aiohttp.__version__)
- }
-
connector = options.pop('connector', None)
- self.session = aiohttp.ClientSession(loop=self.loop, connector=connector)
+ self.http = HTTPClient(connector, loop=self.loop)
self._closed = asyncio.Event(loop=self.loop)
self._is_logged_in = asyncio.Event(loop=self.loop)
@@ -157,23 +149,21 @@ class Client:
filename = hashlib.md5(email.encode('utf-8')).hexdigest()
return os.path.join(tempfile.gettempdir(), 'discord_py', filename)
- @asyncio.coroutine
- def _login_via_cache(self, email, password):
+ def _get_cache_token(self, email, password):
try:
log.info('attempting to login via cache')
cache_file = self._get_cache_filename(email)
self.email = email
with open(cache_file, 'r') as f:
log.info('login cache file found')
- self.token = f.read()
- self.headers['authorization'] = self.token
+ return f.read()
# at this point our check failed
# so we have to login and get the proper token and then
# redo the cache
except OSError:
log.info('a problem occurred while opening login cache')
- pass # file not found et al
+ return None # file not found et al
def _update_cache(self, email, password):
try:
@@ -222,20 +212,30 @@ class Client:
@asyncio.coroutine
def _resolve_destination(self, destination):
- if isinstance(destination, (Channel, PrivateChannel, Server)):
- return destination.id
+ if isinstance(destination, Channel):
+ return destination.id, destination.server.id
+ elif isinstance(destination, PrivateChannel):
+ return destination.id, None
+ elif isinstance(destination, Server):
+ return destination.id, destination.id
elif isinstance(destination, User):
found = self.connection._get_private_channel_by_user(destination.id)
if found is None:
# Couldn't find the user, so start a PM with them first.
channel = yield from self.start_private_message(destination)
- return channel.id
+ return channel.id, None
else:
- return found.id
+ return found.id, None
elif isinstance(destination, Object):
- return destination.id
+ found = self.get_channel(destination.id)
+ if found is not None:
+ return (yield from self._resolve_destination(found))
+
+ # couldn't find it in cache so YOLO
+ return destination.id, destination.id
else:
- raise InvalidArgument('Destination must be Channel, PrivateChannel, User, or Object')
+ fmt = 'Destination must be Channel, PrivateChannel, User, or Object. Received {0.__class__.__name__}'
+ raise InvalidArgument(fmt.format(destination))
def __getattr__(self, name):
if name in ('user', 'servers', 'private_channels', 'messages', 'voice_clients'):
@@ -291,55 +291,25 @@ class Client:
@asyncio.coroutine
def _login_1(self, token, **kwargs):
log.info('logging in using static token')
- self.token = token
- self.email = None
- if kwargs.pop('bot', True):
- self.headers['authorization'] = 'Bot ' + self.token
- else:
- self.headers['authorization'] = self.token
-
- resp = yield from self.session.get(endpoints.ME, headers=self.headers)
- yield from resp.release()
- log.debug(request_logging_format.format(method='GET', response=resp))
-
- if resp.status != 200:
- if resp.status == 401:
- raise LoginFailure('Improper token has been passed.')
- else:
- raise HTTPException(resp, None)
-
- log.info('token auth returned status code {}'.format(resp.status))
+ yield from self.http.static_login(token, bot=kwargs.pop('bot', True))
self._is_logged_in.set()
@asyncio.coroutine
def _login_2(self, email, password, **kwargs):
# attempt to read the token from cache
if self.cache_auth:
- yield from self._login_via_cache(email, password)
- if self.is_logged_in:
+ token = self._get_cache_token()
+ try:
+ self.http.static_login(token, bot=False)
+ except:
+ log.info('cache auth token is out of date')
+ else:
+ self._is_logged_in.set()
return
- payload = {
- 'email': email,
- 'password': password
- }
-
- data = utils.to_json(payload)
- resp = yield from self.session.post(endpoints.LOGIN, data=data, headers=self.headers)
- log.debug(request_logging_format.format(method='POST', response=resp))
- if resp.status != 200:
- yield from resp.release()
- if resp.status == 400:
- raise LoginFailure('Improper credentials have been passed.')
- else:
- raise HTTPException(resp, None)
- log.info('logging in returned status code {}'.format(resp.status))
+ yield from self.http.email_login(email, password)
self.email = email
-
- body = yield from resp.json(encoding='utf-8')
- self.token = body['token']
- self.headers['authorization'] = self.token
self._is_logged_in.set()
# since we went through all this trouble
@@ -395,12 +365,10 @@ class Client:
def logout(self):
"""|coro|
- Logs out of Discord and closes all connections."""
- response = yield from self.session.post(endpoints.LOGOUT, headers=self.headers)
- yield from response.release()
+ Logs out of Discord and closes all connections.
+ """
yield from self.close()
self._is_logged_in.clear()
- log.debug(request_logging_format.format(method='POST', response=response))
@asyncio.coroutine
def connect(self):
@@ -453,7 +421,7 @@ class Client:
yield from self.ws.close()
- yield from self.session.close()
+ yield from self.http.close()
self._closed.set()
self._is_ready.clear()
@@ -774,44 +742,12 @@ class Client:
if not isinstance(user, User):
raise InvalidArgument('user argument must be a User')
- payload = {
- 'recipient_id': user.id
- }
-
- url = '{}/channels'.format(endpoints.ME)
- r = yield from self.session.post(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='POST', response=r))
- yield from utils._verify_successful_response(r)
- data = yield from r.json(encoding='utf-8')
- log.debug(request_success_log.format(response=r, json=payload, data=data))
+ data = yield from self.http.start_private_message(user.id)
channel = PrivateChannel(id=data['id'], user=user)
self.connection._add_private_channel(channel)
return channel
@asyncio.coroutine
- def _retry_helper(self, name, *args, retries=0, **kwargs):
- req_kwargs = {'headers': self.headers}
- req_kwargs.update(kwargs)
- resp = yield from self.session.request(*args, **req_kwargs)
- tmp = request_logging_format.format(method=resp.method, response=resp)
- log_fmt = 'In {}, {}'.format(name, tmp)
- log.debug(log_fmt)
-
- if resp.status == 502 and retries < 5:
- # retry the 502 request unconditionally
- log.info('Retrying the 502 request to ' + name)
- yield from asyncio.sleep(retries + 1)
- return (yield from self._retry_helper(name, *args, retries=retries + 1, **kwargs))
-
- if resp.status == 429:
- retry = float(resp.headers['Retry-After']) / 1000.0
- yield from resp.release()
- yield from asyncio.sleep(retry)
- return (yield from self._retry_helper(name, *args, retries=retries, **kwargs))
-
- return resp
-
- @asyncio.coroutine
def send_message(self, destination, content, *, tts=False):
"""|coro|
@@ -858,23 +794,11 @@ class Client:
The message that was sent.
"""
- channel_id = yield from self._resolve_destination(destination)
+ channel_id, guild_id = yield from self._resolve_destination(destination)
content = str(content)
- url = '{base}/{id}/messages'.format(base=endpoints.CHANNELS, id=channel_id)
- payload = {
- 'content': content,
- 'nonce': random_integer(-2**63, 2**63 - 1)
- }
-
- if tts:
- payload['tts'] = True
-
- resp = yield from self._retry_helper('send_message', 'POST', url, data=utils.to_json(payload))
- yield from utils._verify_successful_response(resp)
- data = yield from resp.json(encoding='utf-8')
- log.debug(request_success_log.format(response=resp, json=payload, data=data))
+ data = yield from self.http.send_message(channel_id, content, guild_id=guild_id, tts=tts)
channel = self.get_channel(data.get('channel_id'))
message = Message(channel=channel, **data)
return message
@@ -895,14 +819,8 @@ class Client:
The location to send the typing update.
"""
- channel_id = yield from self._resolve_destination(destination)
-
- url = '{base}/{id}/typing'.format(base=endpoints.CHANNELS, id=channel_id)
-
- response = yield from self.session.post(url, headers=self.headers)
- log.debug(request_logging_format.format(method='POST', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ channel_id, guild_id = yield from self._resolve_destination(destination)
+ yield from self.http.send_typing(channel_id)
@asyncio.coroutine
def send_file(self, destination, fp, *, filename=None, content=None, tts=False):
@@ -951,34 +869,18 @@ class Client:
The message sent.
"""
- channel_id = yield from self._resolve_destination(destination)
-
- url = '{base}/{id}/messages'.format(base=endpoints.CHANNELS, id=channel_id)
- form = aiohttp.FormData()
-
- if content is not None:
- form.add_field('content', str(content))
-
- form.add_field('tts', 'true' if tts else 'false')
-
- # we don't want the content-type json in this request
- headers = self.headers.copy()
- headers.pop('content-type', None)
+ channel_id, guild_id = yield from self._resolve_destination(destination)
try:
- # attempt to open the file and send the request
with open(fp, 'rb') as f:
- form.add_field('file', f, filename=filename, content_type='application/octet-stream')
- response = yield from self._retry_helper("send_file", "POST", url, data=form, headers=headers)
+ buffer = f.read()
+ if filename is None:
+ filename = fp
except TypeError:
- form.add_field('file', fp, filename=filename, content_type='application/octet-stream')
- response = yield from self._retry_helper("send_file", "POST", url, data=form, headers=headers)
-
- log.debug(request_logging_format.format(method='POST', response=response))
- yield from utils._verify_successful_response(response)
- data = yield from response.json(encoding='utf-8')
- msg = 'POST {0.url} returned {0.status} with {1} response'
- log.debug(msg.format(response, data))
+ buffer = fp
+
+ data = yield from self.http.send_file(channel_id, buffer, guild_id=guild_id,
+ filename=filename, content=content, tts=tts)
channel = self.get_channel(data.get('channel_id'))
message = Message(channel=channel, **data)
return message
@@ -1004,12 +906,8 @@ class Client:
HTTPException
Deleting the message failed.
"""
-
- url = '{}/{}/messages/{}'.format(endpoints.CHANNELS, message.channel.id, message.id)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ channel = message.channel
+ yield from self.http.delete_message(channel.id, message.id, channel.server.id)
@asyncio.coroutine
def delete_messages(self, messages):
@@ -1045,16 +943,9 @@ class Client:
if len(messages) > 100 or len(messages) < 2:
raise ClientException('Can only delete messages in the range of [2, 100]')
- channel_id = messages[0].channel.id
- url = '{0}/{1}/messages/bulk_delete'.format(endpoints.CHANNELS, channel_id)
- payload = {
- 'messages': [m.id for m in messages]
- }
-
- response = yield from self.session.post(url, headers=self.headers, data=utils.to_json(payload))
- log.debug(request_logging_format.format(method='POST', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ channel = messages[0].channel
+ message_ids = [m.id for m in messages]
+ yield from self.http.delete_messages(channel.id, message_ids, channel.server.id)
@asyncio.coroutine
def purge_from(self, channel, *, limit=100, check=None, before=None, after=None):
@@ -1179,19 +1070,9 @@ class Client:
channel = message.channel
content = str(new_content)
- url = '{}/{}/messages/{}'.format(endpoints.CHANNELS, channel.id, message.id)
- payload = {
- 'content': content
- }
-
- response = yield from self._retry_helper('edit_message', 'PATCH', url, data=utils.to_json(payload))
- log.debug(request_logging_format.format(method='PATCH', response=response))
- yield from utils._verify_successful_response(response)
- data = yield from response.json(encoding='utf-8')
- log.debug(request_success_log.format(response=response, json=payload, data=data))
+ data = yield from self.http.edit_message(message.id, channel.id, content, guild_id=channel.server.id)
return Message(channel=channel, **data)
- @asyncio.coroutine
def _logs_from(self, channel, limit=100, before=None, after=None):
"""|coro|
@@ -1242,21 +1123,7 @@ class Client:
if message.author == client.user:
counter += 1
"""
- url = '{}/{}/messages'.format(endpoints.CHANNELS, channel.id)
- params = {
- 'limit': limit
- }
-
- if before:
- params['before'] = before.id
- if after:
- params['after'] = after.id
-
- response = yield from self.session.get(url, params=params, headers=self.headers)
- log.debug(request_logging_format.format(method='GET', response=response))
- yield from utils._verify_successful_response(response)
- messages = yield from response.json(encoding='utf-8')
- return messages
+ return self.http.logs_from(channel.id, limit, before=before, after=after)
if PY35:
def logs_from(self, channel, limit=100, *, before=None, after=None, reverse=False):
@@ -1356,12 +1223,7 @@ class Client:
HTTPException
Kicking failed.
"""
-
- url = '{0}/{1.server.id}/members/{1.id}'.format(endpoints.SERVERS, member)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.kick(member.id, member.server.id)
@asyncio.coroutine
def ban(self, member, delete_message_days=1):
@@ -1390,16 +1252,7 @@ class Client:
HTTPException
Banning failed.
"""
-
- params = {
- 'delete-message-days': delete_message_days
- }
-
- url = '{0}/{1.server.id}/bans/{1.id}'.format(endpoints.SERVERS, member)
- response = yield from self.session.put(url, params=params, headers=self.headers)
- log.debug(request_logging_format.format(method='PUT', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.ban(member.id, member.server.id, delete_message_days)
@asyncio.coroutine
def unban(self, server, user):
@@ -1421,12 +1274,7 @@ class Client:
HTTPException
Unbanning failed.
"""
-
- url = '{0}/{1.id}/bans/{2.id}'.format(endpoints.SERVERS, server, user)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.unban(user.id, server.id)
@asyncio.coroutine
def server_voice_state(self, member, *, mute=False, deafen=False):
@@ -1456,17 +1304,7 @@ class Client:
HTTPException
The operation failed.
"""
-
- url = '{0}/{1.server.id}/members/{1.id}'.format(endpoints.SERVERS, member)
- payload = {
- 'mute': mute,
- 'deaf': deafen
- }
-
- response = yield from self.session.patch(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.server_voice_state(member.id, member.server.id, mute=mute, deafen=deafen)
@asyncio.coroutine
def edit_profile(self, password=None, **fields):
@@ -1527,30 +1365,21 @@ class Client:
if not_bot_account and password is None:
raise ClientException('Password is required for non-bot accounts.')
- payload = {
+ args = {
'password': password,
'username': fields.get('username', self.user.name),
'avatar': avatar
}
if not_bot_account:
- payload['email'] = fields.get('email', self.email)
+ args['email'] = fields.get('email', self.email)
if 'new_password' in fields:
- payload['new_password'] = fields['new_password']
-
-
- r = yield from self.session.patch(endpoints.ME, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=r))
- yield from utils._verify_successful_response(r)
-
- data = yield from r.json(encoding='utf-8')
- log.debug(request_success_log.format(response=r, json=payload, data=data))
+ args['new_password'] = fields['new_password']
+ yield from self.http.edit_profile(**args)
if not_bot_account:
- self.token = data['token']
self.email = data['email']
- self.headers['authorization'] = self.token
if self.cache_auth:
self._update_cache(self.email, password)
@@ -1608,24 +1437,12 @@ class Client:
Changing the nickname failed.
"""
+ nickname = nickname if nickname else ''
+
if member == self.user:
- fmt = '{0}/{1.server.id}/members/@me/nick'
+ yield from self.http.change_my_nickname(member.server.id, nickname)
else:
- fmt = '{0}/{1.server.id}/members/{1.id}'
-
- url = fmt.format(endpoints.SERVERS, member)
-
- payload = {
- # oddly enough, this endpoint requires '' to clear the nickname
- # instead of the more consistent 'null', this might change in the
- # future, or not.
- 'nick': nickname if nickname else ''
- }
-
- r = yield from self.session.patch(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=r))
- yield from utils._verify_successful_response(r)
- yield from r.release()
+ yield from self.http.change_nickname(member.server.id, member.id, nickname)
# Channel management
@@ -1662,26 +1479,7 @@ class Client:
Editing the channel failed.
"""
- url = '{0}/{1.id}'.format(endpoints.CHANNELS, channel)
- payload = {
- 'name': options.get('name', channel.name),
- 'topic': options.get('topic', channel.topic),
- }
-
- user_limit = options.get('user_limit')
- if user_limit is not None:
- payload['user_limit'] = user_limit
-
- bitrate = options.get('bitrate')
- if bitrate is not None:
- payload['bitrate'] = bitrate
-
- r = yield from self.session.patch(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=r))
- yield from utils._verify_successful_response(r)
-
- data = yield from r.json(encoding='utf-8')
- log.debug(request_success_log.format(response=r, json=payload, data=data))
+ yield from self.http.edit_channel(channel.id, **options)
@asyncio.coroutine
def move_channel(self, channel, position):
@@ -1735,13 +1533,7 @@ class Client:
channels.insert(position, channel)
payload = [{'id': c.id, 'position': index } for index, c in enumerate(channels)]
-
- r = yield from self.session.patch(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=r))
- yield from utils._verify_successful_response(r)
-
- yield from r.release()
- log.debug(request_success_log.format(json=payload, response=r, data={}))
+ yield from self.http.patch(url, json=payload, bucket='move_channel')
@asyncio.coroutine
def create_channel(self, server, name, type=None):
@@ -1779,18 +1571,7 @@ class Client:
if type is None:
type = ChannelType.text
- payload = {
- 'name': name,
- 'type': str(type)
- }
-
- url = '{0}/{1.id}/channels'.format(endpoints.SERVERS, server)
- response = yield from self.session.post(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='POST', response=response))
- yield from utils._verify_successful_response(response)
-
- data = yield from response.json(encoding='utf-8')
- log.debug(request_success_log.format(response=response, data=data, json=payload))
+ data = yield from self.http.create_channel(server.id, name, str(type))
channel = Channel(server=server, **data)
return channel
@@ -1817,12 +1598,7 @@ class Client:
HTTPException
Deleting the channel failed.
"""
-
- url = '{}/{}'.format(endpoints.CHANNELS, channel.id)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.delete_channel(channel.id)
# Server management
@@ -1847,12 +1623,7 @@ class Client:
HTTPException
If leaving the server failed.
"""
-
- url = '{}/@me/guilds/{.id}'.format(endpoints.USERS, server)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.leave_server(server.id)
@asyncio.coroutine
def delete_server(self, server):
@@ -1874,11 +1645,7 @@ class Client:
You do not have permissions to delete the server.
"""
- url = '{0}/{1.id}'.format(endpoints.SERVERS, server)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.delete_server(server.id)
@asyncio.coroutine
def create_server(self, name, region=None, icon=None):
@@ -1918,17 +1685,7 @@ class Client:
else:
region = region.name
- payload = {
- 'icon': icon,
- 'name': name,
- 'region': region
- }
-
- r = yield from self.session.post(endpoints.SERVERS, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='POST', response=r))
- yield from utils._verify_successful_response(r)
- data = yield from r.json(encoding='utf-8')
- log.debug(request_success_log.format(response=r, json=payload, data=data))
+ data = yield from self.http.create_server(name, region, icon)
return Server(**data)
@asyncio.coroutine
@@ -1984,30 +1741,18 @@ class Client:
else:
icon = None
- payload = {
- 'region': str(fields.get('region', server.region)),
- 'afk_timeout': fields.get('afk_timeout', server.afk_timeout),
- 'icon': icon,
- 'name': fields.get('name', server.name),
- }
-
- afk_channel = fields.get('afk_channel')
- if afk_channel is None:
- afk_channel = server.afk_channel
-
- payload['afk_channel'] = getattr(afk_channel, 'id', None)
+ fields['icon'] = icon
+ if 'afk_channel' in fields:
+ fields['afk_channel_id'] = fields['afk_channel'].id
if 'owner' in fields:
if server.owner != server.me:
raise InvalidArgument('To transfer ownership you must be the owner of the server.')
- payload['owner_id'] = fields['owner'].id
+ fields['owner_id'] = fields['owner'].id
+
+ yield from self.http.edit_server(server.id, **fields)
- url = '{0}/{1.id}'.format(endpoints.SERVERS, server)
- r = yield from self.session.patch(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=r))
- yield from utils._verify_successful_response(r)
- yield from r.release()
@asyncio.coroutine
def get_bans(self, server):
@@ -2036,11 +1781,7 @@ class Client:
A list of :class:`User` that have been banned.
"""
- url = '{0}/{1.id}/bans'.format(endpoints.SERVERS, server)
- resp = yield from self.session.get(url, headers=self.headers)
- log.debug(request_logging_format.format(method='GET', response=resp))
- yield from utils._verify_successful_response(resp)
- data = yield from resp.json(encoding='utf-8')
+ data = yield from self.http.get_bans(server.id)
return [User(**user['user']) for user in data]
# Invite management
@@ -2092,20 +1833,7 @@ class Client:
The invite that was created.
"""
- payload = {
- 'max_age': options.get('max_age', 0),
- 'max_uses': options.get('max_uses', 0),
- 'temporary': options.get('temporary', False),
- 'xkcdpass': options.get('xkcd', False)
- }
-
- url = '{0}/{1.id}/invites'.format(endpoints.CHANNELS, destination)
- response = yield from self.session.post(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='POST', response=response))
-
- yield from utils._verify_successful_response(response)
- data = yield from response.json(encoding='utf-8')
- log.debug(request_success_log.format(json=payload, response=response, data=data))
+ data = yield from self.http.create_invite(destination.id, **options)
self._fill_invite_data(data)
return Invite(**data)
@@ -2139,12 +1867,8 @@ class Client:
The invite from the URL/ID.
"""
- destination = self._resolve_invite(url)
- rurl = '{0}/invite/{1}'.format(endpoints.API_BASE, destination)
- response = yield from self.session.get(rurl, headers=self.headers)
- log.debug(request_logging_format.format(method='GET', response=response))
- yield from utils._verify_successful_response(response)
- data = yield from response.json(encoding='utf-8')
+ invite_id = self._resolve_invite(url)
+ data = yield from self.http.get_invite(invite_id)
self._fill_invite_data(data)
return Invite(**data)
@@ -2174,11 +1898,7 @@ class Client:
The list of invites that are currently active.
"""
- url = '{0}/{1.id}/invites'.format(endpoints.SERVERS, server)
- resp = yield from self.session.get(url, headers=self.headers)
- log.debug(request_logging_format.format(method='GET', response=resp))
- yield from utils._verify_successful_response(resp)
- data = yield from resp.json(encoding='utf-8')
+ data = yield from self.http.invites_from(server.id)
result = []
for invite in data:
channel = server.get_channel(invite['channel']['id'])
@@ -2210,12 +1930,8 @@ class Client:
The invite is invalid or expired.
"""
- destination = self._resolve_invite(invite)
- url = '{0}/invite/{1}'.format(endpoints.API_BASE, destination)
- response = yield from self.session.post(url, headers=self.headers)
- log.debug(request_logging_format.format(method='POST', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ invite_id = self._resolve_invite(invite)
+ yield from self.http.accept_invite(invite_id)
@asyncio.coroutine
def delete_invite(self, invite):
@@ -2241,12 +1957,8 @@ class Client:
Revoking the invite failed.
"""
- destination = self._resolve_invite(invite)
- url = '{0}/invite/{1}'.format(endpoints.API_BASE, destination)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ invite_id = self._resolve_invite(invite)
+ yield from self.http.delete_invite(invite_id)
# Role management
@@ -2298,13 +2010,7 @@ class Client:
roles.append(role.id)
payload = [{"id": z[0], "position": z[1]} for z in zip(roles, change_range)]
-
- r = yield from self.session.patch(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=r))
- yield from utils._verify_successful_response(r)
-
- data = yield from r.json()
- log.debug(request_success_log.format(json=payload, response=r, data=data))
+ yield from self.http.patch(url, json=payload, bucket='move_role')
@asyncio.coroutine
def edit_role(self, server, role, **fields):
@@ -2345,11 +2051,6 @@ class Client:
Editing the role failed.
"""
- url = '{0}/{1.id}/roles/{2.id}'.format(endpoints.SERVERS, server, role)
- color = fields.get('color')
- if color is None:
- color = fields.get('colour', role.colour)
-
payload = {
'name': fields.get('name', role.name),
'permissions': fields.get('permissions', role.permissions).value,
@@ -2358,12 +2059,7 @@ class Client:
'mentionable': fields.get('mentionable', role.mentionable)
}
- r = yield from self.session.patch(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=r))
- yield from utils._verify_successful_response(r)
-
- data = yield from r.json(encoding='utf-8')
- log.debug(request_success_log.format(json=payload, response=r, data=data))
+ yield from self.http.edit_role(server.id, role.id, **payload)
@asyncio.coroutine
def delete_role(self, server, role):
@@ -2386,24 +2082,11 @@ class Client:
Deleting the role failed.
"""
- url = '{0}/{1.id}/roles/{2.id}'.format(endpoints.SERVERS, server, role)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.delete_role(server.id, role.id)
@asyncio.coroutine
def _replace_roles(self, member, roles):
- url = '{0}/{1.server.id}/members/{1.id}'.format(endpoints.SERVERS, member)
-
- payload = {
- 'roles': roles
- }
-
- r = yield from self.session.patch(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=r))
- yield from utils._verify_successful_response(r)
- yield from r.release()
+ yield from self.http.replace_roles(member.id, member.server.id, roles)
@asyncio.coroutine
def add_roles(self, member, *roles):
@@ -2521,12 +2204,7 @@ class Client:
is stored in cache.
"""
- url = '{0}/{1.id}/roles'.format(endpoints.SERVERS, server)
- r = yield from self.session.post(url, headers=self.headers)
- log.debug(request_logging_format.format(method='POST', response=r))
- yield from utils._verify_successful_response(r)
-
- data = yield from r.json(encoding='utf-8')
+ data = yield from self.http.create_role(server.id)
role = Role(server=server, **data)
# we have to call edit because you can't pass a payload to the
@@ -2581,8 +2259,6 @@ class Client:
or the target type was not :class:`Role` or :class:`Member`.
"""
- url = '{0}/{1.id}/permissions/{2.id}'.format(endpoints.CHANNELS, channel, target)
-
allow = Permissions.none() if allow is None else allow
deny = Permissions.none() if deny is None else deny
@@ -2592,23 +2268,14 @@ class Client:
deny = deny.value
allow = allow.value
- payload = {
- 'id': target.id,
- 'allow': allow,
- 'deny': deny
- }
-
if isinstance(target, Member):
- payload['type'] = 'member'
+ perm_type = 'member'
elif isinstance(target, Role):
- payload['type'] = 'role'
+ perm_type = 'role'
else:
raise InvalidArgument('target parameter must be either discord.Member or discord.Role')
- r = yield from self.session.put(url, data=utils.to_json(payload), headers=self.headers)
- log.debug(request_logging_format.format(method='PUT', response=r))
- yield from utils._verify_successful_response(r)
- yield from r.release()
+ yield from self.http.edit_channel_permissions(channel.id, target.id, allow, deny, perm_type)
@asyncio.coroutine
def delete_channel_permissions(self, channel, target):
@@ -2637,12 +2304,7 @@ class Client:
HTTPException
Deleting channel specific permissions failed.
"""
-
- url = '{0}/{1.id}/permissions/{2.id}'.format(endpoints.CHANNELS, channel, target)
- response = yield from self.session.delete(url, headers=self.headers)
- log.debug(request_logging_format.format(method='DELETE', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.delete_channel_permissions(channel.id, target.id)
# Voice management
@@ -2676,18 +2338,10 @@ class Client:
You do not have permissions to move the member.
"""
- url = '{0}/{1.server.id}/members/{1.id}'.format(endpoints.SERVERS, member)
-
if getattr(channel, 'type', ChannelType.text) != ChannelType.voice:
raise InvalidArgument('The channel provided must be a voice channel.')
- payload = utils.to_json({
- 'channel_id': channel.id
- })
- response = yield from self.session.patch(url, data=payload, headers=self.headers)
- log.debug(request_logging_format.format(method='PATCH', response=response))
- yield from utils._verify_successful_response(response)
- yield from response.release()
+ yield from self.http.move_member(member.id, member.server.id, channel.id)
@asyncio.coroutine
def join_voice_channel(self, channel):
@@ -2817,10 +2471,7 @@ class Client:
HTTPException
Retrieving the information failed somehow.
"""
- url = '{}/@me'.format(endpoints.APPLICATIONS)
- resp = yield from self.session.get(url, headers=self.headers)
- yield from utils._verify_successful_response(resp)
- data = yield from resp.json()
+ data = yield from self.http.application_info()
return AppInfo(id=data['id'], name=data['name'],
description=data['description'], icon=data['icon'])