From 67224dcd8e78395b804577257e37d4e7b5ac6591 Mon Sep 17 00:00:00 2001 From: lixinyu Date: Tue, 16 Jul 2024 14:59:08 +0800 Subject: [PATCH] samples: optimize the samples list * traverse the samples directory through recursive method Signed-off-by: lixinyu --- src/oebuild/app/plugins/samples/samples.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/src/oebuild/app/plugins/samples/samples.py b/src/oebuild/app/plugins/samples/samples.py index e897ff8..7f2f4b3 100644 --- a/src/oebuild/app/plugins/samples/samples.py +++ b/src/oebuild/app/plugins/samples/samples.py @@ -65,10 +65,21 @@ class Samples(OebuildCommand): self.do_exec(samples=samples) def _get_samples(self): - list_samples = os.listdir(self.configure.yocto_samples_dir()) + list_samples = [] + + def recursive_listdir(path): + files = os.listdir(path) + for file in files: + file_path = os.path.join(path, file) + if os.path.isfile(file_path): + list_samples.append(file_path) + if os.path.isdir(file_path): + recursive_listdir(file_path) + recursive_listdir(self.configure.yocto_samples_dir()) + res = {} for index, sample in enumerate(list_samples): - res[str(index + 1)] = sample + res[str(index + 1)] = sample.replace(self.configure.yocto_samples_dir(), "").lstrip("/") return res def do_exec(self, samples: dict): -- Gitee