#!/usr/bin/env python3 import argparse from pathlib import Path from string import Template import subprocess from typing import Callable, Dict, Optional import urllib.parse as url def cmd_player_play(video_uri: str, start: Optional[str] = None): print(f"Play video {video_uri}") subprocess.Popen( args = [e for e in [ str(Path("~/bin/mpvadd").expanduser()), video_uri, f"start={start}" if start is not None else None, ] if e is not None], stdin = subprocess.DEVNULL, stdout = subprocess.DEVNULL, stderr = subprocess.DEVNULL, ) URI_SCHEME = "entertainment-decider" URI_COMMANDS = { "player": { "play": cmd_player_play } } def execute_uri_command(uri: str): parts = url.urlparse(uri, scheme=URI_SCHEME, allow_fragments=False) if parts.scheme != URI_SCHEME: if parts.scheme in {"http", "https"}: return cmd_player_play(uri) raise Exception(f"Cannot parse URI's with scheme {parts.scheme!r}") path = parts.path.strip("/").split("/") options = dict(url.parse_qsl(parts.query)) def unknown_cmd(): raise Exception(f"Unknown command {parts.path}") current = URI_COMMANDS for path_name in path: if callable(current) or path_name not in current: unknown_cmd() current = current[path_name] if not callable(current): unknown_cmd() return current(**options) def misc_generate_desktop(): with Path("./entry.desktop").open("r") as fh: temp = Template(fh.read()) print(temp.substitute(name="Entertainment Decider", exec_path=str(Path(__file__).resolve()))) MISC_COMMANDS: Dict[str, Callable] = { "generate-desktop-file": misc_generate_desktop, } def execute_misc_cmd(cmd: str): if cmd not in MISC_COMMANDS: raise Exception(f"Unknown misc command {cmd!r}") return MISC_COMMANDS[cmd]() def main(): parser = argparse.ArgumentParser(prog="entertainment-decider") subparsers = parser.add_subparsers() # uri parser uri_parser = subparsers.add_parser("exec") uri_parser.add_argument("uri", type=str) uri_parser.set_defaults(parser_cmd=execute_uri_command) # misc parser misc_parser = subparsers.add_parser("misc") misc_parser.add_argument("cmd", type=str, choices=list(MISC_COMMANDS.keys())) misc_parser.set_defaults(parser_cmd=execute_misc_cmd) # parse args args = parser.parse_args() cmd = args.parser_cmd d = vars(args) del d["parser_cmd"] cmd(**d) if __name__ == "__main__": main()