You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

83 lines
2.4 KiB
Python

#!/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:
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()