mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-10-29 18:00:15 +00:00
0f384bb5ee
This comment will: - Transform docstrings above functions into docstrings - Remove some way too verbose comments - Apply some more recommendations from PEP8 forgotten last time - Fix some errors introduced with the first code cleanup Work left to do: - Add params to docstrings - Rename file variables
58 lines
1.7 KiB
Python
58 lines
1.7 KiB
Python
# -*- coding: UTF-8 -*-
|
|
|
|
import spotdl
|
|
|
|
username = 'alex'
|
|
|
|
|
|
def test_user():
|
|
expect_playlists = 7
|
|
playlists = spotdl.spotify.user_playlists(username)
|
|
playlists = len(playlists['items'])
|
|
assert playlists == expect_playlists
|
|
|
|
|
|
def test_playlist():
|
|
expect_tracks = 14
|
|
playlist = spotdl.spotify.user_playlists(username)['items'][0]
|
|
tracks = playlist['tracks']['total']
|
|
assert tracks == expect_tracks
|
|
|
|
|
|
def test_tracks():
|
|
playlist = spotdl.spotify.user_playlists(username)['items'][0]
|
|
expect_lines = playlist['tracks']['total']
|
|
result = spotdl.spotify.user_playlist(
|
|
playlist['owner']['id'], playlist['id'], fields='tracks,next')
|
|
tracks = result['tracks']
|
|
|
|
with open('list.txt', 'a') as fout:
|
|
while True:
|
|
for item in tracks['items']:
|
|
track = item['track']
|
|
try:
|
|
fout.write(track['external_urls']['spotify'] + '\n')
|
|
except KeyError:
|
|
title = track['name'] + ' by '+ track['artists'][0]['name']
|
|
print('Skipping track ' + title + ' (local only?)')
|
|
# 1 page = 50 results
|
|
# check if there are more pages
|
|
if tracks['next']:
|
|
tracks = spotify.next(tracks)
|
|
else:
|
|
break
|
|
|
|
with open('list.txt', 'r') as listed:
|
|
expect_song = (listed.read()).splitlines()[0]
|
|
|
|
spotdl.misc.trim_song('list.txt')
|
|
with open('list.txt', 'a') as myfile:
|
|
myfile.write(expect_song)
|
|
|
|
with open('list.txt', 'r') as listed:
|
|
songs = (listed.read()).splitlines()
|
|
|
|
lines = len(songs)
|
|
song = songs[-1]
|
|
assert (expect_lines == lines and expect_song == song)
|