2019-05-07 06:02:21 +09:00
|
|
|
|
import re
|
|
|
|
|
import logging
|
2020-06-20 22:54:46 +09:00
|
|
|
|
import keyword
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2020-03-22 23:05:09 +09:00
|
|
|
|
from typing import TYPE_CHECKING, Optional, Union
|
|
|
|
|
|
2019-12-23 18:07:11 +09:00
|
|
|
|
# Не используется ujson из-за отсутствия в нём object_hook'a
|
|
|
|
|
# Отправка вообще application/x-www-form-urlencoded, а не JSON'a
|
|
|
|
|
# https://github.com/psf/requests/blob/master/requests/models.py#L508
|
|
|
|
|
import json
|
|
|
|
|
|
2019-12-26 18:01:17 +09:00
|
|
|
|
import requests
|
|
|
|
|
|
2019-05-16 21:45:25 +09:00
|
|
|
|
from yandex_music.utils.response import Response
|
2021-02-03 21:28:10 +09:00
|
|
|
|
from yandex_music.exceptions import (
|
2022-02-21 06:15:27 +09:00
|
|
|
|
UnauthorizedError,
|
|
|
|
|
BadRequestError,
|
2021-02-03 21:28:10 +09:00
|
|
|
|
NetworkError,
|
|
|
|
|
YandexMusicError,
|
2022-02-21 06:15:27 +09:00
|
|
|
|
TimedOutError,
|
|
|
|
|
NotFoundError,
|
2021-02-03 21:28:10 +09:00
|
|
|
|
)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2020-03-22 23:05:09 +09:00
|
|
|
|
if TYPE_CHECKING:
|
|
|
|
|
from yandex_music import Client
|
|
|
|
|
|
|
|
|
|
|
2019-05-07 06:02:21 +09:00
|
|
|
|
USER_AGENT = 'Yandex-Music-API'
|
|
|
|
|
HEADERS = {
|
2023-05-05 00:55:18 +09:00
|
|
|
|
'X-Yandex-Music-Client': 'YandexMusicAndroid/24023231',
|
2019-05-07 06:02:21 +09:00
|
|
|
|
}
|
2022-10-31 22:57:28 +09:00
|
|
|
|
DEFAULT_TIMEOUT = 5
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2021-09-19 22:47:19 +09:00
|
|
|
|
reserved_names = keyword.kwlist + ['client']
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2019-08-25 17:49:02 +09:00
|
|
|
|
logging.getLogger('urllib3').setLevel(logging.WARNING)
|
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
default_timeout = object()
|
|
|
|
|
|
2019-08-25 17:49:02 +09:00
|
|
|
|
|
2019-08-18 18:54:13 +09:00
|
|
|
|
class Request:
|
2020-03-22 23:05:09 +09:00
|
|
|
|
"""Вспомогательный класс для yandex_music, представляющий методы для выполнения POST и GET запросов, скачивания
|
2019-08-25 17:49:02 +09:00
|
|
|
|
файлов.
|
2019-06-04 22:30:33 +09:00
|
|
|
|
|
|
|
|
|
Args:
|
Удаление избыточной информации (#247)
Классы: Account, AutoRenewable, PassportPhone, Permissions, Plus, Price,
Product, Status, Subscription, UserSettings, Album, Label,
TrackPosition, Playlist
У всех классов изменено описание атрибута client
2020-03-22 04:49:20 +09:00
|
|
|
|
client (:obj:`yandex_music.Client`, optional): Клиент Yandex Music.
|
2019-06-13 04:56:38 +09:00
|
|
|
|
headers (:obj:`dict`, optional): Заголовки передаваемые с каждым запросом.
|
2019-11-19 05:54:46 +09:00
|
|
|
|
proxy_url (:obj:`str`, optional): Прокси.
|
2019-06-04 22:30:33 +09:00
|
|
|
|
"""
|
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
def __init__(self, client=None, headers=None, proxy_url=None, timeout=default_timeout):
|
2019-08-23 03:56:02 +09:00
|
|
|
|
self.headers = headers or HEADERS.copy()
|
2019-05-16 23:06:05 +09:00
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
self._timeout = DEFAULT_TIMEOUT
|
|
|
|
|
self.set_timeout(timeout)
|
|
|
|
|
|
2019-11-19 05:54:46 +09:00
|
|
|
|
self.client = self.set_and_return_client(client)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2022-02-20 02:59:53 +09:00
|
|
|
|
# aiohttp
|
|
|
|
|
self.proxy_url = proxy_url
|
|
|
|
|
|
|
|
|
|
# requests
|
2019-11-19 05:54:46 +09:00
|
|
|
|
self.proxies = {'http': proxy_url, 'https': proxy_url} if proxy_url else None
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2020-06-20 22:54:46 +09:00
|
|
|
|
def set_language(self, lang: str) -> None:
|
|
|
|
|
"""Добавляет заголовок языка для каждого запроса.
|
|
|
|
|
|
|
|
|
|
Note:
|
|
|
|
|
Возможные значения `lang`: en/uz/uk/us/ru/kk/hy.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
lang (:obj:`str`): Язык.
|
|
|
|
|
"""
|
|
|
|
|
self.headers.update({'Accept-Language': lang})
|
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
def set_timeout(self, timeout: Union[int, float, object] = default_timeout):
|
|
|
|
|
"""Устанавливает время ожидания для всех запросов.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
timeout (:obj:`int` | :obj:`float`): Время ожидания от сервера.
|
|
|
|
|
"""
|
|
|
|
|
self._timeout = timeout
|
|
|
|
|
if timeout is default_timeout:
|
|
|
|
|
self._timeout = DEFAULT_TIMEOUT
|
|
|
|
|
|
2020-03-22 23:05:09 +09:00
|
|
|
|
def set_authorization(self, token: str) -> None:
|
|
|
|
|
"""Добавляет заголовок авторизации для каждого запроса.
|
|
|
|
|
|
|
|
|
|
Note:
|
|
|
|
|
Используется при передаче своего экземпляра Request'a клиенту.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
token (:obj:`str`): OAuth токен.
|
|
|
|
|
"""
|
2019-05-16 23:06:05 +09:00
|
|
|
|
self.headers.update({'Authorization': f'OAuth {token}'})
|
|
|
|
|
|
2020-03-22 23:05:09 +09:00
|
|
|
|
def set_and_return_client(self, client) -> 'Client':
|
|
|
|
|
"""Принимает клиент и присваивает его текущему объекту. При наличии авторизации добавляет заголовок.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
client (:obj:`yandex_music.Client`): Клиент Yandex Music.
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
|
:obj:`yandex_music.Client`: Клиент Yandex Music.
|
|
|
|
|
"""
|
2019-11-19 05:54:46 +09:00
|
|
|
|
self.client = client
|
|
|
|
|
|
|
|
|
|
if self.client and self.client.token:
|
|
|
|
|
self.set_authorization(self.client.token)
|
|
|
|
|
|
|
|
|
|
return self.client
|
|
|
|
|
|
2019-05-07 06:02:21 +09:00
|
|
|
|
@staticmethod
|
2020-03-22 23:05:09 +09:00
|
|
|
|
def _convert_camel_to_snake(text: str) -> str:
|
|
|
|
|
"""Конвертация CamelCase в SnakeCase.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
text (:obj:`str`): Название переменной в CamelCase.
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
|
:obj:`str`: Название переменной в SnakeCase.
|
|
|
|
|
"""
|
2021-09-19 22:47:19 +09:00
|
|
|
|
s = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', text)
|
|
|
|
|
return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s).lower()
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
|
|
|
|
@staticmethod
|
2020-03-22 23:05:09 +09:00
|
|
|
|
def _object_hook(obj: dict) -> dict:
|
|
|
|
|
"""Нормализация имён переменных пришедших с API.
|
|
|
|
|
|
|
|
|
|
Note:
|
|
|
|
|
В названии переменной заменяет "-" на "_", конвертирует в SnakeCase, если название является
|
2021-09-19 22:47:19 +09:00
|
|
|
|
зарезервированным словом или "client" - добавляет "_" в конец. Если название переменной начинается с цифры -
|
2020-03-22 23:05:09 +09:00
|
|
|
|
добавляет в начало "_".
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
obj (:obj:`dict`): Словарь, где ключ название переменной, а значение - содержимое.
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
|
:obj:`dict`: Тот же словарь, что и на входе, но с нормализованными ключами.
|
|
|
|
|
"""
|
2019-05-07 06:02:21 +09:00
|
|
|
|
cleaned_object = {}
|
|
|
|
|
for key, value in obj.items():
|
|
|
|
|
key = Request._convert_camel_to_snake(key.replace('-', '_'))
|
2020-06-20 22:54:46 +09:00
|
|
|
|
key = key.lower()
|
|
|
|
|
|
|
|
|
|
if key in reserved_names:
|
2019-12-26 18:01:17 +09:00
|
|
|
|
key += '_'
|
2019-06-01 17:23:28 +09:00
|
|
|
|
|
|
|
|
|
if len(key) and key[0].isdigit():
|
|
|
|
|
key = '_' + key
|
|
|
|
|
|
2019-05-07 06:02:21 +09:00
|
|
|
|
cleaned_object.update({key: value})
|
|
|
|
|
|
|
|
|
|
return cleaned_object
|
|
|
|
|
|
2020-03-22 23:05:09 +09:00
|
|
|
|
def _parse(self, json_data: bytes) -> Optional[Response]:
|
|
|
|
|
"""Разбор ответа от API.
|
|
|
|
|
|
|
|
|
|
Note:
|
|
|
|
|
Если данные отсутствуют в `result`, то переформировывает ответ используя данные из корня.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
json_data (:obj:`bytes`): Ответ от API.
|
|
|
|
|
|
|
|
|
|
Returns:
|
|
|
|
|
:obj:`yandex_music.utils.response.Response`: Ответ API.
|
|
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
|
:class:`yandex_music.exceptions.YandexMusicError`: Базовое исключение библиотеки.
|
|
|
|
|
"""
|
2019-05-07 06:02:21 +09:00
|
|
|
|
try:
|
|
|
|
|
decoded_s = json_data.decode('utf-8')
|
|
|
|
|
data = json.loads(decoded_s, object_hook=Request._object_hook)
|
2019-12-23 18:07:11 +09:00
|
|
|
|
|
2019-05-07 06:02:21 +09:00
|
|
|
|
except UnicodeDecodeError:
|
2021-02-03 21:28:10 +09:00
|
|
|
|
logging.getLogger(__name__).debug('Logging raw invalid UTF-8 response:\n%r', json_data)
|
2019-05-10 00:28:46 +09:00
|
|
|
|
raise YandexMusicError('Server response could not be decoded using UTF-8')
|
2019-05-16 21:45:25 +09:00
|
|
|
|
except (AttributeError, ValueError):
|
|
|
|
|
raise YandexMusicError('Invalid server response')
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2019-11-24 09:01:24 +09:00
|
|
|
|
if data.get('result') is None:
|
2019-11-19 22:53:21 +09:00
|
|
|
|
data = {'result': data, 'error': data.get('error'), 'error_description': data.get('error_description')}
|
|
|
|
|
|
2019-05-16 21:45:25 +09:00
|
|
|
|
return Response.de_json(data, self.client)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
|
|
|
|
def _request_wrapper(self, *args, **kwargs):
|
2020-03-22 23:05:09 +09:00
|
|
|
|
"""Обёртка над запросом библиотеки `requests`.
|
|
|
|
|
|
|
|
|
|
Note:
|
|
|
|
|
Добавляет необходимые заголовки для запроса, обрабатывает статус коды, следит за таймаутом, кидает
|
|
|
|
|
необходимые исключения, возвращает ответ. Передаёт пользовательские аргументы в запрос.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
*args: Произвольные аргументы для `requests.request`.
|
|
|
|
|
**kwargs: Произвольные ключевые аргументы для `requests.request`.
|
|
|
|
|
|
|
|
|
|
Returns:
|
2022-02-20 02:59:53 +09:00
|
|
|
|
:obj:`bytes`: Тело ответа.
|
2020-03-22 23:05:09 +09:00
|
|
|
|
|
|
|
|
|
Raises:
|
2022-02-21 06:15:27 +09:00
|
|
|
|
:class:`yandex_music.exceptions.TimedOutError`: При превышении времени ожидания.
|
|
|
|
|
:class:`yandex_music.exceptions.UnauthorizedError`: При невалидном токене, долгом ожидании прямой ссылки на файл.
|
|
|
|
|
:class:`yandex_music.exceptions.BadRequestError`: При неправильном запросе.
|
2020-03-22 23:05:09 +09:00
|
|
|
|
:class:`yandex_music.exceptions.NetworkError`: При проблемах с сетью.
|
|
|
|
|
"""
|
2019-05-07 06:02:21 +09:00
|
|
|
|
if 'headers' not in kwargs:
|
|
|
|
|
kwargs['headers'] = {}
|
|
|
|
|
|
2019-09-18 03:50:58 +09:00
|
|
|
|
kwargs['headers']['User-Agent'] = USER_AGENT
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
if kwargs['timeout'] is default_timeout:
|
|
|
|
|
kwargs['timeout'] = self._timeout
|
|
|
|
|
|
2019-05-07 06:02:21 +09:00
|
|
|
|
try:
|
2019-08-19 01:04:51 +09:00
|
|
|
|
resp = requests.request(*args, **kwargs)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
except requests.Timeout:
|
2022-02-21 06:15:27 +09:00
|
|
|
|
raise TimedOutError()
|
2019-05-16 21:45:25 +09:00
|
|
|
|
except requests.RequestException as e:
|
|
|
|
|
raise NetworkError(e)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
|
|
|
|
if 200 <= resp.status_code <= 299:
|
2022-02-20 02:59:53 +09:00
|
|
|
|
return resp.content
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2022-02-21 05:03:30 +09:00
|
|
|
|
try:
|
|
|
|
|
parse = self._parse(resp.content)
|
|
|
|
|
message = parse.get_error()
|
|
|
|
|
except YandexMusicError:
|
|
|
|
|
message = 'Unknown HTTPError'
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2021-02-28 03:50:04 +09:00
|
|
|
|
if resp.status_code in (401, 403):
|
2022-02-21 06:15:27 +09:00
|
|
|
|
raise UnauthorizedError(message)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
elif resp.status_code == 400:
|
2022-02-21 06:15:27 +09:00
|
|
|
|
raise BadRequestError(message)
|
|
|
|
|
elif resp.status_code == 404:
|
|
|
|
|
raise NotFoundError(message)
|
|
|
|
|
elif resp.status_code in (409, 413):
|
2019-05-16 21:45:25 +09:00
|
|
|
|
raise NetworkError(message)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
|
|
|
|
elif resp.status_code == 502:
|
|
|
|
|
raise NetworkError('Bad Gateway')
|
|
|
|
|
else:
|
2022-02-21 05:03:30 +09:00
|
|
|
|
raise NetworkError(f'{message} ({resp.status_code}): {resp.content}')
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
def get(
|
|
|
|
|
self, url: str, params: dict = None, timeout: Union[int, float] = default_timeout, *args, **kwargs
|
|
|
|
|
) -> Union[dict, str]:
|
2020-03-22 23:05:09 +09:00
|
|
|
|
"""Отправка GET запроса.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
url (:obj:`str`): Адрес для запроса.
|
|
|
|
|
params (:obj:`str`): GET параметры для запроса.
|
|
|
|
|
timeout (:obj:`int` | :obj:`float`): Используется как время ожидания ответа от сервера вместо указанного
|
|
|
|
|
при создании пула.
|
|
|
|
|
*args: Произвольные аргументы для `requests.request`.
|
|
|
|
|
**kwargs: Произвольные ключевые аргументы для `requests.request`.
|
|
|
|
|
|
|
|
|
|
Returns:
|
2022-02-20 08:16:11 +09:00
|
|
|
|
:obj:`dict` | :obj:`str`: Обработанное тело ответа.
|
2020-03-22 23:05:09 +09:00
|
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
|
:class:`yandex_music.exceptions.YandexMusicError`: Базовое исключение библиотеки.
|
|
|
|
|
"""
|
2021-02-03 21:28:10 +09:00
|
|
|
|
result = self._request_wrapper(
|
|
|
|
|
'GET', url, params=params, headers=self.headers, proxies=self.proxies, timeout=timeout, *args, **kwargs
|
|
|
|
|
)
|
2019-05-23 18:10:14 +09:00
|
|
|
|
|
2022-02-20 02:59:53 +09:00
|
|
|
|
return self._parse(result).get_result()
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
def post(self, url, data=None, timeout=default_timeout, *args, **kwargs) -> Union[dict, str]:
|
2020-03-22 23:05:09 +09:00
|
|
|
|
"""Отправка POST запроса.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
url (:obj:`str`): Адрес для запроса.
|
|
|
|
|
data (:obj:`str`): POST тело запроса.
|
|
|
|
|
timeout (:obj:`int` | :obj:`float`): Используется как время ожидания ответа от сервера вместо указанного
|
|
|
|
|
при создании пула.
|
|
|
|
|
*args: Произвольные аргументы для `requests.request`.
|
|
|
|
|
**kwargs: Произвольные ключевые аргументы для `requests.request`.
|
|
|
|
|
|
|
|
|
|
Returns:
|
2022-02-20 08:16:11 +09:00
|
|
|
|
:obj:`dict` | :obj:`str`: Обработанное тело ответа.
|
2020-03-22 23:05:09 +09:00
|
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
|
:class:`yandex_music.exceptions.YandexMusicError`: Базовое исключение библиотеки.
|
|
|
|
|
"""
|
2021-02-03 21:28:10 +09:00
|
|
|
|
result = self._request_wrapper(
|
|
|
|
|
'POST', url, headers=self.headers, proxies=self.proxies, data=data, timeout=timeout, *args, **kwargs
|
|
|
|
|
)
|
2019-05-23 18:10:14 +09:00
|
|
|
|
|
2022-02-20 02:59:53 +09:00
|
|
|
|
return self._parse(result).get_result()
|
2019-05-23 18:10:14 +09:00
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
def retrieve(self, url, timeout=default_timeout, *args, **kwargs) -> bytes:
|
2020-03-22 23:05:09 +09:00
|
|
|
|
"""Отправка GET запроса и получение содержимого без обработки (парсинга).
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
url (:obj:`str`): Адрес для запроса.
|
|
|
|
|
timeout (:obj:`int` | :obj:`float`): Используется как время ожидания ответа от сервера вместо указанного
|
|
|
|
|
при создании пула.
|
|
|
|
|
*args: Произвольные аргументы для `requests.request`.
|
|
|
|
|
**kwargs: Произвольные ключевые аргументы для `requests.request`.
|
|
|
|
|
|
|
|
|
|
Returns:
|
2022-02-20 02:59:53 +09:00
|
|
|
|
:obj:`bytes`: Тело ответа.
|
2020-03-22 23:05:09 +09:00
|
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
|
:class:`yandex_music.exceptions.YandexMusicError`: Базовое исключение библиотеки.
|
|
|
|
|
"""
|
2019-11-19 05:54:46 +09:00
|
|
|
|
return self._request_wrapper('GET', url, proxies=self.proxies, timeout=timeout, *args, **kwargs)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2022-10-31 22:57:28 +09:00
|
|
|
|
def download(self, url, filename, timeout=default_timeout, *args, **kwargs) -> None:
|
2020-03-22 23:05:09 +09:00
|
|
|
|
"""Отправка запроса на получение содержимого и его запись в файл.
|
|
|
|
|
|
|
|
|
|
Args:
|
|
|
|
|
url (:obj:`str`): Адрес для запроса.
|
|
|
|
|
filename (:obj:`str`): Путь и(или) название файла вместе с расширением.
|
|
|
|
|
timeout (:obj:`int` | :obj:`float`): Используется как время ожидания ответа от сервера вместо указанного
|
|
|
|
|
при создании пула.
|
|
|
|
|
*args: Произвольные аргументы для `requests.request`.
|
|
|
|
|
**kwargs: Произвольные ключевые аргументы для `requests.request`.
|
|
|
|
|
|
|
|
|
|
Raises:
|
|
|
|
|
:class:`yandex_music.exceptions.YandexMusicError`: Базовое исключение библиотеки.
|
|
|
|
|
"""
|
2019-11-23 22:35:55 +09:00
|
|
|
|
result = self.retrieve(url, timeout=timeout, *args, *kwargs)
|
2019-05-23 18:10:14 +09:00
|
|
|
|
with open(filename, 'wb') as f:
|
2022-02-20 02:59:53 +09:00
|
|
|
|
f.write(result)
|