270 行
9.1 KiB
Python
270 行
9.1 KiB
Python
import logging
|
|
from datetime import datetime
|
|
|
|
from yandex_music import YandexMusicObject, Status, Settings, PermissionAlerts, Experiments, Artist, Album, Playlist, \
|
|
TracksLikes, Track, AlbumsLikes, ArtistsLikes, PlaylistsLikes, Feed, PromoCodeStatus, DownloadInfo, Search, \
|
|
Suggestions
|
|
from yandex_music.utils.request import Request
|
|
from yandex_music.exceptions import InvalidToken
|
|
|
|
|
|
CLIENT_ID = '23cabbbdc6cd418abb4b39c32c41195d'
|
|
CLIENT_SECRET = '53bc75238f0c4d08a118e51fe9203300'
|
|
|
|
|
|
class Client(YandexMusicObject):
|
|
def __init__(self, username, password, token=None, base_url=None, oauth_url=None, request=None):
|
|
self.logger = logging.getLogger(__name__)
|
|
self.token = token
|
|
|
|
if base_url is None:
|
|
base_url = 'https://api.music.yandex.net'
|
|
if oauth_url is None:
|
|
oauth_url = 'https://oauth.yandex.ru'
|
|
|
|
self.base_url = base_url
|
|
self.oauth_url = oauth_url
|
|
|
|
self._request = request or Request(self)
|
|
|
|
if self.token is None:
|
|
self.token = self._generate_token_by_username_and_password(username, password)
|
|
self.request.set_authorization(self.token)
|
|
|
|
self.account = self.account_status().account
|
|
|
|
@classmethod
|
|
def from_token(cls, token):
|
|
return cls(username=None, password=None, token=token)
|
|
|
|
def _generate_token_by_username_and_password(self, username, password, grant_type='password',
|
|
timeout=None, *args, **kwargs):
|
|
url = f'{self.oauth_url}/token'
|
|
|
|
data = {
|
|
'grant_type': grant_type,
|
|
'client_id': CLIENT_ID,
|
|
'client_secret': CLIENT_SECRET,
|
|
'username': username,
|
|
'password': password
|
|
}
|
|
|
|
result = self._request.post(url, data, timeout=timeout, *args, **kwargs)
|
|
|
|
return result.get('access_token')
|
|
|
|
@staticmethod
|
|
def _validate_token(token):
|
|
if any(x.isspace() for x in token):
|
|
raise InvalidToken()
|
|
|
|
if len(token) != 39:
|
|
raise InvalidToken()
|
|
|
|
return token
|
|
|
|
@property
|
|
def request(self):
|
|
return self._request
|
|
|
|
def account_status(self, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/account/status'
|
|
|
|
result = self._request.get(url, timeout=timeout, *args, **kwargs)
|
|
|
|
return Status.de_json(result, self)
|
|
|
|
def settings(self, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/settings'
|
|
|
|
result = self._request.get(url, timeout=timeout, *args, **kwargs)
|
|
|
|
return Settings.de_json(result, self)
|
|
|
|
def permission_alerts(self, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/permission-alerts'
|
|
|
|
result = self._request.get(url, timeout=timeout, *args, **kwargs)
|
|
|
|
return PermissionAlerts.de_json(result, self)
|
|
|
|
def account_experiments(self, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/account/experiments'
|
|
|
|
result = self._request.get(url, timeout=timeout, *args, **kwargs)
|
|
|
|
return Experiments.de_json(result, self)
|
|
|
|
def consume_promo_code(self, code: str, language='en', timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/account/consume-promo-code'
|
|
|
|
result = self._request.post(url, {'code': code, 'language': language}, timeout=timeout, *args, **kwargs)
|
|
|
|
return PromoCodeStatus.de_json(result, self)
|
|
|
|
def feed(self, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/feed'
|
|
|
|
result = self._request.get(url, timeout=timeout, *args, **kwargs)
|
|
|
|
return Feed.de_json(result, self)
|
|
|
|
def tracks_download_info(self, track_id: str or int, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/tracks/{track_id}/download-info'
|
|
|
|
result = self._request.get(url, timeout=timeout, *args, **kwargs)
|
|
|
|
return DownloadInfo.de_list(result, self)
|
|
|
|
def play_audio(self,
|
|
track_id: str or int,
|
|
from_,
|
|
album_id,
|
|
playlist_id,
|
|
from_cache=False,
|
|
play_id=None,
|
|
uid=None,
|
|
timestamp=None,
|
|
track_length_seconds=0,
|
|
total_played_seconds=0,
|
|
end_position_seconds=0,
|
|
client_now=None,
|
|
timeout=None,
|
|
*args, **kwargs):
|
|
url = f'{self.base_url}/play-audio'
|
|
|
|
data = {
|
|
'track-id': track_id,
|
|
'from-cache': from_cache,
|
|
'from': from_,
|
|
'play-id': play_id or '',
|
|
'uid': uid or self.account.uid,
|
|
'timestamp': timestamp or f'{datetime.now().isoformat()}Z',
|
|
'track-length-seconds': track_length_seconds,
|
|
'total-played-seconds': total_played_seconds,
|
|
'end-position-seconds': end_position_seconds,
|
|
'album-id': album_id,
|
|
'playlist-id': playlist_id,
|
|
'client-now': client_now or f'{datetime.now().isoformat()}Z'
|
|
}
|
|
|
|
result = self._request.post(url, data, timeout=timeout, *args, **kwargs)
|
|
|
|
return result == 'ok'
|
|
|
|
def search(self,
|
|
text,
|
|
nocorrect=False,
|
|
type='all',
|
|
page=0,
|
|
playlist_in_best=True,
|
|
timeout=None,
|
|
*args, **kwargs):
|
|
url = f'{self.base_url}/search'
|
|
|
|
params = {
|
|
'text': text,
|
|
'nocorrect': nocorrect,
|
|
'type': type,
|
|
'page': page,
|
|
'playlist-in-best': playlist_in_best,
|
|
}
|
|
|
|
result = self._request.get(url, params, timeout=timeout, *args, **kwargs)
|
|
|
|
return Search.de_json(result, self)
|
|
|
|
def search_suggest(self, part: str, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/search/suggest'
|
|
|
|
result = self._request.get(url, {'part': part}, timeout=timeout, *args, **kwargs)
|
|
|
|
return Suggestions.de_json(result, self)
|
|
|
|
def artists(self, artist_ids: list or int or str, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/artists'
|
|
|
|
result = self._request.post(url, {'artist-ids': artist_ids}, timeout=timeout, *args, **kwargs)
|
|
|
|
return Artist.de_list(result, self)
|
|
|
|
def albums(self, album_ids: list or int or str, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/albums'
|
|
|
|
result = self._request.post(url, {'album-ids': album_ids}, timeout=timeout, *args, **kwargs)
|
|
|
|
return Album.de_list(result, self)
|
|
|
|
def tracks(self, track_ids: int or str, with_positions=True, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/tracks'
|
|
|
|
result = self._request.post(url, {'track-ids': track_ids, 'with-positions': with_positions},
|
|
timeout=timeout, *args, **kwargs)
|
|
|
|
return Track.de_list(result, self)
|
|
|
|
def playlists_list(self, playlist_ids: list or int or str, timeout=None, *args, **kwargs):
|
|
url = f'{self.base_url}/playlists/list'
|
|
|
|
result = self._request.post(url, {'playlistIds': playlist_ids}, timeout=timeout, *args, **kwargs)
|
|
|
|
return Playlist.de_list(result, self)
|
|
|
|
def users_playlists_list(self, user_id: int or str = None, timeout=None, *args, **kwargs):
|
|
if user_id is None:
|
|
user_id = self.account.uid
|
|
|
|
url = f'{self.base_url}/users/{user_id}/playlists/list'
|
|
|
|
result = self._request.get(url, timeout=timeout, *args, **kwargs)
|
|
|
|
return Playlist.de_list(result, self)
|
|
|
|
def users_likes_tracks(self, user_id: int or str = None, if_modified_since_revision=0, timeout=None, *args, **kwargs):
|
|
if user_id is None:
|
|
user_id = self.account.uid
|
|
|
|
url = f'{self.base_url}/users/{user_id}/likes/tracks'
|
|
|
|
params = {
|
|
'if-modified-since-revision': if_modified_since_revision
|
|
}
|
|
|
|
result = self._request.get(url, params, timeout=timeout, *args, **kwargs).get('library')
|
|
|
|
return TracksLikes.de_json(result, self)
|
|
|
|
def users_likes_albums(self, user_id: int or str = None, rich=True, timeout=None, *args, **kwargs):
|
|
if user_id is None:
|
|
user_id = self.account.uid
|
|
|
|
url = f'{self.base_url}/users/{user_id}/likes/albums'
|
|
|
|
result = self._request.get(url, {'rich': rich}, timeout=timeout, *args, **kwargs)
|
|
|
|
return AlbumsLikes.de_list(result, self)
|
|
|
|
def users_likes_artists(self, user_id: int or str = None, with_timestamps=True, timeout=None, *args, **kwargs):
|
|
if user_id is None:
|
|
user_id = self.account.uid
|
|
|
|
url = f'{self.base_url}/users/{user_id}/likes/artists'
|
|
|
|
params = {
|
|
'with-timestamps': with_timestamps
|
|
}
|
|
|
|
result = self._request.get(url, params, timeout=timeout, *args, **kwargs)
|
|
|
|
return ArtistsLikes.de_list(result, self)
|
|
|
|
def users_likes_playlists(self, user_id: int or str = None, timeout=None, *args, **kwargs):
|
|
if user_id is None:
|
|
user_id = self.account.uid
|
|
|
|
url = f'{self.base_url}/users/{user_id}/likes/playlists'
|
|
|
|
result = self._request.get(url, timeout=timeout, *args, **kwargs)
|
|
|
|
return PlaylistsLikes.de_list(result, self)
|