From 8dc5fa49dc4c669c28ffb1216625ae92475c4ec9 Mon Sep 17 00:00:00 2001
From: Swapnil Negi <swapnil.negi09@gmail.com>
Date: Mon, 21 May 2018 12:50:35 +0530
Subject: utils: modtool: rename: Use SequenceCompleter to suggest blocknames
---
gr-utils/python/modtool/modtool_rename.py | 22 ++++++++++++++++++++++
1 file changed, 22 insertions(+)
(limited to 'gr-utils/python/modtool/modtool_rename.py')
diff --git a/gr-utils/python/modtool/modtool_rename.py b/gr-utils/python/modtool/modtool_rename.py
index 4af860b5d4..60460680e3 100644
--- a/gr-utils/python/modtool/modtool_rename.py
+++ b/gr-utils/python/modtool/modtool_rename.py
@@ -26,12 +26,29 @@ from __future__ import unicode_literals
import os
import re
+import glob
from .util_functions import append_re_line_sequence, ask_yes_no
+from .util_functions import SequenceCompleter
from .cmakefile_editor import CMakeFileEditor
from .modtool_base import ModTool, ModToolException
from .templates import Templates
+def get_block_candidates():
+ cpp_filters = ["*.cc", "*.cpp"]
+ cpp_blocks = []
+ for ftr in cpp_filters:
+ cpp_blocks += filter(lambda x: not (x.startswith('qa_') or
+ x.startswith('test_')),
+ glob.glob1("lib", ftr))
+ python_blocks = filter(lambda x: not (x.startswith('qa_') or
+ x.startswith('build') or
+ x.startswith('__init__')),
+ glob.glob1("python", "*.py"))
+ block_candidates = [x.split('_impl')[0] for x in cpp_blocks] + [x.split('.')[0] for x in python_blocks]
+ return block_candidates
+
+
class ModToolRename(ModTool):
""" Rename a block in the out-of-tree module. """
name = 'rename'
@@ -63,6 +80,11 @@ class ModToolRename(ModTool):
self._info['oldname'] = options.blockname
if self._info['oldname'] is None:
self._info['oldname'] = input("Enter name of block/code to rename (without module name prefix): ")
+ else:
+ block_candidates = get_block_candidates()
+ with SequenceCompleter(block_candidates):
+ self._info['oldname'] = \
+ raw_input("Enter name of block/code to rename (without module name prefix): ")
if not re.match('[a-zA-Z0-9_]+', self._info['oldname']):
raise ModToolException('Invalid block name.')
print("Block/code to rename identifier: " + self._info['oldname'])
--
cgit v1.2.3