yetangitu/funkwhale
1
0
Fork 0
mirror of https://code.eliotberriot.com/funkwhale/funkwhale.git synced 2025-10-04 19:29:20 +02:00
Code Issues Releases Wiki Activity

See #432: added model tagging logic on artists and albums

Browse source
This commit is contained in:
Eliot Berriot 2019-07-15 10:52:57 +02:00
parent 426f6f0d45
commit c8135d38b1
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
5 changed files with 36 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

23
api/funkwhale_api/music/factories.py
View file

@ -6,7 +6,7 @@ from funkwhale_api.factories import registry, NoUpdateOnCreate
from funkwhale_api.federation import factories as federation_factories
from funkwhale_api.music import licenses
from funkwhale_api.tags import models as tags_models
from funkwhale_api.tags import factories as tags_factories
from funkwhale_api.users import factories as users_factories
SAMPLES_PATH = os.path.join(
@ -56,7 +56,9 @@ class LicenseFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
@registry.register
class ArtistFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
class ArtistFactory(
tags_factories.TaggableFactory, NoUpdateOnCreate, factory.django.DjangoModelFactory
):
name = factory.Faker("name")
mbid = factory.Faker("uuid4")
fid = factory.Faker("federation_url")
@ -73,7 +75,9 @@ class ArtistFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
@registry.register
class AlbumFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
class AlbumFactory(
tags_factories.TaggableFactory, NoUpdateOnCreate, factory.django.DjangoModelFactory
):
title = factory.Faker("sentence", nb_words=3)
mbid = factory.Faker("uuid4")
release_date = factory.Faker("date_object")
@ -97,7 +101,9 @@ class AlbumFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
@registry.register
class TrackFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
class TrackFactory(
tags_factories.TaggableFactory, NoUpdateOnCreate, factory.django.DjangoModelFactory
):
fid = factory.Faker("federation_url")
title = factory.Faker("sentence", nb_words=3)
mbid = factory.Faker("uuid4")
@ -127,15 +133,6 @@ class TrackFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):
self.license = LicenseFactory(code=extracted)
self.save()
@factory.post_generation
def set_tags(self, create, extracted, **kwargs):
if not create:
# Simple build, do nothing.
return
if extracted:
tags_models.set_tags(self, *extracted)
@registry.register
class UploadFactory(NoUpdateOnCreate, factory.django.DjangoModelFactory):