104 lines
4.1 KiB
Python
104 lines
4.1 KiB
Python
import os
|
|
import logging
|
|
import cherrypy
|
|
from sqlite3 import IntegrityError
|
|
from pysonic.api import PysonicApi
|
|
from pysonic.library import PysonicLibrary
|
|
from pysonic.database import PysonicDatabase, DuplicateRootException
|
|
|
|
|
|
def main():
|
|
import argparse
|
|
import signal
|
|
|
|
parser = argparse.ArgumentParser(description="Pysonic music streaming server")
|
|
|
|
parser.add_argument('-p', '--port', default=8080, type=int, help="tcp port to listen on")
|
|
parser.add_argument('-d', '--dirs', required=True, nargs='+', help="new music dirs to share")
|
|
parser.add_argument('-u', '--user', nargs='+', type=lambda x: x.split(":"), default=[],
|
|
help="user:password pairs for auth")
|
|
parser.add_argument('--disable-auth', action="store_true", help="disable authentication")
|
|
parser.add_argument('-s', '--database-path', default="./db.sqlite", help="path to persistent sqlite database")
|
|
parser.add_argument('--debug', action="store_true", help="enable development options")
|
|
|
|
group = parser.add_argument_group("app options")
|
|
group.add_argument("--skip-transcode", action="store_true", help="instead of trancoding mp3s, send as-is")
|
|
group.add_argument("--no-rescan", action="store_true", help="don't perform simple scan on startup")
|
|
group.add_argument("--deep-rescap", action="store_true", help="perform deep scan (read id3 etc)")
|
|
group.add_argument("--enable-prune", action="store_true", help="enable removal of media not found on disk")
|
|
group.add_argument("--max-bitrate", type=int, default=320, help="maximum send bitrate")
|
|
group.add_argument("--enable-cors", action="store_true", help="add response headers to allow cors")
|
|
|
|
args = parser.parse_args()
|
|
|
|
logging.basicConfig(level=logging.INFO if args.debug else logging.WARNING)
|
|
|
|
db = PysonicDatabase(path=args.database_path)
|
|
library = PysonicLibrary(db)
|
|
for dirname in args.dirs:
|
|
assert os.path.exists(dirname) and dirname.startswith("/"), "--dirs must be absolute paths and exist!"
|
|
try:
|
|
library.add_dir(dirname)
|
|
except DuplicateRootException:
|
|
pass
|
|
library.update()
|
|
|
|
for username, password in args.user:
|
|
try:
|
|
db.add_user(username, password)
|
|
except IntegrityError:
|
|
db.update_user(username, password)
|
|
|
|
logging.warning("Libraries: {}".format([i["name"] for i in library.get_libraries()]))
|
|
logging.warning("Artists: {}".format([i["name"] for i in library.get_artists()]))
|
|
logging.warning("Albums: {}".format(len(library.get_albums())))
|
|
|
|
api = PysonicApi(db, library, args)
|
|
api_config = {}
|
|
if args.disable_auth:
|
|
logging.warning("starting up with auth disabled")
|
|
else:
|
|
api_config.update({'tools.auth_basic.on': True,
|
|
'tools.auth_basic.realm': 'pysonic',
|
|
'tools.auth_basic.checkpassword': db.validate_password})
|
|
if args.enable_cors:
|
|
def cors():
|
|
cherrypy.response.headers["Access-Control-Allow-Origin"] = "*"
|
|
cherrypy.tools.cors = cherrypy.Tool('before_handler', cors)
|
|
api_config.update({'tools.cors.on': True})
|
|
|
|
cherrypy.tree.mount(api, '/rest/', {'/': api_config})
|
|
|
|
cherrypy.config.update({
|
|
'sessionFilter.on': True,
|
|
'tools.sessions.on': True,
|
|
'tools.sessions.locking': 'explicit',
|
|
'tools.sessions.timeout': 525600,
|
|
'tools.gzip.on': True,
|
|
'request.show_tracebacks': True,
|
|
'server.socket_port': args.port,
|
|
'server.thread_pool': 25,
|
|
'server.socket_host': '0.0.0.0',
|
|
'server.show_tracebacks': True,
|
|
'server.socket_timeout': 5,
|
|
'log.screen': False,
|
|
'engine.autoreload.on': args.debug
|
|
})
|
|
|
|
def signal_handler(signum, stack):
|
|
logging.critical('Got sig {}, exiting...'.format(signum))
|
|
cherrypy.engine.exit()
|
|
|
|
signal.signal(signal.SIGINT, signal_handler)
|
|
signal.signal(signal.SIGTERM, signal_handler)
|
|
|
|
try:
|
|
cherrypy.engine.start()
|
|
cherrypy.engine.block()
|
|
finally:
|
|
logging.info("API has shut down")
|
|
cherrypy.engine.exit()
|
|
|
|
if __name__ == '__main__':
|
|
main()
|