Extract common 3D colors into cadquery_helpers

LibrePCB · Sep 15, 2023 · f28a2d6 · f28a2d6
1 parent 23175f7
commit f28a2d6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/cadquery_helpers.py b/cadquery_helpers.py
@@ -4,6 +4,10 @@
 from OCP.Message import Message, Message_Gravity  # type: ignore
 
 
+class StepColor:
+    LEAD_SMT = cq.Color('gainsboro')
+
+
 class StepAssembly:
     """
     A STEP assembly.

diff --git a/generate_chip.py b/generate_chip.py
@@ -633,7 +633,7 @@ def generate_3d(
 ) -> None:
     import cadquery as cq
 
-    from cadquery_helpers import StepAssembly
+    from cadquery_helpers import StepAssembly, StepColor
 
     print(f'Generating pkg 3D model "{full_name}": {uuid_3d}')
 
@@ -675,8 +675,8 @@ def generate_3d(
 
     assembly = StepAssembly(full_name)
     assembly.add_body(inner, 'inner', inner_color)
-    assembly.add_body(left, 'left', cq.Color("gainsboro"))
-    assembly.add_body(right, 'right', cq.Color("gainsboro"))
+    assembly.add_body(left, 'left', StepColor.LEAD_SMT)
+    assembly.add_body(right, 'right', StepColor.LEAD_SMT)
 
     out_path = path.join('out', library, 'pkg', uuid_pkg, f'{uuid_3d}.step')
     assembly.save(out_path)