[rtems-net-legacy commit] netlegacy: Use os.path to get the file extension

[Vijay Kumar Banerjee]

Module:    rtems-net-legacy
Branch:    main
Commit:    6923bb47129e359fd63a459802e7fbfb35156e7f
Changeset: http://git.rtems.org/rtems-net-legacy/commit/?id=6923bb47129e359fd63a459802e7fbfb35156e7f

Author:    Vijay Kumar Banerjee <vijay at rtems.org>
Date:      Fri Apr  2 19:17:17 2021 -0600

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))