#!/usr/bin/env python # PYTHON_ARGCOMPLETE_OK import argparse from argcomplete import autocomplete from primitive_roots import primitive_roots parser = argparse.ArgumentParser(prog="prim-wallpaper") parser.add_argument("prim1", type=int, choices=primitive_roots) parser.add_argument("prim3", type=int, choices=primitive_roots) parser.add_argument("prim2", type=int, choices=primitive_roots) autocomplete(parser) args = parser.parse_args()