Skip to content
Snippets Groups Projects
Verified Commit ccd600ec authored by Eliot Berriot's avatar Eliot Berriot
Browse files

See #574: Implemented getRandomSongs endpoint

parent 82f2f606
No related branches found
No related tags found
No related merge requests found
...@@ -219,6 +219,27 @@ class SubsonicViewSet(viewsets.GenericViewSet): ...@@ -219,6 +219,27 @@ class SubsonicViewSet(viewsets.GenericViewSet):
data = {"starred2": {"song": serializers.get_starred_tracks_data(favorites)}} data = {"starred2": {"song": serializers.get_starred_tracks_data(favorites)}}
return response.Response(data) return response.Response(data)
@list_route(
methods=["get", "post"], url_name="get_random_songs", url_path="getRandomSongs"
)
def get_random_songs(self, request, *args, **kwargs):
data = request.GET or request.POST
actor = utils.get_actor_from_request(request)
queryset = music_models.Track.objects.all()
queryset = queryset.playable_by(actor)
try:
size = int(data["size"])
except (TypeError, KeyError, ValueError):
size = 50
queryset = queryset.playable_by(actor).prefetch_related('uploads').order_by("?")[:size]
data = {
"randomSongs": {
"song": serializers.GetSongSerializer(queryset, many=True).data
}
}
return response.Response(data)
@list_route(methods=["get", "post"], url_name="get_starred", url_path="getStarred") @list_route(methods=["get", "post"], url_name="get_starred", url_path="getStarred")
def get_starred(self, request, *args, **kwargs): def get_starred(self, request, *args, **kwargs):
favorites = request.user.track_favorites.all() favorites = request.user.track_favorites.all()
......
...@@ -245,6 +245,27 @@ def test_get_starred2(f, db, logged_in_api_client, factories): ...@@ -245,6 +245,27 @@ def test_get_starred2(f, db, logged_in_api_client, factories):
} }
@pytest.mark.parametrize("f", ["json"])
def test_get_random_songs(f, db, logged_in_api_client, factories, mocker):
url = reverse("api:subsonic-get-random-songs")
assert url.endswith("getRandomSongs") is True
track1 = factories["music.Track"]()
track2 = factories["music.Track"]()
factories["music.Track"]()
order_by = mocker.patch.object(
music_models.TrackQuerySet, 'order_by', return_value=[track1, track2]
)
response = logged_in_api_client.get(url, {"f": f, "size": 2})
assert response.status_code == 200
assert response.data == {
"randomSongs": {"song": serializers.GetSongSerializer([track1, track2], many=True).data}
}
order_by.assert_called_once_with("?")
@pytest.mark.parametrize("f", ["xml", "json"]) @pytest.mark.parametrize("f", ["xml", "json"])
def test_get_starred(f, db, logged_in_api_client, factories): def test_get_starred(f, db, logged_in_api_client, factories):
url = reverse("api:subsonic-get-starred") url = reverse("api:subsonic-get-starred")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment