116 行
3.8 KiB
Python
116 行
3.8 KiB
Python
import re
|
|
import json
|
|
import logging
|
|
import requests
|
|
|
|
from yandex_music.utils.response import Response
|
|
from yandex_music.exceptions import Unauthorized, BadRequest, NetworkError, YandexMusicError
|
|
|
|
|
|
USER_AGENT = 'Yandex-Music-API'
|
|
HEADERS = {
|
|
'X-Yandex-Music-Client': 'WindowsPhone/3.17',
|
|
'User-Agent': 'Windows 10',
|
|
# 'X-Yandex-Music-Client': 'Yandex-Music-API',
|
|
# 'User-Agent': 'Yandex-Music-API',
|
|
'Connection': 'Keep-Alive'
|
|
}
|
|
|
|
|
|
class Request(object):
|
|
def __init__(self,
|
|
client,
|
|
headers=None,
|
|
proxies=None):
|
|
|
|
self.client = client
|
|
|
|
self.headers = headers or HEADERS
|
|
|
|
if self.client.token:
|
|
self.set_authorization(self.client.token)
|
|
|
|
self.proxies = proxies # TODO
|
|
|
|
def set_authorization(self, token):
|
|
self.headers.update({'Authorization': f'OAuth {token}'})
|
|
|
|
@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('-', '_'))
|
|
key = key.replace('client', 'client_')
|
|
cleaned_object.update({key: value})
|
|
|
|
return cleaned_object
|
|
|
|
def _parse(self, json_data) -> Response:
|
|
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)
|
|
raise YandexMusicError('Server response could not be decoded using UTF-8')
|
|
except (AttributeError, ValueError):
|
|
raise YandexMusicError('Invalid server response')
|
|
|
|
return Response.de_json(data, self.client)
|
|
|
|
def _request_wrapper(self, *args, **kwargs):
|
|
if 'headers' not in kwargs:
|
|
kwargs['headers'] = {}
|
|
|
|
kwargs['headers']['user-agent'] = USER_AGENT
|
|
|
|
try:
|
|
resp = requests.request(verify=False, *args, **kwargs)
|
|
except requests.Timeout:
|
|
raise TimeoutError()
|
|
except requests.RequestException as e:
|
|
raise NetworkError(e)
|
|
|
|
if 200 <= resp.status_code <= 299:
|
|
return resp
|
|
|
|
message = self._parse(resp.content).error or 'Unknown HTTPError'
|
|
|
|
if resp.status_code in (401, 403):
|
|
raise Unauthorized(message)
|
|
elif resp.status_code == 400:
|
|
raise BadRequest(message)
|
|
elif resp.status_code in (404, 409, 413):
|
|
raise NetworkError(message)
|
|
|
|
elif resp.status_code == 502:
|
|
raise NetworkError('Bad Gateway')
|
|
else:
|
|
raise NetworkError(f'{message} ({resp.status_code})')
|
|
|
|
def get(self, url, params=None, timeout=5, *args, **kwargs):
|
|
result = self._request_wrapper('GET', url, params=params, headers=self.headers, timeout=timeout,
|
|
*args, **kwargs)
|
|
|
|
return self._parse(result.content).result
|
|
|
|
def post(self, url, data=None, timeout=5, *args, **kwargs):
|
|
result = self._request_wrapper('POST', url, headers=self.headers, data=data, timeout=timeout,
|
|
*args, **kwargs)
|
|
|
|
return self._parse(result.content).result
|
|
|
|
def retrieve(self, url, params=None, timeout=5, *args, **kwargs):
|
|
return self._request_wrapper('GET', url, params=params, headers=self.headers, timeout=timeout,
|
|
*args, **kwargs)
|
|
|
|
def download(self, url, filename, timeout=5, *args, **kwargs):
|
|
result = self.retrieve(url, timeout=timeout, *args, *kwargs)
|
|
with open(filename, 'wb') as f:
|
|
f.write(result.content)
|