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.
85 lines
2.5 KiB
Python
85 lines
2.5 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:
|
|
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()
|