Method handle_assignment

cadquery/cqgi.py:485–546

(self, var_name, value_node)

        self.cqModel = cq_model

    def handle_assignment(self, var_name, value_node):
        try:
            if type(value_node) == ast.Name:
                if value_node.id == "True":
                    self.cqModel.add_script_parameter(
                        InputParameter.create(
                            value_node, var_name, BooleanParameterType, True
                        )
                    )
                elif value_node.id == "False":
                    self.cqModel.add_script_parameter(
                        InputParameter.create(
                            value_node, var_name, BooleanParameterType, False
                        )
                    )
            elif type(value_node) == ast.Tuple:
                # Handle multi-length tuples
                tup = ()
                for entry in value_node.elts:
                    tup = tup + (entry.value,)

                self.cqModel.add_script_parameter(
                    InputParameter.create(
                        value_node, var_name, TupleParameterType, tup,
                    )
                )
            elif hasattr(ast, "NameConstant") and type(value_node) == ast.NameConstant:
                if value_node.value == True:
                    self.cqModel.add_script_parameter(
                        InputParameter.create(
                            value_node, var_name, BooleanParameterType, True
                        )
                    )
                else:
                    self.cqModel.add_script_parameter(
                        InputParameter.create(
                            value_node, var_name, BooleanParameterType, False
                        )
                    )

            elif hasattr(ast, "Constant") and type(value_node) == ast.Constant:

                type_dict = {
                    bool: BooleanParameterType,
                    str: StringParameterType,
                    float: NumberParameterType,
                    int: NumberParameterType,
                    tuple: TupleParameterType,
                }

                self.cqModel.add_script_parameter(
                    InputParameter.create(
                        value_node,
                        var_name,
                        type_dict[type(value_node.value)],
                        value_node.value,
                    )
                )