esp-idf/tools/ci/check_tools_files_patterns.py

91 lines
3.4 KiB
Python
Executable File

#!/usr/bin/env python
#
# SPDX-FileCopyrightText: 2021 Espressif Systems (Shanghai) CO LTD
# SPDX-License-Identifier: Apache-2.0
import argparse
import fnmatch
import glob
import os
import sys
import yaml
from idf_ci_utils import IDF_PATH, get_git_files, magic_check, magic_check_bytes, translate
# Monkey patch starts
# glob.glob will ignore all files starts with ``.``
# don't ignore them here
# need to keep the same argument as glob._ishidden
def _ishidden(path): # pylint: disable=W0613
return False
fnmatch.translate = translate
glob.magic_check = magic_check # type: ignore
glob.magic_check_bytes = magic_check_bytes # type: ignore
glob._ishidden = _ishidden # type: ignore # pylint: disable=W0212
# ends here
def check(pattern_yml, exclude_list):
rules_dict = yaml.load(open(pattern_yml), Loader=yaml.FullLoader)
rules_patterns_set = set()
for k, v in rules_dict.items():
if k.startswith('.pattern') and isinstance(v, list):
rules_patterns_set.update(v)
rules_files_set = set()
for pat in rules_patterns_set:
rules_files_set.update(glob.glob(os.path.join(IDF_PATH, pat), recursive=True))
exclude_patterns_set = set()
exclude_patterns_set.update([path.split('#')[0].strip() for path in open(exclude_list).readlines() if path])
exclude_files_set = set()
for pat in exclude_patterns_set:
exclude_files_set.update(glob.glob(os.path.join(IDF_PATH, pat), recursive=True))
missing_files = set()
git_files = get_git_files(os.path.join(IDF_PATH, 'tools'), full_path=True)
for f in git_files:
if f in rules_files_set or f in exclude_files_set:
continue
missing_files.add(os.path.relpath(f, IDF_PATH))
return missing_files, rules_patterns_set.intersection(exclude_patterns_set)
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='check if all tools files are in rules patterns or exclude list')
parser.add_argument('-c', '--pattern-yml',
default=os.path.join(IDF_PATH, '.gitlab', 'ci', 'rules.yml'),
help='yml file path included file patterns')
parser.add_argument('-e', '--exclude-list',
default=os.path.join(IDF_PATH, 'tools', 'ci', 'exclude_check_tools_files.txt'),
help='exclude list path')
args = parser.parse_args()
res = 0
not_included_files, dup_patterns = check(args.pattern_yml, args.exclude_list)
if not_included_files or dup_patterns:
res = 1
print('This test is used for making sure of all the tools dir files are recorded in .gitlab/ci/rules.yml to '
'trigger the related tests, except those files should be excluded.')
if not_included_files:
print('Missing Files:')
for file in not_included_files:
print('\t' + file)
print('Please add these files or glob patterns to ".gitlab/ci/rules.yml" and put related files under '
'".patterns-<test_group>" block to trigger related tests.\n'
'Or add them to "tools/ci/exclude_check_tools_files.txt" to exclude them.')
if dup_patterns:
print('Duplicated Patterns:')
for pattern in dup_patterns:
print('\t' + pattern)
print('Please remove them from tools/ci/exclude_check_tools_files.txt')
sys.exit(res)