250 lines
9.7 KiB
Python
250 lines
9.7 KiB
Python
import sys
|
|
import logging
|
|
import cherrypy
|
|
import subprocess
|
|
from time import time
|
|
from bs4 import BeautifulSoup
|
|
from pysonic.library import LETTER_GROUPS
|
|
from pysonic.types import MUSIC_TYPES
|
|
|
|
|
|
logging = logging.getLogger("api")
|
|
|
|
|
|
class PysonicApi(object):
|
|
def __init__(self, db, library, options):
|
|
self.db = db
|
|
self.library = library
|
|
self.options = options
|
|
|
|
def response(self, status="ok"):
|
|
doc = BeautifulSoup('', features='lxml-xml')
|
|
root = doc.new_tag("subsonic-response", xmlns="http://subsonic.org/restapi", status=status, version="1.15.0")
|
|
doc.append(root)
|
|
return doc, root
|
|
|
|
@cherrypy.expose
|
|
def ping_view(self, **kwargs):
|
|
# Called when the app hits the "test connection" server option
|
|
cherrypy.response.headers['Content-Type'] = 'text/xml; charset=utf-8'
|
|
doc, root = self.response()
|
|
yield doc.prettify()
|
|
|
|
@cherrypy.expose
|
|
def getLicense_view(self, **kwargs):
|
|
# Called after ping.view
|
|
cherrypy.response.headers['Content-Type'] = 'text/xml; charset=utf-8'
|
|
doc, root = self.response()
|
|
root.append(doc.new_tag("license",
|
|
valid="true",
|
|
email="admin@localhost",
|
|
licenseExpires="2100-01-01T00:00:00.000Z",
|
|
trialExpires="2100-01-01T01:01:00.000Z"))
|
|
yield doc.prettify()
|
|
|
|
@cherrypy.expose
|
|
def getMusicFolders_view(self, **kwargs):
|
|
# Get list of configured dirs
|
|
# {'c': 'DSub', 's': 'bfk9mir8is02u3m5as8ucsehn0', 'v': '1.2.0',
|
|
# 't': 'e2b09fb9233d1bfac9abe3dc73017f1e', 'u': 'dave'}
|
|
# Access-Control-Allow-Origin:*
|
|
# Content-Encoding:gzip
|
|
# Content-Type:text/xml; charset=utf-8
|
|
# Server:Jetty(6.1.x)
|
|
# Transfer-Encoding:chunked
|
|
cherrypy.response.headers['Content-Type'] = 'text/xml; charset=utf-8'
|
|
|
|
doc, root = self.response()
|
|
folder_list = doc.new_tag("musicFolders")
|
|
root.append(folder_list)
|
|
|
|
for folder in self.library.get_libraries():
|
|
entry = doc.new_tag("musicFolder", id=folder["id"])
|
|
entry.attrs["name"] = folder["name"]
|
|
folder_list.append(entry)
|
|
yield doc.prettify()
|
|
|
|
@cherrypy.expose
|
|
def getIndexes_view(self, **kwargs):
|
|
# Get listing of top-level dir
|
|
# /rest/getIndexes.view?u=dave&s=bfk9mir8is02u3m5as8ucsehn0
|
|
# &t=e2b09fb9233d1bfac9abe3dc73017f1e&v=1.2.0&c=DSub HTTP/1.1
|
|
cherrypy.response.headers['Content-Type'] = 'text/xml; charset=utf-8'
|
|
doc, root = self.response()
|
|
indexes = doc.new_tag("indexes", lastModified="1502310831000", ignoredArticles="The El La Los Las Le Les")
|
|
doc.append(indexes)
|
|
|
|
for letter in LETTER_GROUPS:
|
|
index = doc.new_tag("index")
|
|
index.attrs["name"] = letter.upper()
|
|
indexes.append(index)
|
|
for artist in self.library.get_artists():
|
|
if artist["name"][0].lower() in letter:
|
|
artist_tag = doc.new_tag("artist")
|
|
artist_tag.attrs.update({"id": artist["id"], "name": artist["name"]})
|
|
index.append(artist_tag)
|
|
yield doc.prettify()
|
|
|
|
@cherrypy.expose
|
|
def getMusicDirectory_view(self, id, **kwargs):
|
|
"""
|
|
List an artist dir
|
|
"""
|
|
dir_id = int(id)
|
|
|
|
cherrypy.response.headers['Content-Type'] = 'text/xml; charset=utf-8'
|
|
doc, root = self.response()
|
|
|
|
dirtag = doc.new_tag("directory")
|
|
|
|
directory = self.library.get_dir(dir_id)
|
|
dir_meta = self.db.decode_metadata(directory["metadata"])
|
|
children = self.library.get_dir_children(dir_id)
|
|
dirtag.attrs.update(name=directory['name'], id=directory['id'],
|
|
parent=directory['parent'], playCount=10)
|
|
root.append(dirtag)
|
|
|
|
for item in children:
|
|
# omit not dirs and media in browser
|
|
if not item["isdir"] and item["type"] not in MUSIC_TYPES:
|
|
continue
|
|
child = doc.new_tag("child",
|
|
id=item["id"],
|
|
parent=directory["id"],
|
|
isDir="true" if item['isdir'] else "false",
|
|
title=item["name"],
|
|
album=item["name"],
|
|
artist=directory["name"],
|
|
# playCount="5",
|
|
# created="2016-04-25T07:31:33.000Z"
|
|
# track="3",
|
|
# year="2012",
|
|
# genre="Other",
|
|
# coverArt="12835",
|
|
# contentType="audio/mpeg"
|
|
# suffix="mp3"
|
|
# size="15838864"
|
|
# duration="395"
|
|
# bitRate="320"
|
|
# path="Cosmic Gate/Sign Of The Times/03 Flatline (featuring Kyler England).mp3"
|
|
# albumId="933"
|
|
# artistId="353"
|
|
# type="music"/>
|
|
)
|
|
item_meta = self.db.decode_metadata(item['metadata'])
|
|
if 'cover' in item_meta:
|
|
child.attrs["coverArt"] = item_meta["cover"]
|
|
elif 'cover' in dir_meta:
|
|
child.attrs["coverArt"] = dir_meta["cover"]
|
|
dirtag.append(child)
|
|
yield doc.prettify()
|
|
|
|
@cherrypy.expose
|
|
def stream_view(self, id, maxBitRate="256", **kwargs):
|
|
maxBitRate = int(maxBitRate)
|
|
assert maxBitRate >= 32 and maxBitRate <= 320
|
|
fpath = self.library.get_filepath(id)
|
|
meta = self.library.get_file_metadata(id)
|
|
cherrypy.response.headers['Content-Type'] = 'audio/mpeg'
|
|
if self.options.skip_transcode and meta["type"] == "audio/mpeg":
|
|
def content():
|
|
with open(fpath, "rb") as f:
|
|
while True:
|
|
data = f.read(16 * 1024)
|
|
if not data:
|
|
break
|
|
yield data
|
|
else:
|
|
def content():
|
|
transcode_args = ["ffmpeg", "-i", fpath, "-map", "0:0", "-b:a",
|
|
"{}k".format(min(maxBitRate, self.options.max_bitrate)),
|
|
"-v", "0", "-f", "mp3", "-"]
|
|
logging.info(' '.join(transcode_args))
|
|
start = time()
|
|
proc = subprocess.Popen(transcode_args, stdout=subprocess.PIPE)
|
|
while True:
|
|
data = proc.stdout.read(16 * 1024)
|
|
if not data:
|
|
break
|
|
yield data
|
|
logging.warning("transcoded {} in {}s".format(id, int(time() - start)))
|
|
return content()
|
|
stream_view._cp_config = {'response.stream': True}
|
|
|
|
@cherrypy.expose
|
|
def getCoverArt_view(self, id, **kwargs):
|
|
# /rest/getCoverArt.view?u=dave&s=bfk9mir8is02u3m5as8ucsehn0
|
|
# &t=e2b09fb9233d1bfac9abe3dc73017f1e&v=1.2.0&c=DSub&id=12833
|
|
fpath = self.library.get_filepath(id)
|
|
type2ct = {
|
|
'jpg': 'image/jpeg',
|
|
'png': 'image/png'
|
|
}
|
|
cherrypy.response.headers['Content-Type'] = type2ct[fpath[-3:]]
|
|
|
|
def content():
|
|
total = 0
|
|
with open(fpath, "rb") as f:
|
|
while True:
|
|
data = f.read(8192)
|
|
if not data:
|
|
break
|
|
total += len(data)
|
|
yield data
|
|
logging.info("\nSent {} bytes for {}".format(total, fpath))
|
|
return content()
|
|
|
|
getCoverArt_view._cp_config = {'response.stream': True}
|
|
|
|
@cherrypy.expose
|
|
def getArtistInfo_view(self, id, includeNotPresent="true", **kwargs):
|
|
# /rest/getArtistInfo.view?
|
|
# u=dave
|
|
# s=gqua9i6c414aomjok8f6b0kdp1
|
|
# t=ed1d31850bbd27690687305d9ccbdabf
|
|
# v=1.2.0
|
|
# c=DSub
|
|
# id=7
|
|
# includeNotPresent=true
|
|
info = self.library.get_artist_info(id)
|
|
cherrypy.response.headers['Content-Type'] = 'text/xml; charset=utf-8'
|
|
doc, root = self.response()
|
|
|
|
dirtag = doc.new_tag("artistInfo")
|
|
root.append(dirtag)
|
|
|
|
for key, value in info.items():
|
|
if key == "similarArtists":
|
|
continue
|
|
tag = doc.new_tag(key)
|
|
tag.append(str(value))
|
|
dirtag.append(tag)
|
|
yield doc.prettify()
|
|
|
|
@cherrypy.expose
|
|
def getUser_view(self, u, username, **kwargs):
|
|
cherrypy.response.headers['Content-Type'] = 'text/xml; charset=utf-8'
|
|
doc, root = self.response()
|
|
user = doc.new_tag("user",
|
|
username="admin",
|
|
email="admin@localhost",
|
|
scrobblingEnabled="false",
|
|
adminRole="false",
|
|
settingsRole="false",
|
|
downloadRole="true",
|
|
uploadRole="false",
|
|
playlistRole="true",
|
|
coverArtRole="false",
|
|
commentRole="false",
|
|
podcastRole="false",
|
|
streamRole="true",
|
|
jukeboxRole="false",
|
|
shareRole="true",
|
|
videoConversionRole="false",
|
|
avatarLastChanged="2017-08-07T20:16:24.596Z")
|
|
root.append(user)
|
|
folder = doc.new_tag("folder")
|
|
folder.append("0")
|
|
user.append(folder)
|
|
yield doc.prettify()
|