Blacked the code

This commit is contained in:
Eliot Berriot 2018-06-09 15:36:16 +02:00
parent b6fc0051fa
commit 62ca3bd736
No known key found for this signature in database
GPG key ID: DD6965E2476E5C27
279 changed files with 8861 additions and 9527 deletions

View file

@ -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("?")