From 6923bb47129e359fd63a459802e7fbfb35156e7f Mon Sep 17 00:00:00 2001 From: Vijay Kumar Banerjee Date: Fri, 2 Apr 2021 19:17:17 -0600 Subject: netlegacy: Use os.path to get the file extension --- bsp_drivers.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'bsp_drivers.py') diff --git a/bsp_drivers.py b/bsp_drivers.py index cb6caa5..3ca10c6 100644 --- a/bsp_drivers.py +++ b/bsp_drivers.py @@ -62,9 +62,10 @@ def bsp_files(bld): source_dir = os.walk(special_case_dirs[bsp]) for root, dirs, files in source_dir: for name in files: - if name[-2:] == '.c': + ext = os.path.splitext(name)[1] + if ext == '.c': source_files[bsp].append(os.path.join(root, name)) - if name[-2:] == '.h': + if ext == '.h': if root not in include_dirs[bsp]: include_dirs[bsp].append(root) if bsp in special_case_sources: -- cgit v1.2.3