mirror of
https://github.com/spl0k/supysonic.git
synced 2024-12-23 01:16:18 +00:00
417 lines
14 KiB
Python
Executable File
417 lines
14 KiB
Python
Executable File
# coding: utf-8
|
|
#
|
|
# This file is part of Supysonic.
|
|
# Supysonic is a Python implementation of the Subsonic server API.
|
|
#
|
|
# Copyright (C) 2013-2019 Alban 'spl0k' Féron
|
|
#
|
|
# Distributed under terms of the GNU AGPLv3 license.
|
|
|
|
import argparse
|
|
import cmd
|
|
import getpass
|
|
import sys
|
|
import time
|
|
|
|
from pony.orm import db_session, select
|
|
from pony.orm import ObjectNotFound
|
|
|
|
from .config import IniConfig
|
|
from .daemon.client import DaemonClient
|
|
from .daemon.exceptions import DaemonUnavailableError
|
|
from .db import Folder, User, init_database, release_database
|
|
from .managers.folder import FolderManager
|
|
from .managers.user import UserManager
|
|
from .scanner import Scanner
|
|
|
|
|
|
class TimedProgressDisplay:
|
|
def __init__(self, stdout, interval=5):
|
|
self.__stdout = stdout
|
|
self.__interval = interval
|
|
self.__last_display = 0
|
|
self.__last_len = 0
|
|
|
|
def __call__(self, name, scanned):
|
|
if time.time() - self.__last_display > self.__interval:
|
|
progress = "Scanning '{0}': {1} files scanned".format(name, scanned)
|
|
self.__stdout.write("\b" * self.__last_len)
|
|
self.__stdout.write(progress)
|
|
self.__stdout.flush()
|
|
|
|
self.__last_len = len(progress)
|
|
self.__last_display = time.time()
|
|
|
|
|
|
class CLIParser(argparse.ArgumentParser):
|
|
def error(self, message):
|
|
self.print_usage(sys.stderr)
|
|
raise RuntimeError(message)
|
|
|
|
|
|
class SupysonicCLI(cmd.Cmd):
|
|
prompt = "supysonic> "
|
|
|
|
def _make_do(self, command):
|
|
def method(obj, line):
|
|
try:
|
|
args = getattr(obj, command + "_parser").parse_args(line.split())
|
|
except RuntimeError as e:
|
|
self.write_error_line(str(e))
|
|
return
|
|
|
|
if hasattr(obj.__class__, command + "_subparsers"):
|
|
try:
|
|
func = getattr(obj, "{}_{}".format(command, args.action))
|
|
except AttributeError:
|
|
return obj.default(line)
|
|
return func(
|
|
**{key: vars(args)[key] for key in vars(args) if key != "action"}
|
|
)
|
|
else:
|
|
try:
|
|
func = getattr(obj, command)
|
|
except AttributeError:
|
|
return obj.default(line)
|
|
return func(**vars(args))
|
|
|
|
return method
|
|
|
|
def __init__(self, config, stderr=None, *args, **kwargs):
|
|
cmd.Cmd.__init__(self, *args, **kwargs)
|
|
|
|
if stderr is not None:
|
|
self.stderr = stderr
|
|
else:
|
|
self.stderr = sys.stderr
|
|
|
|
self.__config = config
|
|
self.__daemon = DaemonClient(config.DAEMON["socket"])
|
|
|
|
# Generate do_* and help_* methods
|
|
for parser_name in filter(
|
|
lambda attr: attr.endswith("_parser") and "_" not in attr[:-7],
|
|
dir(self.__class__),
|
|
):
|
|
command = parser_name[:-7]
|
|
|
|
if not hasattr(self.__class__, "do_" + command):
|
|
setattr(self.__class__, "do_" + command, self._make_do(command))
|
|
|
|
if hasattr(self.__class__, "do_" + command) and not hasattr(
|
|
self.__class__, "help_" + command
|
|
):
|
|
setattr(
|
|
self.__class__,
|
|
"help_" + command,
|
|
getattr(self.__class__, parser_name).print_help,
|
|
)
|
|
if hasattr(self.__class__, command + "_subparsers"):
|
|
for action, subparser in getattr(
|
|
self.__class__, command + "_subparsers"
|
|
).choices.items():
|
|
setattr(
|
|
self, "help_{} {}".format(command, action), subparser.print_help
|
|
)
|
|
|
|
def write_line(self, line=""):
|
|
self.stdout.write(line + "\n")
|
|
|
|
def write_error_line(self, line=""):
|
|
self.stderr.write(line + "\n")
|
|
|
|
def do_EOF(self, line):
|
|
return True
|
|
|
|
do_exit = do_EOF
|
|
|
|
def default(self, line):
|
|
self.write_line("Unknown command %s" % line.split()[0])
|
|
self.do_help(None)
|
|
|
|
def postloop(self):
|
|
self.write_line()
|
|
|
|
def completedefault(self, text, line, begidx, endidx):
|
|
command = line.split()[0]
|
|
parsers = getattr(self.__class__, command + "_subparsers", None)
|
|
if not parsers:
|
|
return []
|
|
|
|
num_words = len(line[len(command) : begidx].split())
|
|
if num_words == 0:
|
|
return [a for a in parsers.choices if a.startswith(text)]
|
|
return []
|
|
|
|
folder_parser = CLIParser(prog="folder", add_help=False)
|
|
folder_subparsers = folder_parser.add_subparsers(dest="action")
|
|
folder_subparsers.add_parser("list", help="Lists folders", add_help=False)
|
|
folder_add_parser = folder_subparsers.add_parser(
|
|
"add", help="Adds a folder", add_help=False
|
|
)
|
|
folder_add_parser.add_argument("name", help="Name of the folder to add")
|
|
folder_add_parser.add_argument(
|
|
"path", help="Path to the directory pointed by the folder"
|
|
)
|
|
folder_del_parser = folder_subparsers.add_parser(
|
|
"delete", help="Deletes a folder", add_help=False
|
|
)
|
|
folder_del_parser.add_argument("name", help="Name of the folder to delete")
|
|
folder_scan_parser = folder_subparsers.add_parser(
|
|
"scan", help="Run a scan on specified folders", add_help=False
|
|
)
|
|
folder_scan_parser.add_argument(
|
|
"folders",
|
|
metavar="folder",
|
|
nargs="*",
|
|
help="Folder(s) to be scanned. If ommitted, all folders are scanned",
|
|
)
|
|
folder_scan_parser.add_argument(
|
|
"-f",
|
|
"--force",
|
|
action="store_true",
|
|
help="Force scan of already know files even if they haven't changed",
|
|
)
|
|
folder_scan_target_group = folder_scan_parser.add_mutually_exclusive_group()
|
|
folder_scan_target_group.add_argument(
|
|
"--background",
|
|
action="store_true",
|
|
help="Scan the folder(s) in the background. Requires the daemon to be running.",
|
|
)
|
|
folder_scan_target_group.add_argument(
|
|
"--foreground",
|
|
action="store_true",
|
|
help="Scan the folder(s) in the foreground, blocking the processus while the scan is running.",
|
|
)
|
|
|
|
@db_session
|
|
def folder_list(self):
|
|
self.write_line("Name\t\tPath\n----\t\t----")
|
|
self.write_line(
|
|
"\n".join(
|
|
"{0: <16}{1}".format(f.name, f.path)
|
|
for f in Folder.select(lambda f: f.root)
|
|
)
|
|
)
|
|
|
|
@db_session
|
|
def folder_add(self, name, path):
|
|
try:
|
|
FolderManager.add(name, path)
|
|
self.write_line("Folder '{}' added".format(name))
|
|
except ValueError as e:
|
|
self.write_error_line(str(e))
|
|
|
|
@db_session
|
|
def folder_delete(self, name):
|
|
try:
|
|
FolderManager.delete_by_name(name)
|
|
self.write_line("Deleted folder '{}'".format(name))
|
|
except ObjectNotFound as e:
|
|
self.write_error_line(str(e))
|
|
|
|
def folder_scan(self, folders, force, background, foreground):
|
|
auto = not background and not foreground
|
|
if auto:
|
|
try:
|
|
self.__folder_scan_background(folders, force)
|
|
except DaemonUnavailableError:
|
|
self.write_error_line(
|
|
"Couldn't connect to the daemon, scanning in foreground"
|
|
)
|
|
self.__folder_scan_foreground(folders, force)
|
|
elif background:
|
|
try:
|
|
self.__folder_scan_background(folders, force)
|
|
except DaemonUnavailableError:
|
|
self.write_error_line(
|
|
"Couldn't connect to the daemon, please use the '--foreground' option"
|
|
)
|
|
elif foreground:
|
|
self.__folder_scan_foreground(folders, force)
|
|
|
|
def __folder_scan_background(self, folders, force):
|
|
self.__daemon.scan(folders, force)
|
|
|
|
def __folder_scan_foreground(self, folders, force):
|
|
try:
|
|
progress = self.__daemon.get_scanning_progress()
|
|
if progress is not None:
|
|
self.write_error_line(
|
|
"The daemon is currently scanning, can't start a scan now"
|
|
)
|
|
return
|
|
except DaemonUnavailableError:
|
|
pass
|
|
|
|
extensions = self.__config.BASE["scanner_extensions"]
|
|
if extensions:
|
|
extensions = extensions.split(" ")
|
|
|
|
scanner = Scanner(
|
|
force=force,
|
|
extensions=extensions,
|
|
follow_symlinks=self.__config.BASE["follow_symlinks"],
|
|
progress=TimedProgressDisplay(self.stdout),
|
|
on_folder_start=self.__unwatch_folder,
|
|
on_folder_end=self.__watch_folder,
|
|
)
|
|
|
|
if folders:
|
|
fstrs = folders
|
|
with db_session:
|
|
folders = select(f.name for f in Folder if f.root and f.name in fstrs)[
|
|
:
|
|
]
|
|
notfound = set(fstrs) - set(folders)
|
|
if notfound:
|
|
self.write_line("No such folder(s): " + " ".join(notfound))
|
|
for folder in folders:
|
|
scanner.queue_folder(folder)
|
|
else:
|
|
with db_session:
|
|
for folder in select(f.name for f in Folder if f.root):
|
|
scanner.queue_folder(folder)
|
|
|
|
scanner.run()
|
|
stats = scanner.stats()
|
|
|
|
self.write_line("Scanning done")
|
|
self.write_line(
|
|
"Added: {0.artists} artists, {0.albums} albums, {0.tracks} tracks".format(
|
|
stats.added
|
|
)
|
|
)
|
|
self.write_line(
|
|
"Deleted: {0.artists} artists, {0.albums} albums, {0.tracks} tracks".format(
|
|
stats.deleted
|
|
)
|
|
)
|
|
if stats.errors:
|
|
self.write_line("Errors in:")
|
|
for err in stats.errors:
|
|
self.write_line("- " + err)
|
|
|
|
def __unwatch_folder(self, folder):
|
|
try:
|
|
self.__daemon.remove_watched_folder(folder.path)
|
|
except DaemonUnavailableError:
|
|
pass
|
|
|
|
def __watch_folder(self, folder):
|
|
try:
|
|
self.__daemon.add_watched_folder(folder.path)
|
|
except DaemonUnavailableError:
|
|
pass
|
|
|
|
user_parser = CLIParser(prog="user", add_help=False)
|
|
user_subparsers = user_parser.add_subparsers(dest="action")
|
|
user_subparsers.add_parser("list", help="List users", add_help=False)
|
|
user_add_parser = user_subparsers.add_parser(
|
|
"add", help="Adds a user", add_help=False
|
|
)
|
|
user_add_parser.add_argument("name", help="Name/login of the user to add")
|
|
user_add_parser.add_argument(
|
|
"-a", "--admin", action="store_true", help="Give admin rights to the new user"
|
|
)
|
|
user_add_parser.add_argument(
|
|
"-p", "--password", help="Specifies the user's password"
|
|
)
|
|
user_add_parser.add_argument(
|
|
"-e", "--email", default="", help="Sets the user's email address"
|
|
)
|
|
user_del_parser = user_subparsers.add_parser(
|
|
"delete", help="Deletes a user", add_help=False
|
|
)
|
|
user_del_parser.add_argument("name", help="Name/login of the user to delete")
|
|
user_admin_parser = user_subparsers.add_parser(
|
|
"setadmin", help="Enable/disable admin rights for a user", add_help=False
|
|
)
|
|
user_admin_parser.add_argument(
|
|
"name", help="Name/login of the user to grant/revoke admin rights"
|
|
)
|
|
user_admin_parser.add_argument(
|
|
"--off",
|
|
action="store_true",
|
|
help="Revoke admin rights if present, grant them otherwise",
|
|
)
|
|
user_pass_parser = user_subparsers.add_parser(
|
|
"changepass", help="Changes a user's password", add_help=False
|
|
)
|
|
user_pass_parser.add_argument(
|
|
"name", help="Name/login of the user to which change the password"
|
|
)
|
|
user_pass_parser.add_argument("password", nargs="?", help="New password")
|
|
|
|
@db_session
|
|
def user_list(self):
|
|
self.write_line("Name\t\tAdmin\tEmail\n----\t\t-----\t-----")
|
|
self.write_line(
|
|
"\n".join(
|
|
"{0: <16}{1}\t{2}".format(u.name, "*" if u.admin else "", u.mail)
|
|
for u in User.select()
|
|
)
|
|
)
|
|
|
|
def _ask_password(self): # pragma: nocover
|
|
password = getpass.getpass()
|
|
confirm = getpass.getpass("Confirm password: ")
|
|
if password != confirm:
|
|
raise ValueError("Passwords don't match")
|
|
return password
|
|
|
|
@db_session
|
|
def user_add(self, name, admin, password, email):
|
|
try:
|
|
if not password:
|
|
password = self._ask_password() # pragma: nocover
|
|
UserManager.add(name, password, email, admin)
|
|
except ValueError as e:
|
|
self.write_error_line(str(e))
|
|
|
|
@db_session
|
|
def user_delete(self, name):
|
|
try:
|
|
UserManager.delete_by_name(name)
|
|
self.write_line("Deleted user '{}'".format(name))
|
|
except ObjectNotFound as e:
|
|
self.write_error_line(str(e))
|
|
|
|
@db_session
|
|
def user_setadmin(self, name, off):
|
|
user = User.get(name=name)
|
|
if user is None:
|
|
self.write_error_line("No such user")
|
|
else:
|
|
user.admin = not off
|
|
self.write_line(
|
|
"{0} '{1}' admin rights".format("Revoked" if off else "Granted", name)
|
|
)
|
|
|
|
@db_session
|
|
def user_changepass(self, name, password):
|
|
try:
|
|
if not password:
|
|
password = self._ask_password() # pragma: nocover
|
|
UserManager.change_password2(name, password)
|
|
self.write_line("Successfully changed '{}' password".format(name))
|
|
except ObjectNotFound as e:
|
|
self.write_error_line(str(e))
|
|
|
|
|
|
def main():
|
|
config = IniConfig.from_common_locations()
|
|
init_database(config.BASE["database_uri"])
|
|
|
|
cli = SupysonicCLI(config)
|
|
if len(sys.argv) > 1:
|
|
cli.onecmd(" ".join(sys.argv[1:]))
|
|
else:
|
|
cli.cmdloop()
|
|
|
|
release_database()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|