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 from yandex_music.utils.request import Request from yandex_music.error 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=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) status = Status.de_json(result, self) return status def settings(self, timeout=None, *args, **kwargs): url = f'{self.base_url}/settings' result = self._request.get(url, timeout=timeout, *args, **kwargs) settings = Settings.de_json(result, self) return settings def permission_alerts(self, timeout=None, *args, **kwargs): url = f'{self.base_url}/permission-alerts' result = self._request.get(url, timeout=timeout, *args, **kwargs) permission_alerts = PermissionAlerts.de_json(result, self) return permission_alerts def account_experiments(self, timeout=None, *args, **kwargs): url = f'{self.base_url}/account/experiments' result = self._request.get(url, timeout=timeout, *args, **kwargs) experiments = Experiments.de_json(result, self) return experiments 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) promo_code_status = PromoCodeStatus.de_json(result, self) return promo_code_status def feed(self, timeout=None, *args, **kwargs): url = f'{self.base_url}/feed' result = self._request.get(url, timeout=timeout, *args, **kwargs) feed = Feed.de_json(result, self) return feed 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) download_info = DownloadInfo.de_list(result, self) return download_info 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' } return self._request.post(url, data=data, timeout=timeout, *args, **kwargs) 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) artists = Artist.de_list(result, self) return artists 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) albums = Album.de_list(result, self) return albums 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) tracks = Track.de_list(result, self) return tracks 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) playlists = Playlist.de_list(result, self) return playlists 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) playlists = Playlist.de_list(result, self) return playlists 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?if-modified-since-revision={if_modified_since_revision}' result = self._request.get(url, timeout=timeout, *args, **kwargs).get('library') tracks_likes = TracksLikes.de_json(result, self) return tracks_likes 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?rich={rich}' result = self._request.get(url, timeout=timeout, *args, **kwargs) albums_likes = AlbumsLikes.de_list(result, self) return albums_likes 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?with-timestamps={with_timestamps}' result = self._request.get(url, timeout=timeout, *args, **kwargs) artists_likes = ArtistsLikes.de_list(result, self) return artists_likes 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) playlists_likes = PlaylistsLikes.de_list(result, self) return playlists_likes