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.
76 lines
2.2 KiB
Python
76 lines
2.2 KiB
Python
3 years ago
|
#!/usr/bin/env python3
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
from pathlib import Path
|
||
|
|
from string import Template
|
||
|
|
import subprocess
|
||
|
|
from typing import Callable, Dict
|
||
|
|
import urllib.parse as url
|
||
|
|
|
||
|
|
|
||
|
|
def cmd_player_play(video_uri: str):
|
||
|
|
print(f"Play video {video_uri}")
|
||
|
|
subprocess.Popen([
|
||
|
|
str(Path("~/bin/mpvadd").expanduser()), video_uri,
|
||
|
|
], 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()
|