mirror of
https://github.com/spl0k/supysonic.git
synced 2024-09-19 19:01:03 +00:00
86 lines
2.5 KiB
Python
Executable File
86 lines
2.5 KiB
Python
Executable File
# coding: utf-8
|
|
|
|
import sys, cmd
|
|
import config
|
|
|
|
class CLI(cmd.Cmd):
|
|
prompt = "supysonic> "
|
|
|
|
def do_EOF(self, line):
|
|
return True
|
|
|
|
def default(self, line):
|
|
print 'Unknown command %s' % line.split()[0]
|
|
self.do_help(None)
|
|
|
|
def postloop(self):
|
|
print
|
|
|
|
def do_folder(self, line):
|
|
action = line.split()[0] if line else 'list'
|
|
args = line.split()[1:] if line else None
|
|
|
|
if action == 'list':
|
|
print 'Name\t\tPath\n----\t\t----'
|
|
print '\n'.join('%s\t\t%s' % (f.name, f.path) for f in db.Folder.query.filter(db.Folder.root == True))
|
|
elif action == 'add':
|
|
if len(args) < 2:
|
|
print 'Missing argument. folder add <name> <path>'
|
|
else:
|
|
ret = FolderManager.add(args[0], args[1])
|
|
if ret != FolderManager.SUCCESS:
|
|
print FolderManager.error_str(ret)
|
|
else:
|
|
print "Folder '%s' added" % args[0]
|
|
elif action == 'delete':
|
|
if len(args) < 1:
|
|
print 'Missing argument. folder delete <name>'
|
|
else:
|
|
ret = FolderManager.delete_by_name(args[0])
|
|
if ret != FolderManager.SUCCESS:
|
|
print FolderManager.error_str(ret)
|
|
else:
|
|
print "Deleted folder '%s'" % args[0]
|
|
elif action == 'scan':
|
|
s = Scanner(db.session)
|
|
if args:
|
|
folders = map(lambda n: db.Folder.query.filter(db.Folder.name == n and db.Folder.root == True).first() or n, args)
|
|
if any(map(lambda f: isinstance(f, basestring), folders)):
|
|
print "No such folder(s): " + ' '.join(f for f in folders if isinstance(f, basestring))
|
|
for folder in filter(lambda f: isinstance(f, db.Folder), folders):
|
|
FolderManager.scan(folder.id, s)
|
|
else:
|
|
for folder in db.Folder.query.filter(db.Folder.root == True):
|
|
FolderManager.scan(folder.id, s)
|
|
|
|
added, deleted = s.stats()
|
|
db.session.commit()
|
|
|
|
print "Scanning done"
|
|
print 'Added: %i artists, %i albums, %i tracks' % (added[0], added[1], added[2])
|
|
print 'Deleted: %i artists, %i albums, %i tracks' % (deleted[0], deleted[1], deleted[2])
|
|
else:
|
|
print "Unknown action '%s'" % action
|
|
|
|
def help_folder(self):
|
|
print "folder\nfolder list\n\tDisplays the list of folders"
|
|
print "folder add <name> <path>\n\tAdds a music folder 'name' pointing to folder 'path'"
|
|
print "folder delete <name>\n\tDeletes folder 'name'"
|
|
print "folder scan [name [name [...]]]\n\tRuns a scan on folder 'name'. If 'name' is ommited, all folders\n\tare scanned"
|
|
|
|
if __name__ == "__main__":
|
|
if not config.check():
|
|
sys.exit(1)
|
|
|
|
import db
|
|
db.init_db()
|
|
|
|
from managers.folder import FolderManager
|
|
from scanner import Scanner
|
|
|
|
if len(sys.argv) > 1:
|
|
CLI().onecmd(' '.join(sys.argv[1:]))
|
|
else:
|
|
CLI().cmdloop()
|
|
|