You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
94 lines
3.2 KiB
94 lines
3.2 KiB
# -*- coding: utf-8 -*-
|
|
|
|
"""
|
|
The MIT License (MIT)
|
|
|
|
Copyright (c) 2015 Rapptz
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a
|
|
copy of this software and associated documentation files (the "Software"),
|
|
to deal in the Software without restriction, including without limitation
|
|
the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
and/or sell copies of the Software, and to permit persons to whom the
|
|
Software is furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
|
DEALINGS IN THE SOFTWARE.
|
|
"""
|
|
|
|
from re import split as re_split
|
|
from .errors import HTTPException, Forbidden, NotFound, InvalidArgument
|
|
import datetime
|
|
from base64 import b64encode
|
|
import asyncio
|
|
|
|
def parse_time(timestamp):
|
|
if timestamp:
|
|
return datetime.datetime(*map(int, re_split(r'[^\d]', timestamp.replace('+00:00', ''))))
|
|
return None
|
|
|
|
def find(predicate, seq):
|
|
"""A helper to return the first element found in the sequence
|
|
that meets the predicate. For example: ::
|
|
|
|
member = find(lambda m: m.name == 'Mighty', channel.server.members)
|
|
|
|
would find the first :class:`Member` whose name is 'Mighty' and return it.
|
|
|
|
This is different from `filter`_ due to the fact it stops the moment it finds
|
|
a valid entry.
|
|
|
|
.. _filter: https://docs.python.org/3.6/library/functions.html#filter
|
|
|
|
:param predicate: A function that returns a boolean-like result.
|
|
:param seq: The sequence to iterate through.
|
|
:return: The first result of the predicate that returned a ``True``-like value or ``None`` if nothing was found.
|
|
"""
|
|
|
|
for element in seq:
|
|
if predicate(element):
|
|
return element
|
|
return None
|
|
|
|
def _null_event(*args, **kwargs):
|
|
pass
|
|
|
|
@asyncio.coroutine
|
|
def _verify_successful_response(response):
|
|
code = response.status
|
|
success = code >= 200 and code < 300
|
|
if not success:
|
|
data = yield from response.json()
|
|
message = data.get('message')
|
|
if code == 403:
|
|
raise Forbidden(response, message)
|
|
elif code == 404:
|
|
raise NotFound(response, message)
|
|
raise HTTPException(response, message)
|
|
|
|
def _get_mime_type_for_image(data):
|
|
if data.startswith(b'\x89\x50\x4E\x47\x0D\x0A\x1A\x0A'):
|
|
return 'image/png'
|
|
elif data.startswith(b'\xFF\xD8') and data.endswith(b'\xFF\xD9'):
|
|
return 'image/jpeg'
|
|
else:
|
|
raise InvalidArgument('Unsupported image type given')
|
|
|
|
def _bytes_to_base64_data(data):
|
|
fmt = 'data:{mime};base64,{data}'
|
|
mime = _get_mime_type_for_image(data)
|
|
b64 = b64encode(data).decode('ascii')
|
|
return fmt.format(mime=mime, data=b64)
|
|
|
|
try:
|
|
create_task = asyncio.ensure_future
|
|
except AttributeError:
|
|
create_task = asyncio.async
|
|
|