From 8d976fe182232580613656aab99fc64953eeba83 Mon Sep 17 00:00:00 2001
From: Dan Buch <daniel.buch@gmail.com>
Date: Mon, 11 Jul 2016 06:02:15 -0400
Subject: [PATCH] Get package choices from source of truth

---
 generate-flag-types | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/generate-flag-types b/generate-flag-types
index e48e7fd..8567f5d 100755
--- a/generate-flag-types
+++ b/generate-flag-types
@@ -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(