vjqDdZddlZddlZddlZddlZdZdZGddeZGddeZ Gd d eZ Gd d eZ Gd deZ Gdde Z Gdde ZGdde ZGdde ZGddZGddeZGddeZGddeZGdd eZGd!d"eZGd#d$ej0ZGd%d&ej0Zy)'z[ The CadQuery Gateway Interface. Provides classes and tools for executing CadQuery scripts Nz ct|}|S)at Parses the script as a model, and returns a model. If you would prefer to access the underlying model without building it, for example, to inspect its available parameters, construct a CQModel object. :param script_source: the script to run. Must be a valid cadquery script :return: a CQModel object that defines the script and allows execution )CQModel) script_sourcemodels :/DATA/.local/lib/python3.12/site-packages/cadquery/cqgi.pyparser s M "E Lc6eZdZdZdZdZdZdZd dZdZ y) ra$ Represents a Cadquery Script. After construction, the metadata property contains a ScriptMetaData object, which describes the model in more detail, and can be used to retrieve the parameters defined by the model. the build method can be used to generate a 3d model ct|_tj|t|_||_|j|jy)z Create an object by parsing the supplied python script. :param script_source: a python script to parse N) ScriptMetadatametadataastrCQSCRIPTast_treer _find_vars_find_descriptions)selfrs r__init__zCQModel.__init__&s@ '(  -: *  !r ct|j}|jjD].}t |t j s|j|0y)za Parse the script, and populate variables that appear to be overridable. N)ConstantAssignmentFinderr rbody isinstancerAssign visit_Assign)rassignment_findernodes rrzCQModel._find_vars6sC5T]]CMM&&D$ +!..t4'r cdt|j}|j|jyN)ParameterDescriptionFinderr visitr)rdescription_finders rrzCQModel._find_descriptionsGs#7 F  /r ctd)z Determine if the supplied parameters are valid. NOT IMPLEMENTED YET-- raises NotImplementedError :param params: a dictionary of parameters znot yet implemented)NotImplementedError)rparamss rvalidatezCQModel.validateKs""788r Nc|si}tj}t} |j|t }t j jjddjd|jjd|jjd|jj}t|jtd}t!|||j#|j$|j'|j(||_tj} | |z |_|S#t,$r}|j/|Yd}~?d}~wwxYw)a" Executes the script, using the optional parameters to override those in the model :param build_parameters: a dictionary of variables. The variables must be assignable to the underlying variable type. These variables override default values in the script :param build_options: build options for how to build the model. Build options include things like timeouts, tessellation tolerances, etc :raises: Nothing. If there is an exception, it will be on the exception property of the result. This is the interface so that we can return other information on the result, such as the build time :return: a BuildResult object, which includes the status of the result, and either a resulting shape or an exception __name____cqgi__ show_objectdebugdescribe_parameterexecN)time perf_counter BuildResultset_param_valuesScriptCallbackEnvironmentBuilderwith_real_builtinswith_cadquery_objects add_entryr)r*r+buildcompilerrr, set_debug debugObjectsset_success_result outputObjectsenv Exceptionset_failure_result buildTime) rbuild_parameters build_optionsstartresult collectorr<cexends rr6z CQModel.buildUs3 ! !!# *  ! !"2 3&(I"$##%&&(:z2=)*?*?@7IOO4/1M1MN  x8A CL   Y33 4  % %i&=&= >FJ !;  *  % %b ) ) *sDE E9E44E9c|jj}|jD]-\}}||vrtd|z||}|j |/y)Nz4Cannot set value '%s': not a parameter of the model.)r parametersitemsInvalidParameterError set_value)rr$model_parameterskvps rr0zCQModel.set_param_valuess[==33LLNDAq((+JQN!#A KKN#r NN) r' __module__ __qualname____doc__rrrr%r6r0r rrrs&" 5"09-^ r rceZdZdZdZy) ShapeResultzN An object created by a build, including the user parameters provided c d|_d|_yr)shapeoptionsrs rrzShapeResult.__init__s  r N)r'rRrSrTrrUr rrWrWs r rWc(eZdZdZdZdZdZdZy)r/aj The result of executing a CadQuery script. The success property contains whether the execution was successful. If successful, the results property contains a list of all results, and the first_result property contains the first result. If unsuccessful, the exception property contains a reference to the stack trace that occurred. cXd|_g|_g|_d|_d|_d|_yNF)r?resultsr9 first_resultsuccess exceptionr[s rrzBuildResult.__init__s.   r c ||_d|_yr^)rbra)rrFs rr>zBuildResult.set_failure_results r c||_yr)r9)rr9s rr8zBuildResult.set_debugs (r cx||_t|jdkDr|jd|_d|_y)NrT)r_lenr`ra)rr_s rr:zBuildResult.set_success_results1 t|| q $ QD  r N)r'rRrSrTrr>r8r:rUr rr/r/s )r r/c"eZdZdZdZdZdZy)r zw Defines the metadata for a parsed CQ Script. the parameters property is a dict of InputParameter objects. ci|_yr)rIr[s rrzScriptMetadata.__init__s r c6||j|j<yr)rIname)rrPs radd_script_parameterz#ScriptMetadata.add_script_parameters"#r c0|j|}||_yr)rIdesc)rrj descriptionrPs radd_parameter_descriptionz(ScriptMetadata.add_parameter_descriptions OOD !r N)r'rRrSrTrrkrorUr rr r s $r r c eZdZy) ParameterTypeNr'rRrSrUr rrqrqr rqc eZdZy)NumberParameterTypeNrrrUr rrurursr ruc eZdZy)StringParameterTypeNrrrUr rrwrwrsr rwc eZdZy)BooleanParameterTypeNrrrUr rryryrsr ryc eZdZy)TupleParameterTypeNrrrUr rr{r{rsr r{c6eZdZdZdZe ddZdZdZy)InputParameteraB Defines a parameter that can be supplied when the model is executed. Name, varType, and default_value are always available, because they are computed from a variable assignment line of code: The others are only available if the script has used define_parameter() to provide additional metadata cXd|_d|_d|_d|_g|_d|_yr) default_valuerjvarTyperm valid_valuesast_noder[s rrzInputParameter.__init__s8"    r Ncv|g}t}||_||_||_||_||_||_|Sr)r}rrrjrmrr)rvar_namevar_typerrrmrPs rcreatezInputParameter.createsF  L   ' %r c t|jdkDrP||jvrBtdjt ||j t |j|j tk(r: t|tr t|}n t|}||j_ y|j tk(rt ||j_ y|j tk(r&|rd|j_ yd|j_ y|j t k(r_g}|D]'}|j#t%j&|)||j_t%j*|jytdt |j #t$r/tdjt ||j wxYw)NrzZCannot set value '{0:s}' for parameter '{1:s}': not a valid value. Valid values are {2:s} zJCannot set value '{0:s}' for parameter '{1:s}': parameter must be numeric.TF)valuezUnknown Type of var: )rfrrKformatstrrjrrurintfloatrr ValueErrorrwryr{appendrConstanteltsfix_missing_locations)r new_valuef constantsnvs rrLzInputParameter.set_values~ t  !A %)4;L;L*L'lss NDIIs43D3D/E  <<. . i-IAi(A&' #\\0 0"%i.DMM  \\1 1&* #&+ # \\/ /I  B!78 !*DMM   % %dmm 44c$,,6GH H/ +`ggI  s =8F..8G&cxd|jdt|jdt|jS)NzInputParameter: {name=z, type=z, defaultValue=)rjrrrr[s r__str__zInputParameter.__str__<s- II   "" #  r rQ) r'rRrSrTr staticmethodrrLrrUr rr}r}s/ &MQ (IT r r}c<eZdZdZdZifdZifdZdZdZdZ y) r1z Allows a script to communicate with the container the show_object() method is exposed to CQ scripts, to allow them to return objects to the execution environment c g|_g|_yr)r;r9r[s rrzScriptCallback.__init__Ksr c |j|t}||_||_|jj |y)z Return an object to the executing environment, with options. :param shape: a cadquery object :param options: a dictionary of options that will be made available to the executing environment N)updaterWrZrYr;r)rrYrZkwargsos rr)zScriptCallback.show_objectOs9 v M  !!!$r cjt}||_||_|jj |y)zH Debug print/output an object, with optional arguments. N)rWrYrZr9r)robjargsss rr*zScriptCallback.debug]s- M    #r cy)zX Do Nothing-- we parsed the ast ahead of execution to get what we need. NrU)rvar_datas rr+z!ScriptCallback.describe_parameterf r cy)ze Not implemented yet: allows scripts to indicate that there are problems with inputs NrU)rparam field_lists r add_errorzScriptCallback.add_errorlrr c2t|jdkDS)Nr)rfr;r[s r has_resultszScriptCallback.has_resultsrs4%%&**r N) r'rRrSrTrr)r*r+rrrUr rr1r1Ds/ *, %!$  +r r1ceZdZdZy)rKzr Raised when an attempt is made to provide a new parameter value that cannot be assigned to the model Nr'rRrSrTrUr rrKrKv   r rKceZdZdZy) NoOutputErrorz` Raised when the script does not execute the show_object() method to return a solid NrrUr rrrrr rc*eZdZdZddZdZdZdZy)ScriptExecutionErrorz| Represents a script syntax error. Useful for helping clients pinpoint issues with the script interactively NcH|d|_n||_|d|_y||_y)NrzUnknown Script Errorlinemessage)rrrs rrzScriptExecutionError.__init__s( <DIDI ?1DL"DLr c"|jSr__repr__r[s r full_messagez!ScriptExecutionError.full_message}}r c"|jSrrr[s rrzScriptExecutionError.__str__rr c:d|jd|jS)NzScriptError [Line z]: rr[s rrzScriptExecutionError.__repr__s.2iiFFr rQ)r'rRrSrTrrrrrUr rrrs #Gr rc4eZdZdZdZdZdZdZdZdZ y) r2z Builds an execution environment for a cadquery script. The environment includes the builtins, as well as the other methods the script will need. ci|_yrr<r[s rrzEnvironmentBuilder.__init__s r c,|jtSr) with_builtins __builtins__r[s rr3z%EnvironmentBuilder.with_real_builtinss!!,//r c$||jd<|S)Nrr)renv_dicts rrz EnvironmentBuilder.with_builtinss#+  r cRt|jd<t|jd<|S)Ncadquerycq)rr<r[s rr4z(EnvironmentBuilder.with_cadquery_objectss"'! r c$||j|<|Srr)rrjrs rr5zEnvironmentBuilder.add_entrys r c|jSrrr[s rr6zEnvironmentBuilder.builds xxr N) r'rRrSrTrr3rr4r5r6rUr rr2r2s% 0 r r2ceZdZdZdZdZy)rzb Visits a parse tree, looking for function calls to describe_parameter(var, description ) c||_yrcqModelrcq_models rrz#ParameterDescriptionFinder.__init__  r c |jjdk(rN|jdj}|jdj}|jj |||S#Y|SxYw)zO Called when we see a function call. Is it describe_parameter? r+r)funcidrrrro)rrvarnamerms r visit_Callz%ParameterDescriptionFinder.visit_Callsh yy||33))A,//yy|)) 66wE     s A'A++A0N)r'rRrSrTrrrUr rrrs r rc"eZdZdZdZdZdZy)rzH Visits a parse tree, and adds script parameters to the cqModel c||_yrrrs rrz!ConstantAssignmentFinder.__init__rr c  t|tjk(r|jdk(r6|jj t j||tdy|jdk(r|jj t j||tdyt|tjk(rYd}|jD]}||jfz}|jj t j||t|yttdrt|tjk(r{|jdk(r6|jj t j||tdy|jj t j||tdyttdrt|tjk(rt tt"t$t&t(t*t(t,ti}|jj t j|||t|j|jyyyy#t/d|zYyxYw) NTrueTFalseFrU NameConstantrz-Unable to handle assignment for variable '%s')typerNamerrrkr}rryTuplerrr{hasattrrrboolrrwrrurtupleprint)rr value_nodetupentry type_dicts rhandle_assignmentz*ConstantAssignmentFinder.handle_assignments< J388+==F*LL55&--&2F  ]]g-LL55&--&2F j!SYY.'__E.C- 11"))"H.@# n-$z2BcFVFV2V##t+LL55&--&2F LL55&--&2F j)d:.>#,,.N.,.,-   11"))" !$z'7'7"89"(( /O)?.d  AHL M s.A I:#AI:)A4I:A0I:5I:B1I::J c |jd}t|tjrytjtj g}t |j|vr(|j|j|j|St |jtjk(rt|tjr(|j|j|j|St|j|jjD]!\}}|j|j|#|S#tjtdtj zY|SxYw)Nrz)Unable to handle assignment for node '%s')targetsrr Attributerrrrrrrzipr traceback print_excrdump)rr left_sideastTypesnrOs rrz%ConstantAssignmentFinder.visit_Assign$s U QI)S]]3#,,/HDJJ8+&&y||TZZ@ djj!SYY.i2**9<<D !$INNDJJOO D1..qttQ7!E   U    ! =@SS T s)EAE A&E3AE7E=N)r'rRrSrTrrrrUr rrrs =~r r)rTrrr-rrrobjectrrWr/r rqrurwryr{r}r1r=rKrrr2NodeTransformerrrrUr rrs    sfsl&&DV" F  -  -  =   _ _ D/+V/+d I  I G9G8<!4!46_s22_r