dt: dt-extract-compatibles: Handle cfile arguments in generator function

Move the handling of the cfile arguments to a separate generator
function to avoid redundancy.

Signed-off-by: Nícolas F. R. A. Prado <nfraprado@collabora.com>
Link: https://lore.kernel.org/r/20230828211424.2964562-2-nfraprado@collabora.com
Signed-off-by: Rob Herring <robh@kernel.org>
This commit is contained in:
Nícolas F. R. A. Prado 2023-08-28 17:13:10 -04:00 committed by Rob Herring
parent 0bb80ecc33
commit eb2139fc0d

View File

@ -49,6 +49,14 @@ def print_compat(filename, compatibles):
else:
print(*compatibles, sep='\n')
def files_to_parse(path_args):
for f in path_args:
if os.path.isdir(f):
for filename in glob.iglob(f + "/**/*.c", recursive=True):
yield filename
else:
yield f
show_filename = False
if __name__ == "__main__":
@ -59,11 +67,6 @@ if __name__ == "__main__":
show_filename = args.with_filename
for f in args.cfile:
if os.path.isdir(f):
for filename in glob.iglob(f + "/**/*.c", recursive=True):
compat_list = parse_compatibles(filename)
print_compat(filename, compat_list)
else:
compat_list = parse_compatibles(f)
print_compat(f, compat_list)
for f in files_to_parse(args.cfile):
compat_list = parse_compatibles(f)
print_compat(f, compat_list)