diff options
| author | Tim Kuester <tpkuester@gmail.com> | 2015-01-21 16:59:12 -0500 |
|---|---|---|
| committer | Tim Kuester <tpkuester@gmail.com> | 2015-01-21 16:59:12 -0500 |
| commit | 10ce6739ea17a3a9de31e10cc5464007e0750d72 (patch) | |
| tree | 8eba4d52164b160019f8733a4775db7baebb54c5 /gr-utils/python | |
| parent | 35d62c6504a0bd523bfa167d36b5c3808dbe54e6 (diff) | |
Added interactive language selection, renamed "code type" to "block
type" to be more descriptive.
Diffstat (limited to 'gr-utils/python')
| -rw-r--r-- | gr-utils/python/modtool/modtool_add.py | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/gr-utils/python/modtool/modtool_add.py b/gr-utils/python/modtool/modtool_add.py index e188b7fd71..a4812c8105 100644 --- a/gr-utils/python/modtool/modtool_add.py +++ b/gr-utils/python/modtool/modtool_add.py @@ -62,7 +62,7 @@ class ModToolAdd(ModTool): ogroup.add_option("--skip-cmakefiles", action="store_true", default=False, help="If given, only source files are written, but CMakeLists.txt files are left unchanged.") ogroup.add_option("-l", "--lang", type="choice", choices=('cpp', 'c++', 'python'), - default='cpp', help="Language (cpp or python)") + default=None, help="Language (cpp or python)") parser.add_option_group(ogroup) return parser @@ -71,13 +71,22 @@ class ModToolAdd(ModTool): self._info['blocktype'] = options.block_type if self._info['blocktype'] is None: + # Print list out of blocktypes to user for reference + print str(self._block_types) while self._info['blocktype'] not in self._block_types: - self._info['blocktype'] = raw_input("Enter code type: ") + self._info['blocktype'] = raw_input("Enter block type: ") if self._info['blocktype'] not in self._block_types: print 'Must be one of ' + str(self._block_types) + + # Allow user to specify language interactively if not set self._info['lang'] = options.lang - if self._info['lang'] == 'c++': - self._info['lang'] = 'cpp' + if self._info['lang'] is None: + while self._info['lang'] not in ['cpp', 'python']: + self._info['lang'] = raw_input("Language (python/cpp): ") + + if self._info['lang'] == 'c++': + self._info['lang'] = 'cpp' + print "Language: %s" % {'cpp': 'C++', 'python': 'Python'}[self._info['lang']] if ((self._skip_subdirs['lib'] and self._info['lang'] == 'cpp') |