Merge pull request #49 from cyberbotics/fix-uppercase-extension-compa…

…rison Fix Case Sensitive File Extension Comparison
cyberbotics · Jul 8, 2020 · 0d56f57 · 0d56f57
2 parents b246949 + 3f31483
commit 0d56f57
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/urdf2webots/parserURDF.py b/urdf2webots/parserURDF.py
@@ -702,7 +702,7 @@ def getVisual(link, node, path):
             extension = os.path.splitext(meshfile)[1].lower()
             if extension == '.dae':
                 getColladaMesh(meshfile, visual, link)
-            elif extension == '.stl' or extension == '.obj':
+            elif extension in ['.stl', '.obj']:
                 name = os.path.splitext(os.path.basename(meshfile))[0]
                 if name in Geometry.reference:
                     visual.geometry = Geometry.reference[name]
@@ -761,10 +761,10 @@ def getCollision(link, node, path):
             if meshfile.count('package'):
                 idx0 = meshfile.find('package://')
                 meshfile = meshfile[idx0 + len('package://'):]
-            extension = os.path.splitext(meshfile)[1]
+            extension = os.path.splitext(meshfile)[1].lower()
             if extension == '.dae':
                 collision.geometry.collada = getColladaMesh(meshfile, collision, link)
-            elif extension == '.stl' or extension == '.obj':
+            elif extension in ['.stl', '.obj']:
                 name = os.path.splitext(os.path.basename(meshfile))[0]
                 if name in Geometry.reference:
                     collision.geometry = Geometry.reference[name]