sanspie/akarpov
1
1
Fork 0
mirror of https://github.com/Alexander-D-Karpov/akarpov synced 2024-11-22 07:26:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

fixed music artist and albums relations

Browse Source
This commit is contained in:
Alexander Karpov 2023-12-17 16:36:30 +03:00
parent 18ba02caab
commit b16ec21486
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
akarpov/music/api/serializers.py
View File

@ -243,10 +243,16 @@ class FullAlbumSerializer(serializers.ModelSerializer):
@extend_schema_field(AuthorSerializer(many=True))
def get_artists(self, obj):
artists = []
qs = Author.objects.cache().filter(
songs__id__in=obj.songs.cache().all().values("id").distinct()
)
for artist in qs:
if artist not in artists:
artists.append(artist)
return AuthorSerializer(
Author.objects.cache().filter(
songs__id__in=obj.songs.cache().all().values("id")
),
artists,
many=True,
).data
@ -265,10 +271,17 @@ class FullAuthorSerializer(serializers.ModelSerializer):
@extend_schema_field(AlbumSerializer(many=True))
def get_albums(self, obj):
qs = Album.objects.cache().filter(
songs__id__in=obj.songs.cache().all().values("id").distinct()
)
albums = []
for album in qs:
# TODO: rewrite to filter
if album not in albums:
albums.append(album)
return AlbumSerializer(
Album.objects.cache().filter(
songs__id__in=obj.songs.cache().all().values("id")
),
albums,
many=True,
).data