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 +++-- netlegacy.py | 5 +++-- 2 files changed, 6 insertions(+), 4 deletions(-) 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: diff --git a/netlegacy.py b/netlegacy.py index 0889548..89176e6 100644 --- a/netlegacy.py +++ b/netlegacy.py @@ -40,9 +40,10 @@ for root, dirs, files in os.walk("."): dirs.append('./bsps/shared/net') include_files[root[2:]] = [] for name in files: - if name[-2:] == '.c': + ext = os.path.splitext(name)[1] + if ext == '.c': source_files.append(os.path.join(root, name)) - if name[-2:] == '.h' and name not in exclude_headers: + if ext == '.h' and name not in exclude_headers: include_files[root[2:]].append(os.path.join(root, name)) -- cgit v1.2.3