mirror of
https://github.com/spl0k/supysonic.git
synced 2024-12-23 01:16:18 +00:00
110 lines
2.9 KiB
Python
Executable File
110 lines
2.9 KiB
Python
Executable File
# coding: utf-8
|
|
|
|
from flask import request
|
|
from web import app
|
|
from db import Folder, Artist, Album, Track
|
|
import uuid, time, string
|
|
|
|
@app.route('/rest/getMusicFolders.view')
|
|
def list_folders():
|
|
return request.formatter({
|
|
'musicFolders': {
|
|
'musicFolder': [ {
|
|
'id': str(f.id),
|
|
'name': f.name
|
|
} for f in Folder.query.filter(Folder.root == True).order_by(Folder.name).all() ]
|
|
}
|
|
})
|
|
|
|
@app.route('/rest/getIndexes.view')
|
|
def list_indexes():
|
|
musicFolderId = request.args.get('musicFolderId')
|
|
ifModifiedSince = request.args.get('ifModifiedSince')
|
|
if ifModifiedSince:
|
|
try:
|
|
ifModifiedSince = int(ifModifiedSince)
|
|
except:
|
|
return request.error_formatter(0, 'Invalid timestamp')
|
|
|
|
if musicFolderId is None:
|
|
folder = Folder.query.filter(Folder.root == True).all()
|
|
else:
|
|
try:
|
|
mfid = uuid.UUID(musicFolderId)
|
|
except:
|
|
return request.error_formatter(0, 'Invalid id')
|
|
|
|
folder = Folder.query.get(mfid)
|
|
|
|
if not folder or (type(folder) is not list and not folder.root):
|
|
return request.error_formatter(70, 'Folder not found')
|
|
|
|
last_modif = max(map(lambda f: f.last_scan, folder)) if type(folder) is list else folder.last_scan
|
|
last_modif_ts = int(time.mktime(last_modif.timetuple()))
|
|
|
|
if (not ifModifiedSince is None) and last_modif_ts < ifModifiedSince:
|
|
return request.formatter({ 'indexes': { 'lastModified': last_modif_ts } })
|
|
|
|
# The XSD lies, we don't return artists but a directory structure
|
|
if type(folder) is list:
|
|
artists = []
|
|
childs = []
|
|
for f in folder:
|
|
artists += f.children
|
|
childs += f.tracks
|
|
else:
|
|
artists = folder.children
|
|
childs = folder.tracks
|
|
|
|
indexes = {}
|
|
for artist in artists:
|
|
index = artist.name[0].upper()
|
|
if index in map(str, xrange(10)):
|
|
index = '#'
|
|
elif index not in string.letters:
|
|
index = '?'
|
|
|
|
if index not in indexes:
|
|
indexes[index] = []
|
|
|
|
indexes[index].append(artist)
|
|
|
|
return request.formatter({
|
|
'indexes': {
|
|
'lastModified': last_modif_ts,
|
|
'index': [ {
|
|
'name': k,
|
|
'artist': [ {
|
|
'id': str(a.id),
|
|
'name': a.name
|
|
} for a in sorted(v, key = lambda a: a.name.lower()) ]
|
|
} for k, v in sorted(indexes.iteritems()) ],
|
|
'child': [ c.as_subsonic_child() for c in sorted(childs, key = lambda t: t.sort_key()) ]
|
|
}
|
|
})
|
|
|
|
@app.route('/rest/getMusicDirectory.view')
|
|
def show_directory():
|
|
did = request.args.get('id')
|
|
if not did:
|
|
return request.error_formatter(10, 'Missing directory id')
|
|
try:
|
|
fid = uuid.UUID(did)
|
|
except:
|
|
return request.error_formatter(0, 'Invalid directory id')
|
|
|
|
folder = Folder.query.get(fid)
|
|
if not folder:
|
|
return request.error_formatter(70, 'Directory not found')
|
|
|
|
directory = {
|
|
'id': str(folder.id),
|
|
'name': folder.name,
|
|
'child': [ f.as_subsonic_child() for f in sorted(folder.children, key = lambda c: c.name) ] + [ t.as_subsonic_child() for t in sorted(folder.tracks, key = lambda t: t.sort_key()) ]
|
|
}
|
|
if not folder.root:
|
|
directory['parent'] = str(folder.parent_id)
|
|
|
|
return request.formatter({ 'directory': directory })
|
|
|