def do_POST(self):
content_length = int(self.headers['Content-Length'])
body = self.rfile.read(content_length)
unquoted_s = ""
if six.PY2:
import urllib
unquoted_s = urllib.unquote(body)
elif six.PY3:
unquoted_s = body.decode('utf-8')
data = json.loads(unquoted_s)
from onshape_client.oas.models import BTDocumentParams
bt_document_params = BTDocumentParams(name=data["doc_name"])
new_doc_response = Client.get_client().documents_api.create11(
bt_document_params, _preload_content=False)
did = get_field(new_doc_response, "id")
wid = get_field(new_doc_response, "defaultWorkspace")["id"]
# Use a fake eid because it isn't used later.
eid = "00000000000000"
self.onshape_element = OnshapeElement.create_from_ids(
did, "w", wid, eid)
if "import_items" in data:
for import_item in data["import_items"]:
self.import_item(import_item)
self.send_response(200)
self.send_header('Content-type', 'application/json')
self.send_header('Access-Control-Allow-Origin', '*')
self.end_headers()
content = {
"document_href": self.onshape_element.get_url(url_type="document")
}
self.wfile.write(sendable(json.dumps(content)))
def import_item(self, import_item):
did = self.onshape_element.did
wid = self.onshape_element.wvmid
path = write_to_file(import_item["file"])
eid = import_file(path, did, wid)
onshape_element = self.onshape_element
onshape_element.eid = eid
url = onshape_element.get_url()
mass_properties = Client.get_client().part_studios_api.get_mass_properties(did, 'w', wid, eid) # type: BTMassPropResponse
volume = mass_properties.bodies["-all-"].volume[0]*1000000000
bounding_box = Client.get_client().part_studios_api.get_bounding_boxes2(did, 'w', wid, eid, _preload_content=False)
data = bounding_box.data
if six.PY3:
data = data.decode('utf-8')
bounding_box = json.loads(data)
x_span = (bounding_box["highX"]-bounding_box["lowX"])*1000
y_span = (bounding_box["highY"]-bounding_box["lowY"])*1000
z_span = (bounding_box["highZ"]-bounding_box["lowZ"])*1000
bounding_box = "L = {x_span}, W = {y_span}, H = {z_span}".format(x_span=x_span, y_span=y_span, z_span=z_span)
if "part_metadata" in import_item or "element_metadata" in import_item:
meta_data_to_be_set = MetaDataBody(OnshapeElement.create_from_ids(did, "w", wid, eid))
if "part_metadata" in import_item:
part_metadata = import_item["part_metadata"]
# Add the volume part volume property from the API:
meta_data_to_be_set.add_to_part_metadata("Part Volume", volume)
meta_data_to_be_set.add_to_part_metadata("Bounding Box", bounding_box)
for k, v in part_metadata.items():
# For the fixed field items, set for the first part
if k != "additionalItems":
meta_data_to_be_set.add_to_part_metadata(k,
v)
else:
for custom_metadata_item in v:
meta_data_to_be_set.add_to_part_metadata(custom_metadata_item["part_id"],
custom_metadata_item["property_name"],
custom_metadata_item["new_value"])
if "element_metadata" in import_item:
element_metadata = import_item["element_metadata"]
for k, v in element_metadata.items():
if k != "additionalItems":
meta_data_to_be_set.add_to_element_metadata(k,v)
else:
for custom_metadata_item in v:
meta_data_to_be_set.add_to_element_metadata(custom_metadata_item["property_name"],
custom_metadata_item["new_value"], eid=eid)
meta_data_to_be_set.send()
if "bounding_box" in import_item and import_item["bounding_box"]:
feature_path = "bound_all_set_metadata.json"
# feature_path = "bound_all_feature.json"
# feature_path = "make_cube_feature.json"
with open(os.path.dirname(__file__) + "/assets/" + feature_path, "r") as f:
body = {"feature": json.loads(f.read())}
Client.get_client().part_studios_api.add_feature1(did, "w", wid, eid, body=body)
def copy_workspace(copied_from):
params = BTCopyDocumentParams(new_name=name_factory(), is_public=True)
doc = client.documents_api.copy_workspace(
copied_from.did,
copied_from.default_workspace,
bt_copy_document_params=params,
_preload_content=False,
)
doc = json.loads(doc.data.decode("utf-8"))
doc = OnshapeElement.create_from_ids(
did=doc["newDocumentId"], wvm="w", wvmid=doc["newWorkspaceId"]
)
created_docs.append(doc)
return doc
