LegalizeAdulthood created this revision.
LegalizeAdulthood added a reviewer: alexfh.
Herald added a subscriber: xazax.hun.
Herald added a reviewer: serge-sans-paille.
Break up the huge main() function
Repository:
rCTE Clang Tools Extra
https://reviews.llvm.org/D56343
Files:
test/clang-tidy/check_clang_tidy.py
Index: test/clang-tidy/check_clang_tidy.py
===================================================================
--- test/clang-tidy/check_clang_tidy.py
+++ test/clang-tidy/check_clang_tidy.py
@@ -42,33 +42,7 @@
def csv(string):
return string.split(',')
-def main():
- parser = argparse.ArgumentParser()
- parser.add_argument('-expect-clang-tidy-error', action='store_true')
- parser.add_argument('-resource-dir')
- parser.add_argument('-assume-filename')
- parser.add_argument('input_file_name')
- parser.add_argument('check_name')
- parser.add_argument('temp_file_name')
- parser.add_argument('-check-suffix', '-check-suffixes',
- default=[''], type=csv,
- help="comma-separated list of FileCheck suffixes")
-
- args, extra_args = parser.parse_known_args()
-
- resource_dir = args.resource_dir
- assume_file_name = args.assume_filename
- input_file_name = args.input_file_name
- check_name = args.check_name
- temp_file_name = args.temp_file_name
- expect_clang_tidy_error = args.expect_clang_tidy_error
-
- file_name_with_extension = assume_file_name or input_file_name
- _, extension = os.path.splitext(file_name_with_extension)
- if extension not in ['.c', '.hpp', '.m', '.mm']:
- extension = '.cpp'
- temp_file_name = temp_file_name + extension
-
+def get_extra_args(resource_dir, extension, extra_args):
clang_tidy_extra_args = extra_args
if len(clang_tidy_extra_args) == 0:
clang_tidy_extra_args = ['--']
@@ -85,9 +59,9 @@
if resource_dir is not None:
clang_tidy_extra_args.append('-resource-dir=%s' % resource_dir)
- with open(input_file_name, 'r') as input_file:
- input_text = input_file.read()
+ return clang_tidy_extra_args
+def get_prefixes(check_suffix, input_text):
check_fixes_prefixes = []
check_messages_prefixes = []
check_notes_prefixes = []
@@ -96,7 +70,7 @@
has_check_messages = False
has_check_notes = False
- for check in args.check_suffix:
+ for check in check_suffix:
if check and not re.match('^[A-Z0-9\-]+$', check):
sys.exit('Only A..Z, 0..9 and "-" are ' +
'allowed in check suffixes list, but "%s" was given' % (check))
@@ -126,56 +100,23 @@
check_messages_prefixes.append(check_messages_prefix)
check_notes_prefixes.append(check_notes_prefix)
- assert has_check_fixes or has_check_messages or has_check_notes
- # Remove the contents of the CHECK lines to avoid CHECKs matching on
- # themselves. We need to keep the comments to preserve line numbers while
- # avoiding empty lines which could potentially trigger formatting-related
- # checks.
- cleaned_test = re.sub('// *CHECK-[A-Z0-9\-]*:[^\r\n]*', '//', input_text)
-
- write_file(temp_file_name, cleaned_test)
-
- original_file_name = temp_file_name + ".orig"
- write_file(original_file_name, cleaned_test)
-
- args = ['clang-tidy', temp_file_name, '-fix', '--checks=-*,' + check_name] + \
- clang_tidy_extra_args
- if expect_clang_tidy_error:
- args.insert(0, 'not')
- print('Running ' + repr(args) + '...')
- try:
- clang_tidy_output = \
- subprocess.check_output(args, stderr=subprocess.STDOUT).decode()
- except subprocess.CalledProcessError as e:
- print('clang-tidy failed:\n' + e.output.decode())
- raise
-
- print('------------------------ clang-tidy output -----------------------\n' +
- clang_tidy_output +
- '\n------------------------------------------------------------------')
-
- try:
- diff_output = subprocess.check_output(
- ['diff', '-u', original_file_name, temp_file_name],
- stderr=subprocess.STDOUT)
- except subprocess.CalledProcessError as e:
- diff_output = e.output
-
- print('------------------------------ Fixes -----------------------------\n' +
- diff_output.decode() +
- '\n------------------------------------------------------------------')
+ return (has_check_fixes, has_check_messages, has_check_notes, \
+ check_fixes_prefixes, check_messages_prefixes, check_notes_prefixes)
+def check_fixes(check_fixes_prefixes, has_check_fixes, input_file_name, temp_file_name):
if has_check_fixes:
try:
subprocess.check_output(
['FileCheck', '-input-file=' + temp_file_name, input_file_name,
- '-check-prefixes=' + ','.join(check_fixes_prefixes),
- '-strict-whitespace'],
+ '-check-prefixes=' + ','.join(check_fixes_prefixes),
+ '-strict-whitespace'],
stderr=subprocess.STDOUT)
except subprocess.CalledProcessError as e:
print('FileCheck failed:\n' + e.output.decode())
raise
+ return
+def check_messages(check_messages_prefixes, has_check_messages, clang_tidy_output, input_file_name, temp_file_name):
if has_check_messages:
messages_file = temp_file_name + '.msg'
write_file(messages_file, clang_tidy_output)
@@ -188,7 +129,9 @@
except subprocess.CalledProcessError as e:
print('FileCheck failed:\n' + e.output.decode())
raise
+ return
+def check_notes(check_notes_prefixes, has_check_notes, clang_tidy_output, input_file_name, temp_file_name):
if has_check_notes:
notes_file = temp_file_name + '.notes'
filtered_output = [line for line in clang_tidy_output.splitlines()
@@ -203,6 +146,85 @@
except subprocess.CalledProcessError as e:
print('FileCheck failed:\n' + e.output.decode())
raise
+ return
+
+def main():
+ parser = argparse.ArgumentParser()
+ parser.add_argument('-expect-clang-tidy-error', action='store_true')
+ parser.add_argument('-resource-dir')
+ parser.add_argument('-assume-filename')
+ parser.add_argument('input_file_name')
+ parser.add_argument('check_name')
+ parser.add_argument('temp_file_name')
+ parser.add_argument('-check-suffix', '-check-suffixes',
+ default=[''], type=csv,
+ help="comma-separated list of FileCheck suffixes")
+
+ args, extra_args = parser.parse_known_args()
+
+ resource_dir = args.resource_dir
+ assume_file_name = args.assume_filename
+ input_file_name = args.input_file_name
+ check_name = args.check_name
+ temp_file_name = args.temp_file_name
+ expect_clang_tidy_error = args.expect_clang_tidy_error
+
+ file_name_with_extension = assume_file_name or input_file_name
+ _, extension = os.path.splitext(file_name_with_extension)
+ if extension not in ['.c', '.hpp', '.m', '.mm']:
+ extension = '.cpp'
+ temp_file_name = temp_file_name + extension
+
+ clang_tidy_extra_args = get_extra_args(resource_dir, extension, extra_args)
+
+ with open(input_file_name, 'r') as input_file:
+ input_text = input_file.read()
+
+ has_check_fixes, has_check_messages, has_check_notes, \
+ check_fixes_prefixes, check_messages_prefixes, check_notes_prefixes = \
+ get_prefixes(args.check_suffix, input_text)
+
+ assert has_check_fixes or has_check_messages or has_check_notes
+
+ # Remove the contents of the CHECK lines to avoid CHECKs matching on
+ # themselves. We need to keep the comments to preserve line numbers while
+ # avoiding empty lines which could potentially trigger formatting-related
+ # checks.
+ cleaned_test = re.sub('// *CHECK-[A-Z0-9\-]*:[^\r\n]*', '//', input_text)
+ write_file(temp_file_name, cleaned_test)
+ original_file_name = temp_file_name + ".orig"
+ write_file(original_file_name, cleaned_test)
+
+ args = ['clang-tidy', temp_file_name, '-fix', '--checks=-*,' + check_name] + \
+ clang_tidy_extra_args
+ if expect_clang_tidy_error:
+ args.insert(0, 'not')
+ print('Running ' + repr(args) + '...')
+ try:
+ clang_tidy_output = \
+ subprocess.check_output(args, stderr=subprocess.STDOUT).decode()
+ except subprocess.CalledProcessError as e:
+ print('clang-tidy failed:\n' + e.output.decode())
+ raise
+
+ print('------------------------ clang-tidy output -----------------------\n' +
+ clang_tidy_output +
+ '\n------------------------------------------------------------------')
+
+ try:
+ diff_output = subprocess.check_output(
+ ['diff', '-u', original_file_name, temp_file_name],
+ stderr=subprocess.STDOUT)
+ except subprocess.CalledProcessError as e:
+ diff_output = e.output
+
+ print('------------------------------ Fixes -----------------------------\n' +
+ diff_output.decode() +
+ '\n------------------------------------------------------------------')
+
+ check_fixes(check_fixes_prefixes, has_check_fixes, input_file_name, temp_file_name)
+ check_messages(check_messages_prefixes, has_check_messages, clang_tidy_output, input_file_name, temp_file_name)
+ check_notes(check_notes_prefixes, has_check_notes, clang_tidy_output, input_file_name, temp_file_name)
if __name__ == '__main__':
main()
_______________________________________________
cfe-commits mailing list
cfe-commits@lists.llvm.org
http://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
