diff --git a/api/albums_songs.py b/api/albums_songs.py index 0e7c565..e761af6 100755 --- a/api/albums_songs.py +++ b/api/albums_songs.py @@ -138,7 +138,7 @@ def now_playing(): def get_starred(): return request.formatter({ 'starred': { - 'artist': [ { 'id': sf.starred.name, 'name': sf.starred_id } for sf in StarredFolder.query.join(User).join(Folder).filter(User.name == request.username).filter(~ Folder.tracks.any()) ], + 'artist': [ { 'id': str(sf.starred_id), 'name': sf.starred.name } for sf in StarredFolder.query.join(User).join(Folder).filter(User.name == request.username).filter(~ Folder.tracks.any()) ], 'album': [ sf.starred.as_subsonic_child(request.user) for sf in StarredFolder.query.join(User).join(Folder).filter(User.name == request.username).filter(Folder.tracks.any()) ], 'song': [ st.starred.as_subsonic_child(request.user) for st in StarredTrack.query.join(User).filter(User.name == request.username) ] } diff --git a/api/search.py b/api/search.py new file mode 100755 index 0000000..afab93c --- /dev/null +++ b/api/search.py @@ -0,0 +1,104 @@ +# coding: utf-8 + +from flask import request +from web import app +from db import Folder, Track, Artist, Album + +@app.route('/rest/search.view') +def old_search(): + artist, album, title, anyf, count, offset, newer_than = map(request.args.get, [ 'artist', 'album', 'title', 'any', 'count', 'offset', 'newerThan' ]) + try: + count = int(count) if count else 20 + offset = int(offset) if offset else 0 + newer_than = int(newer_than) if newer_than else 0 + except: + return request.error_formatter(0, 'Invalid parameter') + + if artist: + query = Folder.query.filter(~ Folder.tracks.any(), Folder.name.contains(artist)) + elif album: + query = Folder.query.filter(Folder.tracks.any(), Folder.name.contains(album)) + elif title: + query = Track.query.filter(Track.title.contains(title)) + elif anyf: + folders = Folder.query.filter(Folder.name.contains(anyf)) + tracks = Track.query.filter(Track.title.contains(anyf)) + res = folders.slice(offset, offset + count).all() + if offset + count > folders.count(): + toff = max(0, offset - folders.count()) + tend = offset + count - folders.count() + res += tracks.slice(toff, tend).all() + + return request.formatter({ 'searchResult': { + 'totalHits': folders.count() + tracks.count(), + 'offset': offset, + 'match': [ r.as_subsonic_child(request.user) for r in res ] + }}) + else: + return request.error_formatter(10, 'Missing search parameter') + + print "---" + print query + + return request.formatter({ 'searchResult': { + 'totalHits': query.count(), + 'offset': offset, + 'match': [ r.as_subsonic_child(request.user) for r in query.slice(offset, offset + count) ] + }}) + +@app.route('/rest/search2.view') +def new_search(): + query, artist_count, artist_offset, album_count, album_offset, song_count, song_offset = map( + request.args.get, [ 'query', 'artistCount', 'artistOffset', 'albumCount', 'albumOffset', 'songCount', 'songOffset' ]) + + try: + artist_count = int(artist_count) if artist_count else 20 + artist_offset = int(artist_offset) if artist_offset else 0 + album_count = int(album_count) if album_count else 20 + album_offset = int(album_offset) if album_offset else 0 + song_count = int(song_count) if song_count else 20 + song_offset = int(song_offset) if song_offset else 0 + except: + return request.error_formatter(0, 'Invalid parameter') + + if not query: + return request.error_formatter(10, 'Missing query parameter') + + artist_query = Folder.query.filter(~ Folder.tracks.any(), Folder.name.contains(query)).slice(artist_offset, artist_offset + artist_count) + album_query = Folder.query.filter(Folder.tracks.any(), Folder.name.contains(query)).slice(album_offset, album_offset + album_count) + song_query = Track.query.filter(Track.title.contains(query)).slice(song_offset, song_offset + song_count) + + return request.formatter({ 'searchResult2': { + 'artist': [ { 'id': str(a.id), 'name': a.name } for a in artist_query ], + 'album': [ f.as_subsonic_child(request.user) for f in album_query ], + 'song': [ t.as_subsonic_child(request.user) for t in song_query ] + }}) + +@app.route('/rest/search3.view') +def search_id3(): + query, artist_count, artist_offset, album_count, album_offset, song_count, song_offset = map( + request.args.get, [ 'query', 'artistCount', 'artistOffset', 'albumCount', 'albumOffset', 'songCount', 'songOffset' ]) + + try: + artist_count = int(artist_count) if artist_count else 20 + artist_offset = int(artist_offset) if artist_offset else 0 + album_count = int(album_count) if album_count else 20 + album_offset = int(album_offset) if album_offset else 0 + song_count = int(song_count) if song_count else 20 + song_offset = int(song_offset) if song_offset else 0 + except: + return request.error_formatter(0, 'Invalid parameter') + + if not query: + return request.error_formatter(10, 'Missing query parameter') + + artist_query = Artist.query.filter(Artist.name.contains(query)).slice(artist_offset, artist_offset + artist_count) + album_query = Album.query.filter(Album.name.contains(query)).slice(album_offset, album_offset + album_count) + song_query = Track.query.filter(Track.title.contains(query)).slice(song_offset, song_offset + song_count) + + return request.formatter({ 'searchResult2': { + 'artist': [ a.as_subsonic_artist(request.user) for a in artist_query ], + 'album': [ a.as_subsonic_album(request.user) for a in album_query ], + 'song': [ t.as_subsonic_child(request.user) for t in song_query ] + }}) + diff --git a/web.py b/web.py index a302c5a..e55b2a8 100755 --- a/web.py +++ b/web.py @@ -61,4 +61,5 @@ import api.albums_songs import api.media import api.annotation import api.chat +import api.search