72 lines
1.8 KiB
Python
72 lines
1.8 KiB
Python
|
|
import cadquery
|
|
from .shapes import Shape
|
|
|
|
import FreeCAD
|
|
import Part
|
|
import sys
|
|
import os
|
|
import urllib as urlreader
|
|
import tempfile
|
|
|
|
class ImportTypes:
|
|
STEP = "STEP"
|
|
|
|
class UNITS:
|
|
MM = "mm"
|
|
IN = "in"
|
|
|
|
|
|
def importShape(importType, fileName):
|
|
"""
|
|
Imports a file based on the type (STEP, STL, etc)
|
|
:param importType: The type of file that we're importing
|
|
:param fileName: THe name of the file that we're importing
|
|
"""
|
|
|
|
#Check to see what type of file we're working with
|
|
if importType == ImportTypes.STEP:
|
|
return importStep(fileName)
|
|
|
|
|
|
#Loads a STEP file into a CQ.Workplane object
|
|
def importStep(fileName):
|
|
"""
|
|
Accepts a file name and loads the STEP file into a cadquery shape
|
|
:param fileName: The path and name of the STEP file to be imported
|
|
"""
|
|
#Now read and return the shape
|
|
try:
|
|
#print fileName
|
|
rshape = Part.read(fileName)
|
|
|
|
#Make sure that we extract all the solids
|
|
solids = []
|
|
for solid in rshape.Solids:
|
|
solids.append(Shape.cast(solid))
|
|
|
|
return cadquery.Workplane("XY").newObject(solids)
|
|
except:
|
|
raise ValueError("STEP File Could not be loaded")
|
|
|
|
#Loads a STEP file from an URL into a CQ.Workplane object
|
|
def importStepFromURL(url):
|
|
#Now read and return the shape
|
|
try:
|
|
webFile = urlreader.urlopen(url)
|
|
tempFile = tempfile.NamedTemporaryFile(suffix='.step', delete=False)
|
|
tempFile.write(webFile.read())
|
|
webFile.close()
|
|
tempFile.close()
|
|
|
|
rshape = Part.read(tempFile.name)
|
|
|
|
#Make sure that we extract all the solids
|
|
solids = []
|
|
for solid in rshape.Solids:
|
|
solids.append(Shape.cast(solid))
|
|
|
|
return cadquery.Workplane("XY").newObject(solids)
|
|
except:
|
|
raise ValueError("STEP File from the URL: " + url + " Could not be loaded")
|