o i(@s ddlmZddlZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZddlmZmZddlZddlmZdd lmZmZmZdd lmZmZdd lmZed d GdddZeGdddejZ Gddde Z!Gddde"Z#ej$dddddZ%dS)) annotationsN)ABCabstractmethod)Iterator) dataclass)Anycast)types)ATTRIBUTES_MESSAGE_TEMPLATE_KEYATTRIBUTES_SAMPLE_RATE_KEYATTRIBUTES_TAGS_KEY) StackInfoget_filepath_attribute)uniquify_sequenceT)frozenc@s:eZdZUdZded<ded<dZded<dZded <dS) LogfireArgszValues passed to `logfire.instrument` and/or values stored in a logfire instance as basic configuration. These determine the arguments passed to the method calls added by the AST transformer. ztuple[str, ...]tagsz float | None sample_rateNz str | None msg_template span_name)__name__ __module__ __qualname____doc____annotations__rrrra/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/ast_utils.pyrs  rc@sveZdZUdZded<ded<ded<ded<dd Zd"d d Zd#ddZd$ddZd%ddZ d&ddZ d'dd Z d!S)(BaseTransformerzXHelper for rewriting ASTs to wrap function bodies in `with {logfire_method_name}(...):`.r logfire_argsstrlogfire_method_namefilename module_namecCs g|_dSN)qualname_stackselfrrr __post_init__0s zBaseTransformer.__post_init__node ast.ClassDefcCs.|j|jttj||}|j|Sr$)r%appendnamerastClassDef generic_visitpopr'r)rrrvisit_ClassDef5s zBaseTransformer.visit_ClassDef&ast.FunctionDef | ast.AsyncFunctionDefcCsP|j|jd|j}|jd|||j|j|||S)N.z)r%r+r,joinr/r0rewrite_functionr'r)qualnamerrrvisit_FunctionDef<s      z!BaseTransformer.visit_FunctionDefast.AsyncFunctionDefcCs ||Sr$)r9r1rrrvisit_AsyncFunctionDefGs z&BaseTransformer.visit_AsyncFunctionDefr8returnast.ASTcs j}g}|r+t|dtjr+t|djtjr+t|djjtr+|| d|rLt |dkrNt|dtj sLt|dtjrNt|djtjrNStj tj ||dg|jd}||fddjD}||d<tdi|}tt|S) Nrr ) context_expr)itemsbody type_commentcsi|] }|t|dqSr$)getattr).0r,r)rr ssz4BaseTransformer.rewrite_function..r@r)r@copy isinstancer-ExprvalueConstantr r+r0lenPassWithwithitemlogfire_method_call_noderA_fieldstypefix_missing_locations copy_location)r'r)r8r@new_bodyspankwargsnew_noderrDrr6JsB     z BaseTransformer.rewrite_functionast.CallcCstr$)NotImplementedErrorr7rrrrOxsz(BaseTransformer.logfire_method_call_nodelinenoint0tuple[str, dict[str, otel_types.AttributeValue]]c Csit|j||d}i|}|j}|jpd|jd|}||t<|jp'|}|jr2t|j|t <|j }|dvr=||t <||fS)N)z code.linenoz code.functionzCalling r4)Nr ) rr"rrr#r rrrr rr ) r'r8rZ stack_info attributesrrrrrrrlogfire_method_arg_values{s" z)BaseTransformer.logfire_method_arg_valuesN)r)r*)r)r3)r)r:)r)r3r8r r<r=)r)r3r8r r<rX)r8r rZr[r<r\) rrrrrr(r2r9r;r6rOr_rrrrr's     .rc@s`eZdZdZdddZddd Zedd d ZedddZedddZ d ddZ ddZ dS)!CallNodeFinderaBase class for finding the `ast.Call` node corresponding to a function call in a given frame. Uses `executing`, then falls back to a heuristic if that fails. The heuristic is defined by subclasses which depends on what we're looking for, but in general `executing` is expected to work. Warns appropriately when things don't work. Only used when `inspect_arguments=True` in `logfire.configure()`. frametypes.FrameTypecCs,||_tj||_|jj|_||_dSr$)ra executingSourceexsource_get_call_noder))r'rarrr__init__s zCallNodeFinder.__init__r<ast.Call | Nonecstjjtjr jjSjjsddSd}tj dddkr&|d7}fdd D}t |dkr>|dS|d S) NzNo source code available. This happens when running in an interactive shell, using exec(), or running .pyc files without the source .py files.z"`executing` failed to find a node.) zM This may be caused by a combination of using Python < 3.11 and auto-tracing.cs6g|]}t|D]}t|tjr |r |q qSr)r-walkrGCallheuristic_call_node_filter)rC main_noder)r&rr s z1CallNodeFinder._get_call_node..r r) rGrer)r-rnrftextwarn_inspect_argumentssys version_infoheuristic_main_nodesrK)r'msg call_nodesrr&rrgs"   zCallNodeFinder._get_call_nodeIterator[ast.AST]cCdS)zFAST nodes (e.g. statements) to search for potential call nodes inside.Nrr&rrrrvz#CallNodeFinder.heuristic_main_nodesr)rXboolcCrz)zKCondition that a potential call node must satisfy to be considered a match.Nrr1rrrror{z)CallNodeFinder.heuristic_call_node_filterr cCrz)zMiddle part of the warning message for `warn_inspect_arguments`. Should describe the consequences of the failure. Nrr&rrrwarn_inspect_arguments_middler{z,CallNodeFinder.warn_inspect_arguments_middlerwcCs>ddl}d|d|}tj|t|d||dS)NrzHFailed to introspect calling code. Please report this issue to Logfire. z_ Set inspect_arguments=False in logfire.configure() to suppress this warning. The problem was: ) stacklevel)logfirer}warningswarnInspectArgumentsFailedWarningget_stacklevel)r'rwrrrrrssz%CallNodeFinder.warn_inspect_argumentscCs6t}d}|r||jkr |S|d7}|j}|s|S)Nrr )inspect currentframeraf_back)r' current_framer~rrrrs zCallNodeFinder.get_stacklevelN)rarb)r<ri)r<ry)r)rXr<r|)r<r )rwr ) rrrrrhrgrrvror}rsrrrrrr`s (     r`c@s eZdZdS)rN)rrrrrrrrsri)maxsizer)r= ex_sourceexecuting.SourcecCs\t|}t|j|p d}z ttj|dd}Wn ty%d}Ynw||kr,|S|S)a]Returns some Python source code representing `node`. Preferably the actual original code given by `ast.get_source_segment`, but falling back to `ast.unparse(node)` if the former is incorrect. This happens sometimes due to Python bugs (especially for older Python versions) in the source positions of AST nodes inside f-strings. eval)mode)r-unparseget_source_segmentrrparse Exception)r)rsource_unparsedsource_segmentsource_segment_unparsedrrrget_node_source_texts  r)r)r=rr)& __future__rr- functoolsrrtr rabcrrcollections.abcr dataclassesrtypingrrrcopentelemetry.util otel_types constantsr r r r]rrutilsrrNodeTransformerrr`Warningr lru_cacherrrrrs0      ka