fix mesh extension bug (#305)

kscalelabs · Aug 20, 2024 · b667bf0 · b667bf0
1 parent 1cf5bf2
commit b667bf0
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/store/app/crud/urdf.py b/store/app/crud/urdf.py
@@ -88,7 +88,6 @@ async def set_urdf(
                     tmesh = trimesh.load(data, file_type=suffix)
                     assert isinstance(tmesh, trimesh.Trimesh)
                 except Exception:
-                    print(data.read())
                     raise BadArtifactError(f"Invalid mesh file: {name}")
                 meshes.append((name, tmesh))
 
@@ -106,7 +105,6 @@ async def set_urdf(
                 raise BadArtifactError("Mesh element missing filename attribute.")
             filepath = Path(filename).relative_to(".")
             if filepath not in mesh_names:
-                print(mesh_names)
                 raise BadArtifactError(f"Mesh referenced in URDF was not uploaded: {filepath}")
             mesh_names.remove(filepath)
             mesh.set("filename", str(filepath.with_suffix(".obj")))
@@ -119,7 +117,8 @@ async def set_urdf(
             for name, tmesh in meshes:
                 out_file = io.BytesIO()
                 tmesh.export(out_file, file_type="obj")
-                info = tarfile.TarInfo(name)
+                obj_name = Path(name).with_suffix(".obj").as_posix()
+                info = tarfile.TarInfo(obj_name)
                 info.size = out_file.tell()
                 out_file.seek(0)
                 tar.addfile(info, out_file)