Get package choices from source of truth

8 years ago · 8d976fe182
parent c3b03b8437
commit 8d976fe182
1 changed files with 1 additions and 1 deletions
--- a/2
+++ b/2
@ -78,7 +78,7 @@ def main(sysargs=sys.argv[:]):
        formatter_class=_FancyFormatter)
    parser.add_argument(
        'package',
-        type=str, default='cli', choices=['cli', 'altsrc'],
+        type=str, default='cli', choices=_WRITEFUNCS.keys(),
        help='Package for which flag types will be generated'
    )
    parser.add_argument(