This commit is contained in:
IlyaShurupov 2024-07-11 10:26:27 +03:00
parent 59a3be1108
commit 3744bd1c56
3 changed files with 57 additions and 9 deletions

20
main.py
View file

@ -12,9 +12,12 @@ def get_playlist_names(pl):
return [name for name, items in pl.items()]
def print_playlist_tracks(name):
def print_playlist_tracks(pl_id: int):
pls = list(playlists)
name = pls[pl_id]
pl = playlists[name]
print(name)
for track in playlists[name]:
for track in pl:
track_data = track['track']
track_name = track_data['name']
artists = get_artists_str(track_data['artists'])
@ -46,9 +49,11 @@ def print_stats():
def print_tracks():
track_idx = 0
for trackItem in user_tracks:
track = trackItem['track']
print(f" '{track['name']}' - {get_artists_str(track['artists'])}")
print(f" {track_idx}: '{track['name']}' by '{get_artists_str(track['artists'])}'")
track_idx += 1
class Interpreter(cmd.Cmd):
@ -57,13 +62,16 @@ class Interpreter(cmd.Cmd):
def do_playlists(self, arg):
"""prints user playlists"""
print("\n".join(get_playlist_names(playlists)))
pl_id = 0
for pl in get_playlist_names(playlists):
print(f"{pl_id} - {pl}")
pl_id += 1
def do_playlist_tracks(self, arg):
"""prints playlist [name]"""
try:
name = arg.split()[0]
print_playlist_tracks(name)
pl_id = int(arg.split()[0])
print_playlist_tracks(pl_id)
except IndexError:
print("Invalid input")