Compare commits

..

No commits in common. "a4bf0f69c08d33d271fae8d237e85a7e747e7258" and "5c3e459cfebb5032fa413879e8e062274285315a" have entirely different histories.

5 changed files with 19 additions and 63 deletions

View file

@ -7,17 +7,15 @@ from loguru import logger
fzf = FzfPrompt() fzf = FzfPrompt()
@logger.catch @logger.catch
def list_albums(albums=None, pg=None, search=None, artist=None, include_channels=None, refresh=False): def list_albums(albums=None, pg=None, search=None, artist=None, include_channels=None):
albums_next = None albums_next = None
albums_prev = None albums_prev = None
play_artist_albums = False play_artist_albums = False
if not albums: if not albums:
albums = get_albums(q=search, artist=artist, include_channels=include_channels, refresh=refresh, pg=pg) albums = get_albums(q=search, artist=artist, include_channels=include_channels, pg=pg)
albums_next = albums.get('next') albums_next = albums.get('next')
albums_prev = albums.get('previous') albums_prev = albums.get('previous')
albums_results = albums.get('results') albums_results = albums.get('results')
if artist:
play_artist_albums = True
else: else:
play_artist_albums = True play_artist_albums = True
albums_results = albums albums_results = albums
@ -42,9 +40,6 @@ def list_albums(albums=None, pg=None, search=None, artist=None, include_channels
print('Search by albums: ') print('Search by albums: ')
list_albums(search=input()) list_albums(search=input())
elif select == 'Play all': elif select == 'Play all':
if artist:
src.fw_artists.play_artist(artist)
else:
src.fw_artists.play_artist(albums_results[0].get('artist')) src.fw_artists.play_artist(albums_results[0].get('artist'))
else: else:
play_album(album_id=albums_results[int(select)].get('id')) play_album(album_id=albums_results[int(select)].get('id'))

View file

@ -1,27 +1,9 @@
from src.mpv_control import set_http_header from src.mpv_control import set_http_header
import requests, json, time import requests, json, time
import os
from loguru import logger from loguru import logger
auth_file = '.auth.json'
if os.path.exists(auth_file):
with open('.auth.json', 'rt') as f: with open('.auth.json', 'rt') as f:
auth = json.loads(f.read()) auth = json.loads(f.read())
else:
# The default umask is 0o22 which turns off write permission of group and others
os.umask(0)
descriptor = os.open(
path=auth_file,
flags=(
os.O_WRONLY # access mode: write only
| os.O_CREAT # create if not exists
| os.O_TRUNC # truncate the file to zero
),
mode=0o600)
with open(descriptor, 'wt') as f:
f.write('{}')
auth = {}
s = requests.Session() s = requests.Session()
instance = 'fw.ponychord.rocks' instance = 'fw.ponychord.rocks'
@ -42,7 +24,7 @@ else:
def select_instance(new_instance=None): def select_instance(new_instance=None):
global instance global instance
instance = new_instance instance = new_instance
with open(auth_file, 'rt') as f: with open('.auth.json', 'rt') as f:
auth = json.loads(f.read()) auth = json.loads(f.read())
new_token = auth.get(instance) new_token = auth.get(instance)
s.headers.update({"Authorization": None, s.headers.update({"Authorization": None,
@ -81,15 +63,14 @@ def get_tracks(page=None, q=None, artist=None, album=None, favourites=None, incl
@logger.catch @logger.catch
def get_artists(page=None, q=None, artist=None, album=None, favourites=None, refresh=False, pg=None): def get_artists(page=None, q=None, artist=None, album=None, favourites=None, pg=None):
'''This function get artists by params''' '''This function get artists by params'''
params = { params = {
'page': page, 'page': page,
'q': q, 'q': q,
'artist': artist, 'artist': artist,
'album': album, 'album': album,
'favourites': favourites, 'favourites': favourites
'refresh': refresh
} }
if pg: if pg:
r = s.get(pg) r = s.get(pg)
@ -99,14 +80,13 @@ def get_artists(page=None, q=None, artist=None, album=None, favourites=None, ref
@logger.catch @logger.catch
def get_albums(page=None, q=None, artist=None, include_channels=None, refresh=False, pg=None): def get_albums(page=None, q=None, artist=None, include_channels=None, pg=None):
'''This function get artists by params''' '''This function get artists by params'''
params = { params = {
'page': page, 'page': page,
'q': q, 'q': q,
'artist': artist, 'artist': artist,
'include_channels': include_channels, 'include_channels': include_channels
'refresh': refresh
} }
if pg: if pg:
r = s.get(pg) r = s.get(pg)

View file

@ -35,21 +35,11 @@ def list_artists(pg=None, search=None):
def play_artist(artist_id): def play_artist(artist_id):
tracks = get_tracks(artist=artist_id, include_channels=True, pg=None) tracks = get_tracks(artist=artist_id)
tracks_next = tracks.get('next')
tracks_count = tracks.get('count')
storage = {}
if tracks_count > 50:
print(f'Loading {tracks_count} tracks...')
while True:
tracks_results = tracks.get('results') tracks_results = tracks.get('results')
tracks_next = tracks.get('next') storage = {}
for i in tracks_results: for i in tracks_results:
listen_url = concatinate_endpoint(i.get('listen_url')) listen_url = concatinate_endpoint(i.get('listen_url'))
storage[listen_url] = i storage[listen_url] = i
player.loadfile(listen_url, 'append-play') player.loadfile(listen_url, 'append-play')
if tracks_next:
tracks = get_tracks(artist=artist_id, include_channels=True, pg=tracks_next)
else:
break
player_menu("Artist playing...", storage) player_menu("Artist playing...", storage)

View file

@ -31,7 +31,4 @@ def list_channels(pg=None, search=None):
print('Search by channel:') print('Search by channel:')
list_channels(search=input()) list_channels(search=input())
else: else:
refresh = False list_albums(artist=channels_results[int(select)].get('artist').get('id'), include_channels=True)
if channels_results[int(select)].get('artist').get('is_local') == False:
refresh = True
list_albums(artist=channels_results[int(select)].get('artist').get('id'), include_channels=True, refresh=refresh)

View file

@ -13,19 +13,13 @@ def mpv_log(loglevel, component, message):
elif loglevel == 'error': elif loglevel == 'error':
logger.error(f'{component} {message}') logger.error(f'{component} {message}')
player = mpv.MPV(log_handler=mpv_log) player = mpv.MPV(log_handler=mpv_log, ytdl=False,
prefetch_playlist=True)
list_options = dir(player)
if 'ytdl' in list_options:
player.ytdl = False
if 'prefetch_playlist' in list_options:
player.prefetch_playlist = True
def set_http_header(headers=[]): def set_http_header(headers=[]):
player.http_header_fields = headers player.http_header_fields = headers
@logger.catch @logger.catch
def player_menu(header=None, storage={}): def player_menu(header=None, storage={}):
while True: while True: