devscripts/bash-completion.py

   1 #!/usr/bin/env python3
   2 import os
   3 import sys
   4
   5 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
   6
   7 import yt_dlp
   8
   9 BASH_COMPLETION_FILE = "completions/bash/yt-dlp"
  10 BASH_COMPLETION_TEMPLATE = "devscripts/bash-completion.in"
  11
  12
  13 def build_completion(opt_parser):
  14     opts_flag = []
  15     for group in opt_parser.option_groups:
  16         for option in group.option_list:
  17             # for every long flag
  18             opts_flag.append(option.get_opt_string())
  19     with open(BASH_COMPLETION_TEMPLATE) as f:
  20         template = f.read()
  21     with open(BASH_COMPLETION_FILE, "w") as f:
  22         # just using the special char
  23         filled_template = template.replace("{{flags}}", " ".join(opts_flag))
  24         f.write(filled_template)
  25
  26
  27 parser = yt_dlp.parseOpts(ignore_config_files=True)[0]
  28 build_completion(parser)