More MusicBrainz refinement
Moved cover check inside the release check loop, and other things.
This commit is contained in:
markjfine
GitHub
parent
395bd5782c
commit
1bc01daf5b
146
nrsc5-dui.py
146
nrsc5-dui.py
@ -344,16 +344,65 @@ class NRSC5_DUI(object):
|
||||
result = group[arg]
|
||||
return result
|
||||
|
||||
def check_terms(self,inStr,terms):
|
||||
result = False
|
||||
for term in terms:
|
||||
result=(term in inStr)
|
||||
if result:
|
||||
break
|
||||
return result
|
||||
|
||||
def check_musicbrainz_cover(self,inID):
|
||||
result = False
|
||||
imageList = None
|
||||
|
||||
try:
|
||||
imageList = musicbrainzngs.get_image_list(inID)
|
||||
except:
|
||||
print("MusicBrainz image list retrieval error")
|
||||
#pass
|
||||
|
||||
if (imageList is not None) and ('images' in imageList):
|
||||
#print('has images')
|
||||
for (idx, image) in enumerate(imageList['images']):
|
||||
imgTypes = self.check_value('types', image, None)
|
||||
imgApproved = self.check_value('approved', image, "False")
|
||||
result = ('Front' in imgTypes) and imgApproved
|
||||
if (result):
|
||||
break
|
||||
return result
|
||||
|
||||
def save_musicbrainz_cover(self,inID,saveStr):
|
||||
imgData = None
|
||||
result = False
|
||||
|
||||
try:
|
||||
imgData = musicbrainzngs.get_image_front(inID, size="500")
|
||||
except:
|
||||
print("MusicBrainz image retrieval error")
|
||||
#pass
|
||||
|
||||
if (imgData is not None) and (len(imgData) > 0):
|
||||
dataBytes = io.BytesIO(imgData)
|
||||
imgCvr = Image.open(dataBytes)
|
||||
imgCvr.save(saveStr)
|
||||
result = True
|
||||
return result
|
||||
|
||||
def get_cover_image_online(self):
|
||||
global aasDir
|
||||
got_cover = False
|
||||
albumExclude = ['hitzone','now that’s what i call music']
|
||||
|
||||
# only care about the first artist listed if separated by slashes
|
||||
newArtist = self.fix_artist()
|
||||
newArtist = self.fix_artist().replace("'","’")
|
||||
newTitle = self.streamInfo["Title"].replace("'","’")
|
||||
baseStr = str(newArtist+" - "+self.streamInfo["Title"])
|
||||
saveStr = os.path.join(aasDir, baseStr.replace(" ","_").replace("/","_").replace(":","_")+".jpg")
|
||||
|
||||
if (newArtist=="") and (newTitle==""):
|
||||
return
|
||||
|
||||
# does it already exist?
|
||||
if (os.path.isfile(saveStr)):
|
||||
self.coverImage = saveStr
|
||||
@ -365,14 +414,14 @@ class NRSC5_DUI(object):
|
||||
result = None
|
||||
|
||||
#print()
|
||||
#print("searching for {} - {}".format(self.streamInfo["Artist"],newTitle))
|
||||
#print("searching for {} - {}".format(newArtist,newTitle))
|
||||
|
||||
try:
|
||||
result = musicbrainzngs.search_recordings(artist=self.streamInfo["Artist"], recording=newTitle)
|
||||
result = musicbrainzngs.search_recordings(strict=True, artist= newArtist, recording=newTitle, type='Album', status='Official')
|
||||
#print("recording search succeeded")
|
||||
except:
|
||||
#print("mb recording search error")
|
||||
pass
|
||||
print("MusicBrainz recording search error")
|
||||
#pass
|
||||
|
||||
if (result is not None) and ('recording-list' in result):
|
||||
#print("got recording search result")
|
||||
@ -380,83 +429,66 @@ class NRSC5_DUI(object):
|
||||
# loop through the list until you get a match
|
||||
for (idx, release) in enumerate(result['recording-list']):
|
||||
#print(release)
|
||||
resultid = self.check_value('id',release,"")
|
||||
resultID = self.check_value('id',release,"")
|
||||
resultScore = self.check_value('ext:score',release,"0")
|
||||
resultArtist = self.check_value('artist-credit-phrase',release,"")
|
||||
resultTitle = self.check_value('title',release,"")
|
||||
scoreMatch = (int(resultScore) > 90)
|
||||
scoreMatch = (int(resultScore) > 85)
|
||||
artistMatch = (newArtist.lower() in resultArtist.lower())
|
||||
titleMatch = (newTitle.lower() in resultTitle.lower())
|
||||
if ('release-list' in release):
|
||||
recordingMatch = (artistMatch and titleMatch and scoreMatch)
|
||||
|
||||
# don't bother dealing with releases if artist, title and score don't match
|
||||
resultStatus = ""
|
||||
resultType = ""
|
||||
resultAlbum = ""
|
||||
resultArtist2 = ""
|
||||
releaseMatch = False
|
||||
imageMatch = False
|
||||
if recordingMatch and ('release-list' in release):
|
||||
#print("got release-list")
|
||||
for (idx2, release2) in enumerate(release['release-list']):
|
||||
#print(release2)
|
||||
resultid = self.check_value('id',release2,"")
|
||||
resultStatus = self.check_value('status',release2,"")
|
||||
imageMatch = False
|
||||
resultID = self.check_value('id',release2,"")
|
||||
resultStatus = self.check_value('status',release2,"Official")
|
||||
resultType = self.check_value('type',self.check_value('release-group',release2,""),"")
|
||||
resultAlbum = self.check_value('title',release2,"")
|
||||
resultArtist2 = self.check_value('artist-credit-phrase',release2,"")
|
||||
typeMatch = (resultType in ['Single','Album','EP'])
|
||||
statusMatch = (resultStatus == 'Official')
|
||||
albumMatch = (not self.check_terms(resultAlbum, albumExclude))
|
||||
artistMatch2 = not ('Various' in resultArtist2)
|
||||
releaseMatch = (artistMatch2 and albumMatch and typeMatch and statusMatch)
|
||||
# don't bother checking for covers unless album, type, and status match
|
||||
if releaseMatch:
|
||||
imageMatch = self.check_musicbrainz_cover(resultID)
|
||||
#print(" release-list: #{} of {}".format(idx2, len(release['release-list'])))
|
||||
if (typeMatch and statusMatch and ((idx2+1) < len(release['release-list']))):
|
||||
if (releaseMatch and imageMatch and ((idx2+1) < len(release['release-list']))):
|
||||
break
|
||||
#print("#{} {} {}: {} - {} - {}, {} {}%".format(idx, resultStatus, resultType, resultArtist, resultTitle, resultAlbum, resultid, resultScore))
|
||||
if (artistMatch and titleMatch and scoreMatch and typeMatch and statusMatch):
|
||||
resultID=resultid
|
||||
#print("#{} {}: Track: {} - {}, {}: {} - {}, {} {}%".format(idx, resultStatus, resultArtist, resultTitle, resultType, resultArtist2, resultAlbum, resultID, resultScore))
|
||||
|
||||
# got a match, now get the cover art
|
||||
#print("Found {}: {} - {} - {}, {} {}%".format(resultType, resultArtist, resultTitle, resultAlbum, resultID, resultScore))
|
||||
imageList = None
|
||||
if (recordingMatch and releaseMatch and imageMatch):
|
||||
|
||||
try:
|
||||
imageList = musicbrainzngs.get_image_list(resultID)
|
||||
except:
|
||||
#print("mb image error")
|
||||
pass
|
||||
# got a full match, now get the cover art
|
||||
#print("Found {}: Track: {} - {}, {}: {} - {}, {} {}%".format(resultStatus, resultArtist, resultTitle, resultType, resultArtist2, resultAlbum, resultID, resultScore))
|
||||
if self.save_musicbrainz_cover(resultID,saveStr):
|
||||
self.coverImage = saveStr
|
||||
imgSaved = True
|
||||
if (self.streamInfo['Album'] == ""):
|
||||
self.streamInfo['Album']=resultAlbum
|
||||
|
||||
if (imageList is not None) and ('images' in imageList):
|
||||
#print('has images')
|
||||
for (idx3, image) in enumerate(imageList['images']):
|
||||
imgTypes = self.check_value('types', image, None)
|
||||
imgApproved = self.check_value('approved', image, "False")
|
||||
#print("image: {} {}".format(imgTypes, imgApproved))
|
||||
if ('Front' in imgTypes) and imgApproved:
|
||||
|
||||
# now save it
|
||||
#print("Found approved image")
|
||||
imgData = None
|
||||
|
||||
try:
|
||||
imgData = musicbrainzngs.get_image_front(resultID, size="500")
|
||||
except:
|
||||
#print("mb image retrieval error")
|
||||
pass
|
||||
|
||||
#if (len(imgData) > 0):
|
||||
if (imgData is not None) and (len(imgData) > 0):
|
||||
dataBytes = io.BytesIO(imgData)
|
||||
imgCvr = Image.open(dataBytes)
|
||||
imgCvr.save(saveStr)
|
||||
self.coverImage = saveStr
|
||||
imgSaved = True
|
||||
|
||||
#print("image-list: #{} of {}".format(idx3, len(imageList['images'])))
|
||||
if (imgSaved) and ((idx3+1) < len(imageList['images'])):
|
||||
break
|
||||
#print("End of image-list")
|
||||
|
||||
#print("recording-list: #{} of {}".format(idx, len(result['recording-list'])))
|
||||
if (imgSaved) and ((idx+1) < len(result['recording-list'])) or (not scoreMatch):
|
||||
break
|
||||
#print("End of recording-list")
|
||||
|
||||
# If no match use the station logo if there is one
|
||||
if (not imgSaved):
|
||||
#print("No image found, using logo")
|
||||
self.coverImage = os.path.join(aasDir, self.stationLogos[self.stationStr][self.streamNum])
|
||||
self.streamInfo['Album']=""
|
||||
except:
|
||||
#print("general error")
|
||||
pass
|
||||
print("general error in the musicbrainz routine")
|
||||
#pass
|
||||
|
||||
# now display it by simulating a window resize
|
||||
self.on_cover_resize(self.mainWindow)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user