From a0d07426641ce6902ce00b441d41f0807b891f25 Mon Sep 17 00:00:00 2001 From: Florent Kermarrec Date: Fri, 1 Aug 2014 12:50:38 +0200 Subject: [PATCH] mibuild/generic_platform: add recursive parameter to add_source_dir --- mibuild/generic_platform.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/mibuild/generic_platform.py b/mibuild/generic_platform.py index d972ee262..1ede87a4e 100644 --- a/mibuild/generic_platform.py +++ b/mibuild/generic_platform.py @@ -222,12 +222,20 @@ class GenericPlatform: for f in filenames: self.add_source(os.path.join(path, f), language) - def add_source_dir(self, path): - for root, dirs, files in os.walk(path): - for filename in files: - language = tools.language_by_filename(filename) - if language is not None: - self.add_source(os.path.join(root, filename), language) + def add_source_dir(self, path, recursive=True): + dir_files = [] + if recursive: + for root, dirs, files in os.walk(path): + for filename in files: + dir_files.append(os.path.join(root, filename)) + else: + for item in os.listdir(path): + if os.path.isfile(os.path.join(path, item)): + dir_files.append(os.path.join(path, item)) + for filename in dir_files: + language = tools.language_by_filename(filename) + if language is not None: + self.add_source(filename, language) def add_verilog_include_path(self, path): self.verilog_include_paths.append(os.path.abspath(path))