Error!

import json import os import webbrowser import pytest import six from onshape_client.client import Client from onshape_client.compatible_imports import HTTPServer, HTTPHandler, sendable from onshape_client.onshape_url import OnshapeElement from onshape_client.utility import write_to_file, get_field from test.example_programs import MetaDataBody from test.example_programs.import_file import import_file class myHandler(HTTPHandler): def do_GET(self): try: # Say we are at an https port so that OAuth package doesn't complain. This isn't a security concern because # it is just so that the authorization code is correctly parsed. self.send_response(200) self.send_header("Content-type", "text/html") self.end_headers() content = open(os.path.dirname(__file__) + "/assets/form.html", "r").read() self.wfile.write(sendable(content)) except BaseException as e: self.send_response(500) self.send_header("Content-type", "text/html") self.end_headers() content = """ Error!

Something happened and here is what we know: {}

You may close this tab.

""".format( e ) self.wfile.write(sendable(content)) 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 ) class MyServer(HTTPServer, object): def server_activate(self): super(MyServer, self).server_activate() webbrowser.open("{}:{}".format(self.server_address[0], self.server_port)) @pytest.mark.skip(reason="no way of currently testing this on the CI") def test_run_import_server(): server = MyServer(("localhost", 9000), myHandler) server.serve_forever()