Implemented logErrors setting + some fixes
This commit is contained in:
parent
dfb29b2397
commit
6e18b51ac0
16
README.md
16
README.md
|
@ -3,6 +3,16 @@ Install the dependencies using `pip install -r requirements.txt`<br>
|
||||||
Run `python -m deemix --help` to see how to use the app
|
Run `python -m deemix --help` to see how to use the app
|
||||||
|
|
||||||
# TODO
|
# TODO
|
||||||
Making the download work in multi-threading<br>
|
Finish porting all features:
|
||||||
Finish porting all features<br>
|
- logging
|
||||||
Make the GUI after all is implemented
|
- ?
|
||||||
|
|
||||||
|
Settings not yet implemented:
|
||||||
|
- fallbackBitrate (on by default)
|
||||||
|
- fallbackSearch (off by default)
|
||||||
|
- logSearched
|
||||||
|
- savePlaylistAsCompilation
|
||||||
|
- removeAlbumVersion
|
||||||
|
- moveFeaturedToTitle
|
||||||
|
- titleCasing
|
||||||
|
- artistCasing
|
||||||
|
|
|
@ -8,7 +8,6 @@ from requests import get
|
||||||
from requests.exceptions import HTTPError
|
from requests.exceptions import HTTPError
|
||||||
from tempfile import gettempdir
|
from tempfile import gettempdir
|
||||||
from concurrent.futures import ThreadPoolExecutor
|
from concurrent.futures import ThreadPoolExecutor
|
||||||
import json
|
|
||||||
|
|
||||||
TEMPDIR = os.path.join(gettempdir(), 'deezloader-imgs')
|
TEMPDIR = os.path.join(gettempdir(), 'deezloader-imgs')
|
||||||
if not os.path.isdir(TEMPDIR):
|
if not os.path.isdir(TEMPDIR):
|
||||||
|
@ -206,12 +205,13 @@ def getTrackData(dz, trackAPI_gw, trackAPI = None, albumAPI_gw = None, albumAPI
|
||||||
except APIError:
|
except APIError:
|
||||||
if not albumAPI_gw:
|
if not albumAPI_gw:
|
||||||
albumAPI_gw = dz.get_album_gw(track['album']['id'])
|
albumAPI_gw = dz.get_album_gw(track['album']['id'])
|
||||||
track['album']['artist'] = {
|
track['album']['mainArtist'] = {
|
||||||
'id': albumAPI_gw['ART_ID'],
|
'id': albumAPI_gw['ART_ID'],
|
||||||
'name': albumAPI_gw['ART_NAME']
|
'name': albumAPI_gw['ART_NAME']
|
||||||
}
|
}
|
||||||
artistAPI = dz.get_artist(track['album']['artist']['id'])
|
artistAPI = dz.get_artist(track['album']['mainArtist']['id'])
|
||||||
track['album']['artist']['pic'] = artistAPI['picture_small'][artistAPI['picture_small'].find('artist/')+7:-24]
|
track['album']['artists'] = albumAPI_gw['ART_NAME']
|
||||||
|
track['album']['mainArtist']['pic'] = artistAPI['picture_small'][artistAPI['picture_small'].find('artist/')+7:-24]
|
||||||
track['album']['trackTotal'] = albumAPI_gw['NUMBER_TRACK']
|
track['album']['trackTotal'] = albumAPI_gw['NUMBER_TRACK']
|
||||||
track['album']['discTotal'] = albumAPI_gw['NUMBER_DISK']
|
track['album']['discTotal'] = albumAPI_gw['NUMBER_DISK']
|
||||||
track['album']['recordType'] = "Album"
|
track['album']['recordType'] = "Album"
|
||||||
|
@ -278,8 +278,8 @@ def getTrackData(dz, trackAPI_gw, trackAPI = None, albumAPI_gw = None, albumAPI
|
||||||
|
|
||||||
# Create artists strings
|
# Create artists strings
|
||||||
track['mainArtistsString'] = ""
|
track['mainArtistsString'] = ""
|
||||||
|
if 'Main' in track['artist']:
|
||||||
tot = len(track['artist']['Main'])
|
tot = len(track['artist']['Main'])
|
||||||
if tot > 0:
|
|
||||||
for i, art in enumerate(track['artist']['Main']):
|
for i, art in enumerate(track['artist']['Main']):
|
||||||
track['mainArtistsString'] += art
|
track['mainArtistsString'] += art
|
||||||
if tot != i+1:
|
if tot != i+1:
|
||||||
|
@ -289,8 +289,8 @@ def getTrackData(dz, trackAPI_gw, trackAPI = None, albumAPI_gw = None, albumAPI
|
||||||
track['mainArtistsString'] += ", "
|
track['mainArtistsString'] += ", "
|
||||||
else:
|
else:
|
||||||
track['mainArtistsString'] = track['mainArtist']['name']
|
track['mainArtistsString'] = track['mainArtist']['name']
|
||||||
|
if 'Featured' in track['artist']:
|
||||||
tot = len(track['artist']['Featured'])
|
tot = len(track['artist']['Featured'])
|
||||||
if tot > 0:
|
|
||||||
track['featArtistsString'] = "feat. "
|
track['featArtistsString'] = "feat. "
|
||||||
for i, art in enumerate(track['artist']['Featured']):
|
for i, art in enumerate(track['artist']['Featured']):
|
||||||
track['featArtistsString'] += art
|
track['featArtistsString'] += art
|
||||||
|
@ -303,7 +303,7 @@ def getTrackData(dz, trackAPI_gw, trackAPI = None, albumAPI_gw = None, albumAPI
|
||||||
# Create title with feat
|
# Create title with feat
|
||||||
if "(feat." in track['title'].lower():
|
if "(feat." in track['title'].lower():
|
||||||
track['title_feat'] = track['title']
|
track['title_feat'] = track['title']
|
||||||
elif len(artists['Featured'])>0:
|
elif 'Featured' in track['artist']:
|
||||||
track['title_feat'] = track['title']+" ({})".format(track['featArtistsString'])
|
track['title_feat'] = track['title']+" ({})".format(track['featArtistsString'])
|
||||||
else:
|
else:
|
||||||
track['title_feat'] = track['title']
|
track['title_feat'] = track['title']
|
||||||
|
@ -332,6 +332,10 @@ def downloadTrackObj(dz, trackAPI, settings, overwriteBitrate=False, extraTrack=
|
||||||
return downloadTrackObj(dz, trackAPI, settings, extraTrack=track)
|
return downloadTrackObj(dz, trackAPI, settings, extraTrack=track)
|
||||||
else:
|
else:
|
||||||
print("ERROR: Track not yet encoded!")
|
print("ERROR: Track not yet encoded!")
|
||||||
|
result['error'] = {
|
||||||
|
'message': "Track not yet encoded!",
|
||||||
|
'data': track
|
||||||
|
}
|
||||||
return result
|
return result
|
||||||
|
|
||||||
# Get the selected bitrate
|
# Get the selected bitrate
|
||||||
|
@ -378,7 +382,9 @@ def downloadTrackObj(dz, trackAPI, settings, overwriteBitrate=False, extraTrack=
|
||||||
# Generate artist tag if needed
|
# Generate artist tag if needed
|
||||||
if settings['multitagSeparator'] != "default":
|
if settings['multitagSeparator'] != "default":
|
||||||
if settings['multitagSeparator'] == "andFeat":
|
if settings['multitagSeparator'] == "andFeat":
|
||||||
track['artistsString'] = track['mainArtistsString'] + " " + track['featArtistsString']
|
track['artistsString'] = track['mainArtistsString']
|
||||||
|
if 'featArtistsString' in track:
|
||||||
|
track['artistsString'] += " "+track['featArtistsString']
|
||||||
else:
|
else:
|
||||||
track['artistsString'] = settings['multitagSeparator'].join(track[artists])
|
track['artistsString'] = settings['multitagSeparator'].join(track[artists])
|
||||||
|
|
||||||
|
@ -401,6 +407,10 @@ def downloadTrackObj(dz, trackAPI, settings, overwriteBitrate=False, extraTrack=
|
||||||
return downloadTrackObj(dz, trackAPI, settings, extraTrack=track)
|
return downloadTrackObj(dz, trackAPI, settings, extraTrack=track)
|
||||||
else:
|
else:
|
||||||
print("ERROR: Track not available on deezer's servers!")
|
print("ERROR: Track not available on deezer's servers!")
|
||||||
|
result['error'] = {
|
||||||
|
'message': "Track not available on deezer's servers!",
|
||||||
|
'data': track
|
||||||
|
}
|
||||||
return result
|
return result
|
||||||
if track['selectedFormat'] in [3, 1, 8]:
|
if track['selectedFormat'] in [3, 1, 8]:
|
||||||
tagID3(writepath, track, settings['tags'], settings['saveID3v1'], settings['useNullSeparator'])
|
tagID3(writepath, track, settings['tags'], settings['saveID3v1'], settings['useNullSeparator'])
|
||||||
|
@ -409,6 +419,33 @@ def downloadTrackObj(dz, trackAPI, settings, overwriteBitrate=False, extraTrack=
|
||||||
print("Done!")
|
print("Done!")
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
def after_download(tracks, settings):
|
||||||
|
extrasPath = None
|
||||||
|
playlist = [None] * len(tracks)
|
||||||
|
errors = ""
|
||||||
|
for index in range(len(tracks)):
|
||||||
|
result = tracks[index].result()
|
||||||
|
if 'error' in result:
|
||||||
|
errors += f"{result['error']['data']['id']} | {result['error']['data']['mainArtist']['name']} - {result['error']['data']['title']} | {result['error']['message']}\r\n"
|
||||||
|
if not extrasPath and 'extrasPath' in result:
|
||||||
|
extrasPath = result['extrasPath']
|
||||||
|
if settings['saveArtwork'] and result['albumPath']:
|
||||||
|
downloadImage(result['albumURL'], result['albumPath'])
|
||||||
|
if settings['saveArtworkArtist'] and result['artistPath']:
|
||||||
|
downloadImage(result['artistURL'], result['artistPath'])
|
||||||
|
if 'playlistPosition' in result:
|
||||||
|
playlist[index] = result['playlistPosition']
|
||||||
|
else:
|
||||||
|
playlist[index] = ""
|
||||||
|
if settings['logErrors'] and extrasPath and errors != "":
|
||||||
|
with open(os.path.join(extrasPath, 'errors.txt'), 'w') as f:
|
||||||
|
f.write(errors)
|
||||||
|
if settings['createM3U8File'] and extrasPath:
|
||||||
|
with open(os.path.join(extrasPath, 'playlist.m3u8'), 'w') as f:
|
||||||
|
for line in playlist:
|
||||||
|
f.write(line+"\n")
|
||||||
|
return extrasPath
|
||||||
|
|
||||||
def download_track(dz, id, settings, overwriteBitrate=False):
|
def download_track(dz, id, settings, overwriteBitrate=False):
|
||||||
trackAPI = dz.get_track_gw(id)
|
trackAPI = dz.get_track_gw(id)
|
||||||
trackAPI['FILENAME_TEMPLATE'] = settings['tracknameTemplate']
|
trackAPI['FILENAME_TEMPLATE'] = settings['tracknameTemplate']
|
||||||
|
@ -437,24 +474,7 @@ def download_album(dz, id, settings, overwriteBitrate=False):
|
||||||
trackAPI['FILENAME_TEMPLATE'] = settings['albumTracknameTemplate']
|
trackAPI['FILENAME_TEMPLATE'] = settings['albumTracknameTemplate']
|
||||||
playlist[pos-1] = executor.submit(downloadTrackObj, dz, trackAPI, settings, overwriteBitrate)
|
playlist[pos-1] = executor.submit(downloadTrackObj, dz, trackAPI, settings, overwriteBitrate)
|
||||||
|
|
||||||
extrasPath = None
|
return after_download(playlist, settings)
|
||||||
for index in range(len(playlist)):
|
|
||||||
result = playlist[index].result()
|
|
||||||
if not extrasPath:
|
|
||||||
extrasPath = result['extrasPath']
|
|
||||||
if settings['saveArtwork'] and result['albumPath']:
|
|
||||||
downloadImage(result['albumURL'], result['albumPath'])
|
|
||||||
if settings['saveArtworkArtist'] and result['artistPath']:
|
|
||||||
downloadImage(result['artistURL'], result['artistPath'])
|
|
||||||
if 'playlistPosition' in result:
|
|
||||||
playlist[index] = result['playlistPosition']
|
|
||||||
else:
|
|
||||||
playlist[index] = ""
|
|
||||||
if settings['createM3U8File'] and extrasPath:
|
|
||||||
with open(os.path.join(extrasPath, 'playlist.m3u8'), 'w') as f:
|
|
||||||
for line in playlist:
|
|
||||||
f.write(line+"\n")
|
|
||||||
return extrasPath
|
|
||||||
|
|
||||||
def download_artist(dz, id, settings, overwriteBitrate=False):
|
def download_artist(dz, id, settings, overwriteBitrate=False):
|
||||||
artistAPI = dz.get_artist_albums(id)
|
artistAPI = dz.get_artist_albums(id)
|
||||||
|
@ -472,22 +492,4 @@ def download_playlist(dz, id, settings, overwriteBitrate=False):
|
||||||
trackAPI['POSITION'] = pos
|
trackAPI['POSITION'] = pos
|
||||||
trackAPI['FILENAME_TEMPLATE'] = settings['playlistTracknameTemplate']
|
trackAPI['FILENAME_TEMPLATE'] = settings['playlistTracknameTemplate']
|
||||||
playlist[pos-1] = executor.submit(downloadTrackObj, dz, trackAPI, settings, overwriteBitrate)
|
playlist[pos-1] = executor.submit(downloadTrackObj, dz, trackAPI, settings, overwriteBitrate)
|
||||||
|
return after_download(playlist, settings)
|
||||||
extrasPath = None
|
|
||||||
for index in range(len(playlist)):
|
|
||||||
result = playlist[index].result()
|
|
||||||
if not extrasPath:
|
|
||||||
extrasPath = result['extrasPath']
|
|
||||||
if settings['saveArtwork'] and result['albumPath']:
|
|
||||||
downloadImage(result['albumURL'], result['albumPath'])
|
|
||||||
if settings['saveArtworkArtist'] and result['artistPath']:
|
|
||||||
downloadImage(result['artistURL'], result['artistPath'])
|
|
||||||
if 'playlistPosition' in result:
|
|
||||||
playlist[index] = result['playlistPosition']
|
|
||||||
else:
|
|
||||||
playlist[index] = ""
|
|
||||||
if settings['createM3U8File'] and extrasPath:
|
|
||||||
with open(os.path.join(extrasPath, 'playlist.m3u8'), 'w') as f:
|
|
||||||
for line in playlist:
|
|
||||||
f.write(line+"\n")
|
|
||||||
return extrasPath
|
|
||||||
|
|
Loading…
Reference in New Issue