138 lines
3.8 KiB
Python
138 lines
3.8 KiB
Python
import re
|
|
import string
|
|
from deezer import TrackFormats
|
|
import os
|
|
|
|
def generateReplayGainString(trackGain):
|
|
return "{0:.2f} dB".format((float(trackGain) + 18.4) * -1)
|
|
|
|
def getBitrateInt(txt):
|
|
txt = str(txt).lower()
|
|
if txt in ['flac', 'lossless', '9']:
|
|
return TrackFormats.FLAC
|
|
elif txt in ['mp3', '320', '3']:
|
|
return TrackFormats.MP3_320
|
|
elif txt in ['128', '1']:
|
|
return TrackFormats.MP3_128
|
|
elif txt in ['360', '360_hq', '15']:
|
|
return TrackFormats.MP4_RA3
|
|
elif txt in ['360_mq', '14']:
|
|
return TrackFormats.MP4_RA2
|
|
elif txt in ['360_lq', '13']:
|
|
return TrackFormats.MP4_RA1
|
|
else:
|
|
return None
|
|
|
|
|
|
def changeCase(str, type):
|
|
if type == "lower":
|
|
return str.lower()
|
|
elif type == "upper":
|
|
return str.upper()
|
|
elif type == "start":
|
|
return string.capwords(str)
|
|
elif type == "sentence":
|
|
return str.capitalize()
|
|
else:
|
|
return str
|
|
|
|
|
|
def removeFeatures(title):
|
|
clean = title
|
|
if "(feat." in clean.lower():
|
|
pos = clean.lower().find("(feat.")
|
|
tempTrack = clean[:pos]
|
|
if ")" in clean:
|
|
tempTrack += clean[clean.find(")", pos + 1) + 1:]
|
|
clean = tempTrack.strip()
|
|
return clean
|
|
|
|
|
|
def andCommaConcat(lst):
|
|
tot = len(lst)
|
|
result = ""
|
|
for i, art in enumerate(lst):
|
|
result += art
|
|
if tot != i + 1:
|
|
if tot - 1 == i + 1:
|
|
result += " & "
|
|
else:
|
|
result += ", "
|
|
return result
|
|
|
|
|
|
def getIDFromLink(link, type):
|
|
if '?' in link:
|
|
link = link[:link.find('?')]
|
|
if link.endswith("/"):
|
|
link = link[:-1]
|
|
|
|
if link.startswith("http") and 'open.spotify.com/' in link:
|
|
if type == "spotifyplaylist":
|
|
return link[link.find("/playlist/") + 10:]
|
|
if type == "spotifytrack":
|
|
return link[link.find("/track/") + 7:]
|
|
if type == "spotifyalbum":
|
|
return link[link.find("/album/") + 7:]
|
|
elif link.startswith("spotify:"):
|
|
if type == "spotifyplaylist":
|
|
return link[link.find("playlist:") + 9:]
|
|
if type == "spotifytrack":
|
|
return link[link.find("track:") + 6:]
|
|
if type == "spotifyalbum":
|
|
return link[link.find("album:") + 6:]
|
|
elif type == "artisttop":
|
|
return re.search(r"\/artist\/(\d+)\/top_track", link)[1]
|
|
elif type == "artistdiscography":
|
|
return re.search(r"\/artist\/(\d+)\/discography", link)[1]
|
|
else:
|
|
return link[link.rfind("/") + 1:]
|
|
|
|
|
|
def getTypeFromLink(link):
|
|
type = ''
|
|
if 'spotify' in link:
|
|
type = 'spotify'
|
|
if 'playlist' in link:
|
|
type += 'playlist'
|
|
elif 'track' in link:
|
|
type += 'track'
|
|
elif 'album' in link:
|
|
type += 'album'
|
|
elif 'deezer' in link:
|
|
if '/track' in link:
|
|
type = 'track'
|
|
elif '/playlist' in link:
|
|
type = 'playlist'
|
|
elif '/album' in link:
|
|
type = 'album'
|
|
elif re.search("\/artist\/(\d+)\/top_track", link):
|
|
type = 'artisttop'
|
|
elif re.search("\/artist\/(\d+)\/discography", link):
|
|
type = 'artistdiscography'
|
|
elif '/artist' in link:
|
|
type = 'artist'
|
|
return type
|
|
|
|
|
|
def uniqueArray(arr):
|
|
for iPrinc, namePrinc in enumerate(arr):
|
|
for iRest, nRest in enumerate(arr):
|
|
if iPrinc!=iRest and namePrinc.lower() in nRest.lower():
|
|
del arr[iRest]
|
|
return arr
|
|
|
|
def removeDuplicateArtists(artist, artists):
|
|
artists = uniqueArray(artists)
|
|
for role in artist.keys():
|
|
artist[role] = uniqueArray(artist[role])
|
|
return (artist, artists)
|
|
|
|
def checkFolder(folder):
|
|
try:
|
|
os.makedirs(folder, exist_ok=True)
|
|
except Exception as e:
|
|
print(str(e))
|
|
return False
|
|
return os.access(folder, os.W_OK)
|