secstate/vrobbler
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix fetching release group and cover images

Browse Source
This commit is contained in:
Colin Powell
2023-02-08 20:00:09 -05:00
parent 8ba8ceefb8
commit 9f272df99c
1 changed files with 26 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
vrobbler/apps/music/models.py
View File

@ -72,11 +72,19 @@ class Album(TimeStampedModel):
return self.artists.first() return self.artists.first()
def fix_metadata(self): def fix_metadata(self):
if not self.musicbrainz_albumartist_id or not self.year: if (
not self.musicbrainz_albumartist_id
or not self.year
or not self.musicbrainz_releasegroup_id
):
musicbrainzngs.set_useragent('vrobbler', '0.3.0') musicbrainzngs.set_useragent('vrobbler', '0.3.0')
mb_data = musicbrainzngs.get_release_by_id( mb_data = musicbrainzngs.get_release_by_id(
self.musicbrainz_id, includes=['artists'] self.musicbrainz_id, includes=['artists', 'release-groups']
) )
if not self.musicbrainz_releasegroup_id:
self.musicbrainz_releasegroup_id = mb_data['release'][
'release-group'
]['id']
if not self.musicbrainz_albumartist_id: if not self.musicbrainz_albumartist_id:
self.musicbrainz_albumartist_id = mb_data['release'][ self.musicbrainz_albumartist_id = mb_data['release'][
'artist-credit' 'artist-credit'
@ -89,7 +97,13 @@ class Album(TimeStampedModel):
except IndexError: except IndexError:
pass pass
self.save(update_fields=['musicbrainz_albumartist_id', 'year']) self.save(
update_fields=[
'musicbrainz_albumartist_id',
'musicbrainz_releasegroup_id',
'year',
]
)
new_artist = Artist.objects.filter( new_artist = Artist.objects.filter(
musicbrainz_id=self.musicbrainz_albumartist_id musicbrainz_id=self.musicbrainz_albumartist_id
@ -99,6 +113,11 @@ class Album(TimeStampedModel):
if not new_artist: if not new_artist:
for t in self.track_set.all(): for t in self.track_set.all():
self.artists.add(t.artist) self.artists.add(t.artist)
if (
not self.cover_image
or self.cover_image == 'default-image-replace-me'
):
self.fetch_artwork()
def fetch_artwork(self, force=False): def fetch_artwork(self, force=False):
if not self.cover_image and not force: if not self.cover_image and not force:
@ -115,7 +134,10 @@ class Album(TimeStampedModel):
f'No cover art found for {self.name} by release' f'No cover art found for {self.name} by release'
) )
if not self.cover_image and self.musicbrainz_releasegroup_id: if (
not self.cover_image
or self.cover_image == "default-image-replace-me"
) and self.musicbrainz_releasegroup_id:
try: try:
img_data = musicbrainzngs.get_release_group_image_front( img_data = musicbrainzngs.get_release_group_image_front(
self.musicbrainz_releasegroup_id self.musicbrainz_releasegroup_id
@ -131,8 +153,6 @@ class Album(TimeStampedModel):
logger.debug( logger.debug(
f"No cover art found for release or release group for {self.name}, setting to default" f"No cover art found for release or release group for {self.name}, setting to default"
) )
# TODO Get a placeholder image in here
self.cover_image = 'default-image-replace-me'
self.save() self.save()
@property @property