o i!5 @sddlmZddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z ddl Z ddl m Z ddlmZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl%m'Z'ddl(m)Z)ddl*m+Z+ddl*m,Z,ddl*m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3ddl2m4Z4ddl5m6Z6dd l5m7Z7e j8Z9e1e9Z:e:d!krdd"l;mfd%d&Z?d$e@e>e>ffd'd(ZAd)e e>fd*d+ZBeCd,e@e&d,e/ejDd-d.d/eBeDd0d1d2ZEd3ZFd4ZGd5d6ZHd7d8ZId9d:ZJd;d<ZKd=d>ZLd?d@ZMdAdBZNdCdDZOdEdFZPdGedHeQdIe@d$eQeQe@ffdJdKZRdLe e>eefdMdNZSdOedPeTd$e@fdQdRZUdSeTedTe6d$dfdUdVZVdWdXZWdS)Y)Callable)Iterable)StringION)Optional)Union)MiddlewareManager) GraphQLError)ExecutionResult)Source)config)Pin)_SPAN_MEASURED_KEY) ERROR_MSG) ERROR_STACK) ERROR_TYPE) trace_utils) SpanTypes)core) COMPONENT)schematize_service_name)schematize_url_operation) SpanDirection) ArgumentError)get_argument_value)set_argument_value)asbool) parse_version)unwrap)wrap)Span)tracerr)Document) DocumentNodereturncCstSN)_graphql_version_strr(r(Z/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/contrib/internal/graphql/patch.py get_version2sr*cCsddiS)Ngraphqlz>=3.1r(r(r(r(r)_supported_versions6sr,error_extensionscCs$|durdd|dD}|SdS)z)Parse the user provided error extensions.NcSsg|]}|qSr()strip).0er(r(r) =z+_parse_error_extensions..,)split)r-fieldsr(r(r)_parse_error_extensions:sr6r+"DD_TRACE_GRAPHQL_RESOLVERS_ENABLEDF)default!DD_TRACE_GRAPHQL_ERROR_EXTENSIONS)_default_serviceresolvers_enabled_error_extensionszgraphql.sourcezgraphql.operation.typezgraphql.operation.namecCsFttddrdSdt_tttD] \}}}tt|||qdS)N_datadog_patchFT)getattrr+r=r onto_get_patching_candidates_update_patchingr module_str func_namewrapperr(r(r)patchQs  rFcCsBttddr tdkr dStD] \}}}tt|||qdt_dS)Nr=F)r)r>r+_graphql_versionr@rArr=rBr(r(r)unpatch[s  rIcCsPtdkrddtfddtfddtfdd tfgSdd tfddtfd dtfd d tfgS) Nr!zgraphql.graphqlexecute_graphqlzgraphql.language.parserparsezgraphql.validation.validationvalidatezgraphql.execution.executorexecute graphql_implzgraphql.validation.validatezgraphql.execution.execute)rH _traced_query _traced_parse_traced_validate_traced_executer(r(r(r)r@esr@cCs"tj|}t||}|||dSr&)sysmodulesr>) operationrCrDrEmodulefuncr(r(r)rAus  rAcCstt}|r |s||i|St||dd}t|}tjdt |t jt j d}| tt jj| t|||i|WdS1sKwYdS)Nrsourcez graphql.parsenameservice span_type)r get_fromr+enabledr_get_source_strr tracer int_servicer rGRAPHQL _set_tag_strrintegration_name_GRAPHQL_SOURCE)rWargskwargspinrX source_strspanr(r(r)rP{s     $rPcCstt}|r |s||i|St||dd}t|}tjdt |t jt j d$}| tt jj| t|||i|}t|||WdS1sRwYdS)Nastzgraphql.validaterY)r r]r+r^rr_r r`rrar rrbrcrrdre_set_span_errors)rWrfrgrhdocumentrirjerrorsr(r(r)rQs"     $rQcCs tt}|r |s||i|Stjjrtt||\}}tdkr*t ||dd}nt ||dd}t |}t j d|p;dt |tjtjd5}|ttjj|tdt|||t|||i|}t|trrt|j||WdS1s~wYdS)Nr!rk document_astrnzgraphql.executerZresourcer[r\)r r]r+r^r r; _inject_trace_middleware_to_args_resolver_middlewarerHrr_r r`rrarrbrcrrd set_metricr _set_span_operation_tagsre isinstancer rmro)rWrfrgrhrnrirjresultr(r(r)rRs2        $rRcCstt}|r |s||i|St||dd}t|}tjtddt j d|p)dt |t jtjd*}|tt jj|td||i|}t|trUt|j||WdS1sawYdS)NrkrXzgraphql.requestr+)protocol directionrq)r r]r+r^rr_r r`rrINBOUNDrrar rrbrcrrdrur rwr rmro)rWrfrgrhrXrrrjrxr(r(r)rOs&      $rOcKstt}|r |s|||fi|Stjd|jtjd}| t t jj |||fi|WdS1s:wYdS)z trace middleware which wraps the resolvers of graphql fields. Note - graphql middlewares can not be a partial. It must be a class or a function. zgraphql.resolve)rZrrr\N) r r]r+r^r r` field_namerrbrcrr rd)next_middlewarerootinforfrhrjr(r(r)rts  $rttrace_middlewarerfrgcCsxd}tdkrd}zt|||dpg}t|tr|j}Wn ty&g}Ynwt||g}t|||d|\}}||fS)zJ Adds a trace middleware to graphql.execute(..., middleware, ...) )r"rG middleware)rHrrwr middlewaresrlistr)rrfrgmiddlewares_argrr(r(r)rss  rsobjcCsXt|tr|}nt|tr|j}nt|tr!|jdur!|jjj}nd}tdd| S)zo Parses graphql Documents and "Source" objects to retrieve the graphql source input for a request. Nz\s+ ) rwstrr bodyr#locrXresubr.)rrir(r(r)r_s   r_errorerror_extension_fieldscCsTi}|D]#}||jvr't|j|tttfr|j|||<qt|j|||<q|S)zValidate user-provided extensions format and return the formatted extensions. All extensions values MUST be stringified, EXCEPT for numeric values and boolean values, which remain in their original type. ) extensionsrwintfloatboolr)rrr-fieldr(r(r)_validate_error_extensions*s rrorjcCs`|sdSd|_dtjtjf}|t|ddd|D}|t||D]}|j| dd}|j rAd d|j D}||d <|j rut |||j }}} t } tj||| | tjd | } | |d <|t| td |||| f|jdurddd|jD} | |d<tjj} | durt|| }|r|D] }|||d|<q|jd|dq'dS)zC Set tags on error span and set span events on each error. Nrkz%s.%s cSg|]}t|qSr(r)r/rr(r(r)r1Er2z$_set_span_errors..z error.type)messagetypecSsg|] }|jd|jqS):)linecolumn)r/rr(r(r)r1Ms locations)filelimit stacktracezspan.exceptionr3cSrr(r)r/path_objr(r(r)r1\r2pathz extensions.zdd.graphql.query.error)rZ attributes)rr __module____name__rcrjoinrrget_tagr __traceback__rr tracebackprint_exceptionr _span_traceback_max_sizegetvaluerrdispatchrr+r<r _add_event)rorj exc_type_str error_msgsrrrexc_typeexc_valexc_tbbufftbrrrkeyr(r(r)rm:sH     rmcCsXt|}|s dStdkr|t|jn|t|jj|jr*|t|jjdSdS)Nr!) r+get_operation_astrHrc_GRAPHQL_OPERATION_TYPErUvaluerZ_GRAPHQL_OPERATION_NAME)rjrn operation_defr(r(r)rvks rv)Xcollections.abcrriorosrrSrtypingrrr+r graphql.errorrgraphql.executionr graphql.language.sourcer ddtracer ddtrace._trace.pinr ddtrace.constantsr rrrddtrace.contribr ddtrace.extrddtrace.internalrddtrace.internal.constantsrddtrace.internal.schemarr-ddtrace.internal.schema.span_attribute_schemarddtrace.internal.utilsrrrddtrace.internal.utils.formatsrddtrace.internal.utils.versionrddtrace.internal.wrappingrr ddtrace.tracerr __version__r'rHgraphql.language.astr#r$rr*dictr,r6_addgetenvrerrrFrIr@rArPrQrRrOrttuplersr_rrrmrvr(r(r(r)s                                   %" 1