mirror of
https://code.eliotberriot.com/funkwhale/funkwhale.git
synced 2025-10-05 04:09:25 +02:00
Blacked the code
This commit is contained in:
parent
b6fc0051fa
commit
62ca3bd736
279 changed files with 8861 additions and 9527 deletions
|
@ -12,7 +12,6 @@ from .registries import registry
|
|||
|
||||
|
||||
class SimpleRadio(object):
|
||||
|
||||
def clean(self, instance):
|
||||
return
|
||||
|
||||
|
@ -37,13 +36,13 @@ class SessionRadio(SimpleRadio):
|
|||
self.session = session
|
||||
|
||||
def start_session(self, user, **kwargs):
|
||||
self.session = models.RadioSession.objects.create(user=user, radio_type=self.radio_type, **kwargs)
|
||||
self.session = models.RadioSession.objects.create(
|
||||
user=user, radio_type=self.radio_type, **kwargs
|
||||
)
|
||||
return self.session
|
||||
|
||||
def get_queryset(self, **kwargs):
|
||||
qs = Track.objects.annotate(
|
||||
files_count=Count('files')
|
||||
)
|
||||
qs = Track.objects.annotate(files_count=Count("files"))
|
||||
return qs.filter(files_count__gt=0)
|
||||
|
||||
def get_queryset_kwargs(self):
|
||||
|
@ -57,7 +56,9 @@ class SessionRadio(SimpleRadio):
|
|||
return queryset
|
||||
|
||||
def filter_from_session(self, queryset):
|
||||
already_played = self.session.session_tracks.all().values_list('track', flat=True)
|
||||
already_played = self.session.session_tracks.all().values_list(
|
||||
"track", flat=True
|
||||
)
|
||||
queryset = queryset.exclude(pk__in=already_played)
|
||||
return queryset
|
||||
|
||||
|
@ -76,60 +77,51 @@ class SessionRadio(SimpleRadio):
|
|||
return data
|
||||
|
||||
|
||||
@registry.register(name='random')
|
||||
@registry.register(name="random")
|
||||
class RandomRadio(SessionRadio):
|
||||
def get_queryset(self, **kwargs):
|
||||
qs = super().get_queryset(**kwargs)
|
||||
return qs.order_by('?')
|
||||
return qs.order_by("?")
|
||||
|
||||
|
||||
@registry.register(name='favorites')
|
||||
@registry.register(name="favorites")
|
||||
class FavoritesRadio(SessionRadio):
|
||||
|
||||
def get_queryset_kwargs(self):
|
||||
kwargs = super().get_queryset_kwargs()
|
||||
if self.session:
|
||||
kwargs['user'] = self.session.user
|
||||
kwargs["user"] = self.session.user
|
||||
return kwargs
|
||||
|
||||
def get_queryset(self, **kwargs):
|
||||
qs = super().get_queryset(**kwargs)
|
||||
track_ids = kwargs['user'].track_favorites.all().values_list('track', flat=True)
|
||||
track_ids = kwargs["user"].track_favorites.all().values_list("track", flat=True)
|
||||
return qs.filter(pk__in=track_ids)
|
||||
|
||||
|
||||
@registry.register(name='custom')
|
||||
@registry.register(name="custom")
|
||||
class CustomRadio(SessionRadio):
|
||||
|
||||
def get_queryset_kwargs(self):
|
||||
kwargs = super().get_queryset_kwargs()
|
||||
kwargs['user'] = self.session.user
|
||||
kwargs['custom_radio'] = self.session.custom_radio
|
||||
kwargs["user"] = self.session.user
|
||||
kwargs["custom_radio"] = self.session.custom_radio
|
||||
return kwargs
|
||||
|
||||
def get_queryset(self, **kwargs):
|
||||
qs = super().get_queryset(**kwargs)
|
||||
return filters.run(
|
||||
kwargs['custom_radio'].config,
|
||||
candidates=qs,
|
||||
)
|
||||
return filters.run(kwargs["custom_radio"].config, candidates=qs)
|
||||
|
||||
def validate_session(self, data, **context):
|
||||
data = super().validate_session(data, **context)
|
||||
try:
|
||||
user = data['user']
|
||||
user = data["user"]
|
||||
except KeyError:
|
||||
user = context['user']
|
||||
user = context["user"]
|
||||
try:
|
||||
assert (
|
||||
data['custom_radio'].user == user or
|
||||
data['custom_radio'].is_public)
|
||||
assert data["custom_radio"].user == user or data["custom_radio"].is_public
|
||||
except KeyError:
|
||||
raise serializers.ValidationError(
|
||||
'You must provide a custom radio')
|
||||
raise serializers.ValidationError("You must provide a custom radio")
|
||||
except AssertionError:
|
||||
raise serializers.ValidationError(
|
||||
"You don't have access to this radio")
|
||||
raise serializers.ValidationError("You don't have access to this radio")
|
||||
return data
|
||||
|
||||
|
||||
|
@ -139,15 +131,17 @@ class RelatedObjectRadio(SessionRadio):
|
|||
def clean(self, instance):
|
||||
super().clean(instance)
|
||||
if not instance.related_object:
|
||||
raise ValidationError('Cannot start RelatedObjectRadio without related object')
|
||||
raise ValidationError(
|
||||
"Cannot start RelatedObjectRadio without related object"
|
||||
)
|
||||
if not isinstance(instance.related_object, self.model):
|
||||
raise ValidationError('Trying to start radio with bad related object')
|
||||
raise ValidationError("Trying to start radio with bad related object")
|
||||
|
||||
def get_related_object(self, pk):
|
||||
return self.model.objects.get(pk=pk)
|
||||
|
||||
|
||||
@registry.register(name='tag')
|
||||
@registry.register(name="tag")
|
||||
class TagRadio(RelatedObjectRadio):
|
||||
model = Tag
|
||||
|
||||
|
@ -155,7 +149,8 @@ class TagRadio(RelatedObjectRadio):
|
|||
qs = super().get_queryset(**kwargs)
|
||||
return Track.objects.filter(tags__in=[self.session.related_object])
|
||||
|
||||
@registry.register(name='artist')
|
||||
|
||||
@registry.register(name="artist")
|
||||
class ArtistRadio(RelatedObjectRadio):
|
||||
model = Artist
|
||||
|
||||
|
@ -164,7 +159,7 @@ class ArtistRadio(RelatedObjectRadio):
|
|||
return qs.filter(artist=self.session.related_object)
|
||||
|
||||
|
||||
@registry.register(name='less-listened')
|
||||
@registry.register(name="less-listened")
|
||||
class LessListenedRadio(RelatedObjectRadio):
|
||||
model = User
|
||||
|
||||
|
@ -174,5 +169,5 @@ class LessListenedRadio(RelatedObjectRadio):
|
|||
|
||||
def get_queryset(self, **kwargs):
|
||||
qs = super().get_queryset(**kwargs)
|
||||
listened = self.session.user.listenings.all().values_list('track', flat=True)
|
||||
return qs.exclude(pk__in=listened).order_by('?')
|
||||
listened = self.session.user.listenings.all().values_list("track", flat=True)
|
||||
return qs.exclude(pk__in=listened).order_by("?")
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue