def IdentifyTrack(self, term):
sFktname = "search_track"
# build search term. so connect artist and
# title, delete some characters and split on
# white spaces
search_term_list = term.strip()
search_term_list = search_term_list.replace("(","").replace(")","").lower()
search_term_list = search_term_list.split()
ePrint(2, sFktname, "looking for term: {}".format(search_term_list))
# to find the track, we constuct a ranking, with the term on the highest rank, wich
# made the most hits
max_hits = 0
match_list = []
for item in self.TrackList:
item_hits = 0
# create target term
search_target_list = item.Artist.lower() + " " + item.Title.lower()
search_target_list = ReplaceChars("/_()-.:,", " ", search_target_list)
search_target_list = search_target_list.split()
ePrint(2, sFktname, "Matching against: "+ str(search_target_list))
# match
for s in search_term_list:
if s in search_target_list:
item_hits += 1
# calculate a value wich is relative to the length of the search_term_list
item_hits = item_hits / len(search_term_list)
ePrint(2, sFktname, "Ranking points: " + str(item_hits))
# create a matchcount, item tupel
match_list.append((item_hits, item))
if len(match_list) == 0:
return None
# sort matchlist
match_list = sorted(match_list, key = lambda tup: tup[0], reverse = True)
return match_list[0][1]
def Run(self):
print()
filled_tracks = []
unfilled_tracks = []
##
## Handle Directories (Release Search)
##
if self.Settings.HasDirs():
# here we only link the files to specific dirs
# the next loop then cares about and first makes
# a search in a release cache
idx = 0
for folder in self.Settings.DirList:
#
# figure out the directory entries
#
file_list = os.listdir(folder)
if not file_list:
ePrint(1, self.__ClassName, "Skipping directory " + folder)
continue
ePrint(2, folder, "<-- Linking files")
ePrint(2, "Containing: ", str(file_list) + "\n")
# append them to the FileList
for file_path in file_list:
f = File(os.path.join(folder, file_path))
f.RelDir = folder
f.RelDirIdx = idx
#ePrint(2, "Fileending: ", str(file_list) + "\n")
if f.Type in ["mp3", "flac", "wav", "wave", "ogg"]:
self.Settings.FileList.append(f)
idx += 1
##
## Handle the FileList if existent
##
if self.Settings.HasFiles():
track_list = []
for f in self.Settings.FileList:
#
# at first try to gather some data from the tags, so
# create a track object
#
new_track = Track()
# associate the track with its file and try to read
# some info from the tags
new_track.FileInstance = f
new_track.FillFromFile()
ePrint(1, ColorString(str(new_track.FileInstance.NameBody), "green"), "<-- Try to determine info")
# if a track is fully filled continue with the next
if (new_track.FilledEnough(self.Settings.Pattern)):
ePrint(1, self.__ClassName, "Track is filled enough: " + str(new_track) + "\n")
filled_tracks.append(new_track)
continue
#
# Look in the cache
#
rel_page = None
for t in filled_tracks:
if new_track.FileInstance.RelDir != "" and \
new_track.FileInstance.RelDir == t.FileInstance.RelDir and \
t.Release:
ePrint(2, self.__ClassName, "Using cache")
rel_page = t.Release
break
if not rel_page:
#
# look at chemical if nothing is found in tags or cache
#
# first we check for an explicit searchterm
idx = new_track.FileInstance.RelDirIdx
if idx < len(self.Settings.SearchTermList):
search_term = self.Settings.SearchTermList[idx]
else:
# if we have artist and filename availible set this as searchterm
if new_track.Artist and new_track.Title:
search_term = new_track.Artist + " " + new_track.Title
else:
# otherwise create search_term from filename
search_term = new_track.FileInstance.NameBody
ePrint(2, self.__ClassName, "Setting searchterm: " + search_term)
# link it up on BeatportWeb
res_page = BeatportWeb.ResultPage(search_term)
# if we found nothing, we cut the searchterm
if not res_page.GetReleaseList():
# at first remove signs symbols and retry
search_term = ReplaceChars("/_()-.:,&?", " ", search_term)
# strip "feat" terms
p = re.compile(" feat ", re.IGNORECASE)
search_term = p.sub("", search_term)
res_page = BeatportWeb.ResultPage(search_term)
if not res_page.GetReleaseList():
# second remove numbers
search_term = ReplaceChars("0123456789", " ", search_term)
res_page = BeatportWeb.ResultPage(search_term)
# now strip words from the term
if not res_page.GetReleaseList():
search_term_word_list = search_term.split()
while len(search_term_word_list) > 1:
search_term_word_list.pop()
search_term = " ".join(search_term_word_list)
res_page = BeatportWeb.ResultPage(search_term)
if res_page.GetReleaseList():
break
# if we finally found nothing, skip
if not res_page.GetReleaseList():
ePrint(1, self.__ClassName, "Could not determine all fields for: " +
new_track.FileInstance.Basename + "\n")
unfilled_tracks.append(new_track)
continue
#
# now we process the result list
#
release_list = res_page.GetReleaseList()
# invoke the user if we have found more than one results
if len(release_list) > 1:
ePrint(1, self.__ClassName,
"Multiple possilble releases found for: " + new_track.FileInstance.Basename +
"Please type number. 0 to skip.")
# TODO print sorted
release_list = res_page.GetReleaseList()
# sort the release list for better choosing
release_list = sorted(release_list, key = lambda rel: str(rel), reverse = False)
c = 1
for r in release_list:
print("{:4d} : {!s:.150}".format(c, r))
c += 1
choice = -1
while choice < 0 or choice > len(release_list):
choice = int(input(" <-- "))
if choice == 0:
ePrint(1, self.__ClassName, "Skipping: " + new_track.FileInstance.Basename + "\n")
unfilled_tracks.append(new_track)
continue
# choose result, pay attention on index
release_candidate = release_list[choice - 1]
else:
release_candidate = release_list[0]
# create a ReleasePage, from the release candidate . This determines all relevant
# information for the release
rel_page = BeatportWeb.ReleasePage(release_candidate)
#
# identify the track in the release
#
# create a search term
if new_track.Artist and new_track.Title:
search_term = new_track.Artist + " " + new_track.Title
else:
search_term = new_track.FileInstance.NameBody
search_term = ReplaceChars("/_()-.:,", " ", search_term)
cor_track = rel_page.IdentifyTrack(search_term)
ePrint(1, self.__ClassName, "corresponding track: " + ColorString(str(cor_track), 'green'))
# since we only have a corresponding track copy on demand
if not new_track.Artist:
new_track.Artist = cor_track.Artist
if not new_track.Title:
new_track.Title = cor_track.Title
if not new_track.Key:
new_track.Key = cor_track.Key
if not new_track.Number:
new_track.Number = cor_track.Number
if new_track.FilledEnough(self.Settings.Pattern):
ePrint(1, self.__ClassName, "Track is filled enough: " + str(new_track))
# link to the found release
new_track.Release = rel_page
filled_tracks.append(new_track)
else:
ePrint(1, self.__ClassName, "Could not determine all fields for: " + new_track.FileInstance.Basename)
unfilled_tracks.append(new_track)
print()
#
# Rename files
#
for track in filled_tracks:
dest_name = self.Settings.Pattern.GetResolvedString(artist=track.Artist, title=track.Title,
number=track.Number, key=track.Key)
# function only needs the basename without extension
if self.Settings.MakeCopy:
#track.FileInstance.Copy(dest_name)
pass
else:
if self.Settings.Ask == True:
track.FileInstance.RenameQuery(dest_name)
else:
track.FileInstance.Rename(dest_name)
#
# handle dir renaming
#
handled_dirs = []
for track in filled_tracks:
if track.FileInstance.RelDir and track.FileInstance.RelDir not in handled_dirs:
#TODO remove code here
# build a artist string, maximum 2
#artists = ""
#c = 0
#for t in track.Release.TrackList:
# artists += t.Artist + " "
# if c >= 1:
# break
# c += 1
#print("Artist name: " + artists)
#
## create destination string
#artists = artists.strip()
#artists = ReplaceChars(" ", ".", artists)
#artists = artists.lower()
rel_name = track.Release.Name.strip()
rel_name = ReplaceChars("/'", "", rel_name)
rel_name = ReplaceChars(" ", ".", rel_name)
rel_name = rel_name.lower()
dest = track.Release.Catid.lower() + ".-." + rel_name
ePrint(2, self.__ClassName, track.FileInstance.RelDir)
ePrint(2, self.__ClassName, dest)
# download artwork if necessary
if self.Settings.GetImage == True:
image_path = os.path.join(track.FileInstance.RelDir, "artwork.jpg")
if os.path.exists(image_path):
ePrint(1, self.__ClassName, "Artwork already in release folder")
else:
ePrint(1, self.__ClassName, "Downloading Artwork")
DownloadFile(track.Release.PictureLink, image_path)
RenameDirQuery(track.FileInstance.RelDir, dest)
handled_dirs.append(track.FileInstance.RelDir)
