2019-05-07 06:02:21 +09:00
|
|
|
|
import re
|
|
|
|
|
import json
|
|
|
|
|
import logging
|
|
|
|
|
import requests
|
|
|
|
|
|
2019-11-19 22:53:21 +09:00
|
|
|
|
from yandex_music.utils.captcha_response import CaptchaResponse
|
2019-05-16 21:45:25 +09:00
|
|
|
|
from yandex_music.utils.response import Response
|
2019-11-20 00:35:09 +09:00
|
|
|
|
from yandex_music.exceptions import Unauthorized, BadRequest, NetworkError, YandexMusicError, CaptchaRequired, \
|
|
|
|
|
CaptchaWrong
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
|
|
|
|
USER_AGENT = 'Yandex-Music-API'
|
|
|
|
|
HEADERS = {
|
2019-09-18 03:50:58 +09:00
|
|
|
|
'X-Yandex-Music-Client': 'WindowsPhone/3.20',
|
2019-05-07 06:02:21 +09:00
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
2019-08-25 17:49:02 +09:00
|
|
|
|
logging.getLogger('urllib3').setLevel(logging.WARNING)
|
|
|
|
|
|
|
|
|
|
|
2019-08-18 18:54:13 +09:00
|
|
|
|
class Request:
|
2019-08-25 17:49:02 +09:00
|
|
|
|
"""Вспомогателньный класс для yandex_music предоставляющий методы для выполнения POST и GET запросов, скачивания
|
|
|
|
|
файлов.
|
2019-06-04 22:30:33 +09:00
|
|
|
|
|
|
|
|
|
Args:
|
2019-07-05 02:21:54 +09:00
|
|
|
|
client (:obj:`yandex_music.Client`): Объект класса :class:`yandex_music.Client` представляющий клиент Yandex
|
2019-07-02 20:40:41 +09:00
|
|
|
|
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
|
|
|
|
"""
|
|
|
|
|
|
2019-05-07 06:02:21 +09:00
|
|
|
|
def __init__(self,
|
2019-11-19 05:54:46 +09:00
|
|
|
|
client=None,
|
2019-05-07 06:02:21 +09:00
|
|
|
|
headers=None,
|
2019-11-19 05:54:46 +09:00
|
|
|
|
proxy_url=None):
|
2019-08-23 03:56:02 +09:00
|
|
|
|
self.headers = headers or HEADERS.copy()
|
2019-05-16 23:06:05 +09:00
|
|
|
|
|
2019-11-19 05:54:46 +09:00
|
|
|
|
self.client = self.set_and_return_client(client)
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
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
|
|
|
|
|
2019-05-16 23:06:05 +09:00
|
|
|
|
def set_authorization(self, token):
|
|
|
|
|
self.headers.update({'Authorization': f'OAuth {token}'})
|
|
|
|
|
|
2019-11-19 05:54:46 +09:00
|
|
|
|
def set_and_return_client(self, client):
|
|
|
|
|
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
|
|
|
|
|
def _convert_camel_to_snake(text):
|
|
|
|
|
s1 = re.sub('(.)([A-Z][a-z]+)', r'\1_\2', text)
|
|
|
|
|
return re.sub('([a-z0-9])([A-Z])', r'\1_\2', s1).lower()
|
|
|
|
|
|
|
|
|
|
@staticmethod
|
|
|
|
|
def _object_hook(obj):
|
|
|
|
|
cleaned_object = {}
|
|
|
|
|
for key, value in obj.items():
|
|
|
|
|
key = Request._convert_camel_to_snake(key.replace('-', '_'))
|
New supported objects: Landing, Block, BlockEntity, Chart, ChartItem, MixLink, PersonalPlaylistsData, PlayContext, PlayContextsData, Promotion, TrackId, TrackShorOld (YEAH!)
The following method are wrapped:
- /landing3
Added the ability to download covers
The following field are optional: Playlist.tags
Now, when parsing json, the "client" key is replaced with "client_"
2019-05-25 02:10:39 +09:00
|
|
|
|
key = key.replace('client', 'client_')
|
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
|
|
|
|
|
|
2019-05-16 21:45:25 +09:00
|
|
|
|
def _parse(self, json_data) -> Response:
|
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)
|
|
|
|
|
except UnicodeDecodeError:
|
|
|
|
|
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-19 22:53:21 +09:00
|
|
|
|
if not data.get('result'):
|
|
|
|
|
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):
|
|
|
|
|
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
|
|
|
|
|
|
|
|
|
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:
|
|
|
|
|
raise TimeoutError()
|
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:
|
2019-05-23 18:10:14 +09:00
|
|
|
|
return resp
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2019-11-19 22:53:21 +09:00
|
|
|
|
parse = self._parse(resp.content)
|
|
|
|
|
message = parse.error or 'Unknown HTTPError'
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
2019-11-20 00:35:09 +09:00
|
|
|
|
if 'CAPTCHA' in message:
|
|
|
|
|
exception = CaptchaWrong if 'Wrong' in message else CaptchaRequired
|
|
|
|
|
raise exception(message, CaptchaResponse.de_json(parse.result, self.client))
|
2019-11-19 22:53:21 +09:00
|
|
|
|
elif resp.status_code in (401, 403):
|
2019-05-07 06:02:21 +09:00
|
|
|
|
raise Unauthorized(message)
|
|
|
|
|
elif resp.status_code == 400:
|
|
|
|
|
raise BadRequest(message)
|
|
|
|
|
elif resp.status_code in (404, 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:
|
2019-05-16 23:06:05 +09:00
|
|
|
|
raise NetworkError(f'{message} ({resp.status_code})')
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
New supported objects: Search, Suggestions, Video, Best, AlbumSearchResult, ArtistSearchResult, PlaylistSearchResult, TrackSearchResult, VideoSearchResult,
The following methods are wrapped:
- /search
- /search/suggest
The following classes received optional fields instead of required ones: Album, Artist, Playlist, Track
Added "params" arg to the GET request
Code refactoring
2019-05-18 01:20:34 +09:00
|
|
|
|
def get(self, url, params=None, timeout=5, *args, **kwargs):
|
2019-11-19 05:54:46 +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
|
|
|
|
|
|
|
|
|
return self._parse(result.content).result
|
2019-05-07 06:02:21 +09:00
|
|
|
|
|
New supported objects: Search, Suggestions, Video, Best, AlbumSearchResult, ArtistSearchResult, PlaylistSearchResult, TrackSearchResult, VideoSearchResult,
The following methods are wrapped:
- /search
- /search/suggest
The following classes received optional fields instead of required ones: Album, Artist, Playlist, Track
Added "params" arg to the GET request
Code refactoring
2019-05-18 01:20:34 +09:00
|
|
|
|
def post(self, url, data=None, timeout=5, *args, **kwargs):
|
2019-11-19 05:54:46 +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
|
|
|
|
|
|
|
|
|
return self._parse(result.content).result
|
|
|
|
|
|
2019-09-18 03:50:58 +09:00
|
|
|
|
def retrieve(self, url, timeout=5, *args, **kwargs):
|
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
|
|
|
|
|
2019-05-23 18:10:14 +09:00
|
|
|
|
def download(self, url, filename, timeout=5, *args, **kwargs):
|
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:
|
|
|
|
|
f.write(result.content)
|