1
0
mirror of https://github.com/mikf/gallery-dl.git synced 2024-11-25 12:12:34 +01:00
gallery-dl/scripts/completion_zsh.py
Mike Fährmann 74494b43d3
let zsh completion immediately suggest cmdline options
instead of expecting an URL and trying to complete it.
2020-07-05 22:36:00 +02:00

51 lines
1.2 KiB
Python
Executable File

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# Copyright 2020 Mike Fährmann
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
"""Generate zsh completion script from gallery-dl's argument parser"""
import util
from gallery_dl import option
TEMPLATE = """#compdef gallery-dl
local curcontext="$curcontext"
typeset -A opt_args
local rc=1
_arguments -C -S \\
%(opts)s && rc=0
return rc
"""
opts = []
for action in option.build_parser()._actions:
if not action.option_strings:
continue
elif len(action.option_strings) == 1:
opt = action.option_strings[0]
else:
opt = "{" + ",".join(action.option_strings) + "}"
opt += "'[" + action.help.replace("'", '"') + "]'"
if action.metavar:
opt += ":'<" + action.metavar.lower() + ">'"
if action.metavar in ("FILE", "CFG", "DEST"):
opt += ":_files"
opts.append(opt)
PATH = util.path("data/completion/_gallery-dl")
with open(PATH, "w", encoding="utf-8") as file:
file.write(TEMPLATE % {"opts": " \\\n".join(opts)})