commit 55db207c1167b0b5a0b48b3023c9d30422adc654 Author: dave Date: Sat Aug 17 10:25:11 2019 -0700 initial move-only prototype diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..318db42 --- /dev/null +++ b/.gitignore @@ -0,0 +1,4 @@ +.DS_Store +__pycache__ +/*.egg-info/ +/virtualenv/ \ No newline at end of file diff --git a/mediaweb/__init__.py b/mediaweb/__init__.py new file mode 100644 index 0000000..080d688 --- /dev/null +++ b/mediaweb/__init__.py @@ -0,0 +1,124 @@ +import os +import cherrypy +import logging +from jinja2 import Environment, FileSystemLoader, select_autoescape +from deluge_client import DelugeRPCClient +from urllib.parse import urlparse + + +APPROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "../")) + + +class MediaWeb(object): + def __init__(self, rpc, templater, uioptions): + self.tpl = templater + self.rpc = rpc + self.uioptions = uioptions + + def render(self, template, **kwargs): + """ + Render a template + """ + return self.tpl.get_template(template).render(**kwargs, options=self.uioptions, **self.get_default_vars()) + + def get_default_vars(self): + return {} + + @cherrypy.expose + def index(self): + torrents = self.rpc.core.get_torrents_status({"label": "sickrage"}, ['name', 'label', 'save_path', 'is_seed', 'is_finished', 'progress']) + return self.render("index.html", torrents=torrents) + + @cherrypy.expose + def move(self, thash, dest=None, otherdest=None): + torrent = self.rpc.core.get_torrent_status(thash, []) # TODO reduce to needed fields + + if cherrypy.request.method == "POST" and (dest or otherdest): + target = otherdest or dest + self.rpc.core.move_storage([thash], target) + raise cherrypy.HTTPRedirect("/") + + return self.render("moveform.html", torrent=torrent) + + +def main(): + import argparse + import signal + + parser = argparse.ArgumentParser(description="mediaweb server") + + parser.add_argument('-p', '--port', help="tcp port to listen on", + default=int(os.environ.get("MEDIAWEB_PORT", 8080)), type=int) + parser.add_argument('-i', '--download-dirs', help="download directories", nargs="+", + default=os.environ.get("MEDIAWEB_DLDIR")) + parser.add_argument('-o', '--library', default=os.environ.get("STORAGE_URL"), help="media library path") + parser.add_argument('--debug', action="store_true", help="enable development options") + parser.add_argument('-s', '--server', help="deluge uris", action="append", required=True) + parser.add_argument('--ui-movedests', help="move destination options in the UI", nargs="+", required=True) + + args = parser.parse_args() + + uioptions = { + "movedests": args.ui_movedests, + } + + # TODO smarter argparser that understands env vars + if not args.library: + parser.error("--download-dirs or MEDIAWEB_DLDIR is required") + + logging.basicConfig(level=logging.INFO if args.debug else logging.WARNING, + format="%(asctime)-15s %(levelname)-8s %(filename)s:%(lineno)d %(message)s") + + tpl_dir = os.path.join(APPROOT, "templates") if not args.debug else "templates" + tpl = Environment(loader=FileSystemLoader(tpl_dir), + autoescape=select_autoescape(['html', 'xml'])) + # self.tpl.filters.update(basename=os.path.basename, + # ceil=math.ceil + + def validate_password(realm, user, passw): + return user == passw # lol + + # assume 1 deluge server for now + uri = urlparse(args.server[0]) + assert uri.scheme == "deluge" + rpc = DelugeRPCClient(uri.hostname, uri.port if uri.port else 58846, uri.username, uri.password, decode_utf8=True) + + web = MediaWeb(rpc, tpl, uioptions) + cherrypy.tree.mount(web, '/', {'/': {'tools.auth_basic.on': True, + 'tools.auth_basic.realm': 'mediaweb', + 'tools.auth_basic.checkpassword': validate_password, }}) + + # General config options + cherrypy.config.update({ + 'tools.sessions.on': False, + 'request.show_tracebacks': True, + 'server.show_tracebacks': True, + 'server.socket_port': args.port, + 'server.thread_pool': 1 if args.debug else 5, + 'server.socket_host': '0.0.0.0', + 'log.screen': False, + 'engine.autoreload.on': args.debug + }) + + # Setup signal handling and run it. + 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() + + +# https://github.com/deluge-torrent/deluge/blob/1.3-stable/deluge/ui/console/commands/info.py#L46 +# https://deluge.readthedocs.io/en/latest/reference/api.html?highlight=rpc diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..4eaefbd --- /dev/null +++ b/requirements.txt @@ -0,0 +1,13 @@ +backports.functools-lru-cache==1.5 +cheroot==6.5.5 +CherryPy==18.1.2 +deluge-client==1.7.1 +jaraco.functools==2.0 +Jinja2==2.10.1 +MarkupSafe==1.1.1 +more-itertools==7.2.0 +portend==2.5 +pytz==2019.2 +six==1.12.0 +tempora==1.14.1 +zc.lockfile==2.0 diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..692ad24 --- /dev/null +++ b/setup.py @@ -0,0 +1,25 @@ +#!/usr/bin/env python3 + +from setuptools import setup + + +__version__ = "0.0.0" + + +setup(name='mediaweb', + version=__version__, + description='web based tool for sorting your media in an opinionated way', + url='https://git.davepedu.com/dave/mediasort', + author='dpedu', + author_email='dave@davepedu.com', + packages=['mediaweb'], + install_requires=[], # TODO + entry_points={ + "console_scripts": [ + "mswebd = mediaweb:main" + ] + }, + include_package_data=True, + package_data={'mediaweb': ['../templates/*.html', + '../assets/*']}, + zip_safe=False) diff --git a/templates/index.html b/templates/index.html new file mode 100644 index 0000000..fb26960 --- /dev/null +++ b/templates/index.html @@ -0,0 +1,52 @@ +{% extends "page.html" %} +{% block body %} + +
+

Completed

+ + + + + + + + + + {% for torid, tor in torrents.items() %}{% if tor.is_finished %} + + + + + + + + + {% endif %}{% endfor %} +
hashnamepathdeststatusactions
{{ torid[0:6] }}{{ tor.name }}{{ tor.save_path }}x{{ "complete" if tor.is_finished else "pending" }} + + +
+

Pending

+ + + + + + + + + {% for torid, tor in torrents.items() %}{% if not tor.is_finished %} + + + + + + + + {% endif %}{% endfor %} +
hashnamepathpercentactions
{{ torid[0:6] }}{{ tor.name }}{{ tor.save_path }}{{ tor.progress }}% + +
+
+ +{% endblock %} \ No newline at end of file diff --git a/templates/moveform.html b/templates/moveform.html new file mode 100644 index 0000000..e7efabf --- /dev/null +++ b/templates/moveform.html @@ -0,0 +1,29 @@ +{% extends "page.html" %} +{% block body %} +
+

Move {{ torrent.name }}

+

{{ torrent.hash }}

+ +
+ current path + {{ torrent.save_path }} +
+ +
+ +
+ +
+ new path + + + +
+
+
+{% endblock %} \ No newline at end of file diff --git a/templates/page.html b/templates/page.html new file mode 100644 index 0000000..cdc18c3 --- /dev/null +++ b/templates/page.html @@ -0,0 +1,30 @@ + + + + + Media Sort + + + +
+{% block body %}{% endblock %} +
+ + \ No newline at end of file