mirror of
https://github.com/KevinMidboe/spotify-downloader.git
synced 2025-10-29 18:00:15 +00:00
587f907ed8
This refactoring includes:
- Two empty lines before each global function
- Using '{0} {1}'.format(str1, str2) instead of str1 + ' ' + str2
Sometimes this will make lines longer, sometimes shorter.
- Starting all comments with # + space + comment
- Make lines not longer than 80 characters in most cases
- Renaming some variables to make more sense
- Add some missing code like returns and Exceptions
Not included, but follows:
- Make some comments docstrings
- Rename all 'file' variables, only for Python 2
- Remove some way too verbose comments ;)
57 lines
1.7 KiB
Python
57 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)
|