github/supysonic
1
0
Fork 0
mirror of https://github.com/spl0k/supysonic.git synced 2024-12-22 08:56:17 +00:00
Code Issues Releases Wiki Activity

Porting supysonic.api.browse

Browse Source
This commit is contained in:
Alban Féron 2022-12-18 18:03:51 +01:00
parent 7401b4dec9
commit 95f77cc170
No known key found for this signature in database
GPG Key ID: 8CE0313646D16165
4 changed files with 112 additions and 117 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
supysonic/api/__init__.py
View File

@ -96,8 +96,7 @@ def get_entity(cls, param="id"):
eid = int(eid)
else:
eid = uuid.UUID(eid)
entity = cls[eid]
return entity
return cls[eid]
def get_entity_id(cls, eid):
@ -107,12 +106,12 @@ def get_entity_id(cls, eid):
raise GenericError("Invalid ID")
try:
return int(eid)
except ValueError:
raise GenericError("Invalid ID")
except ValueError as e:
raise GenericError("Invalid ID") from e
try:
return uuid.UUID(eid)
except (AttributeError, ValueError):
raise GenericError("Invalid ID")
except (AttributeError, ValueError) as e:
raise GenericError("Invalid ID") from e
def get_root_folder(id):
@ -121,14 +120,13 @@ def get_root_folder(id):
try:
fid = int(id)
except ValueError:
raise ValueError("Invalid folder ID")
except ValueError as e:
raise ValueError("Invalid folder ID") from e
folder = Folder.get(id=fid, root=True)
if folder is None:
raise NotFound("Folder")
return folder
try:
return Folder.get(id=fid, root=True)
except Folder.DoesNotExist as e:
raise NotFound("Folder") from e
from .errors import *

17
supysonic/api/browse.py
View File

@ -9,6 +9,7 @@ import re
import string
from flask import current_app, request
from peewee import fn
from ..db import Folder, Artist, Album, Track
@ -22,7 +23,7 @@ def list_folders():
{
"musicFolder": [
{"id": str(f.id), "name": f.name}
for f in Folder.select(lambda f: f.root).order_by(Folder.name)
for f in Folder.select().where(Folder.root).order_by(Folder.name)
]
},
)
@ -77,7 +78,7 @@ def list_indexes():
ifModifiedSince = int(ifModifiedSince) / 1000
if musicFolderId is None:
folders = Folder.select(lambda f: f.root)[:]
folders = Folder.select().where(Folder.root)[:]
else:
folders = [get_root_folder(musicFolderId)]
@ -95,8 +96,8 @@ def list_indexes():
artists = []
children = []
for f in folders:
artists += f.children.select()[:]
children += f.tracks.select()[:]
artists += f.children[:]
children += f.tracks[:]
indexes = build_indexes(artists)
return request.formatter(
@ -137,9 +138,11 @@ def list_genres():
{
"genre": [
{"value": genre, "songCount": sc, "albumCount": ac}
for genre, sc, ac in select(
(t.genre, count(), count(t.album)) for t in Track if t.genre
for genre, sc, ac in Track.select(
Track.genre, fn.count(), fn.count(Track.album.distinct())
)
.group_by(Track.genre)
.tuples()
]
},
)
@ -152,7 +155,7 @@ def list_artists():
query = Artist.select()
if mfid is not None:
folder = get_root_folder(mfid)
query = Artist.select(lambda a: folder in a.tracks.root_folder)
query = Artist.select().join(Track).where(Track.root_folder == folder)
indexes = build_indexes(query)
return request.formatter(

2
supysonic/db.py
View File

@ -150,7 +150,7 @@ class Folder(PathMixin, db.Model):
"name": self.name,
"child": [
f.as_subsonic_child(user)
for f in self.children.order_by(lambda c: c.name.lower())
for f in self.children.order_by(fn.lower(Folder.name))
]
+ [
t.as_subsonic_child(user, client)

26
tests/api/test_browse.py
View File

@ -9,8 +9,6 @@ import time
import unittest
import uuid
from pony.orm import db_session
from supysonic.db import Folder, Artist, Album, Track
from .apitestbase import ApiTestBase
@ -20,30 +18,31 @@ class BrowseTestCase(ApiTestBase):
def setUp(self):
super().setUp()
with db_session:
self.empty_root = Folder(root=True, name="Empty root", path="/tmp")
self.root = Folder(root=True, name="Root folder", path="tests/assets")
self.empty_root = Folder.create(root=True, name="Empty root", path="/tmp")
self.root = Folder.create(root=True, name="Root folder", path="tests/assets")
for letter in "ABC":
folder = Folder(
folder = Folder.create(
name=letter + "rtist",
path="tests/assets/{}rtist".format(letter),
root=False,
parent=self.root,
)
artist = Artist(name=letter + "rtist")
artist = Artist.create(name=letter + "rtist")
for lether in "AB":
afolder = Folder(
afolder = Folder.create(
name=letter + lether + "lbum",
path="tests/assets/{0}rtist/{0}{1}lbum".format(letter, lether),
root=False,
parent=folder,
)
album = Album(name=letter + lether + "lbum", artist=artist)
album = Album.create(name=letter + lether + "lbum", artist=artist)
for num, song in enumerate(["One", "Two", "Three"]):
Track(
Track.create(
disc=1,
number=num,
title=song,
@ -61,7 +60,7 @@ class BrowseTestCase(ApiTestBase):
)
self.assertEqual(Folder.select().count(), 11)
self.assertEqual(Folder.select(lambda f: f.root).count(), 2)
self.assertEqual(Folder.select().where(Folder.root).count(), 2)
self.assertEqual(Artist.select().count(), 3)
self.assertEqual(Album.select().count(), 6)
self.assertEqual(Track.select().count(), 18)
@ -86,7 +85,6 @@ class BrowseTestCase(ApiTestBase):
)
self.assertEqual(len(child), 0)
with db_session:
fid = Folder.get(name="Empty root").id
rv, child = self._make_request(
"getIndexes", {"musicFolderId": str(fid)}, tag="indexes"
@ -106,7 +104,6 @@ class BrowseTestCase(ApiTestBase):
self._make_request("getMusicDirectory", {"id": 1234567890}, error=70)
# should test with folders with both children folders and tracks. this code would break in that case
with db_session:
for f in Folder.select():
rv, child = self._make_request(
"getMusicDirectory", {"id": str(f.id)}, tag="directory"
@ -158,7 +155,6 @@ class BrowseTestCase(ApiTestBase):
self._make_request("getArtist", {"id": "artist"}, error=0)
self._make_request("getArtist", {"id": str(uuid.uuid4())}, error=70)
with db_session:
for ar in Artist.select():
rv, child = self._make_request(
"getArtist", {"id": str(ar.id)}, tag="artist"
@ -181,7 +177,6 @@ class BrowseTestCase(ApiTestBase):
self._make_request("getAlbum", {"id": "nastynasty"}, error=0)
self._make_request("getAlbum", {"id": str(uuid.uuid4())}, error=70)
with db_session:
a = Album.select().first()
rv, child = self._make_request("getAlbum", {"id": str(a.id)}, tag="album")
self.assertEqual(child.get("id"), str(a.id))
@ -201,7 +196,6 @@ class BrowseTestCase(ApiTestBase):
self._make_request("getSong", {"id": "nastynasty"}, error=0)
self._make_request("getSong", {"id": str(uuid.uuid4())}, error=70)
with db_session:
s = Track.select().first()
self._make_request("getSong", {"id": str(s.id)}, tag="song")