o ;i @s\ddlZddlZddlZddlZddlZddlmZdeeeffddZ defddZ dS)N)FunctionWithAioreturncCst|}t|}dtfdd}t|}tt|d}t |tj tj fs*Jdd|j D}t |j|jfg|}t dd|jD}|j|||d|d |d|d d fS) a3Get function signature, including decorators and comments, from source code Traverses functools.wraps-wrappings to get source of underlying function. Has the advantage over inspect.signature that it can get decorators, default arguments and comments verbatim from the function definition. rcSsL|d}||d|}|dd||d<|d|d|d<d|S)N r)splitjoin)srcfromlinefromcoltolinetocollinesrN/home/ubuntu/.local/lib/python3.10/site-packages/modal_docs/mdmd/signatures.pyget_source_segments  z/_signature_from_ast..get_source_segmentrcSsg|] }|j|jdfqS)rlineno col_offset.0itemrrr sz'_signature_from_ast..css|] }|j|jfVqdS)Nrrrrr "sz&_signature_from_ast..r)inspect getsourcetextwrapdedentstrastparselistiter_child_nodes isinstance FunctionDefAsyncFunctionDefdecorator_listminrrbodynamestrip)funcr rtreefunc_defdecorator_startsdeclaration_start body_startrrr_signature_from_ast s   &r1cCst|st|st|tst|dsJ|j}zt|\}}Wnty?t d|d|}d|t |}Ynw||krP| d|d|}d|vrpt |spt|sp| dd}| dd}| d d }td d |\}}|S) aA problem with using *only* this method is that the wrapping method signature will not be respected. TODO: use source parsing *only* to extract default arguments, comments (and possibly decorators) and "merge" that definition with the outer-most definition.__call__z#Could not get source signature for z. Using fallback.zdef z async defdefasynccontextmanagercontextmanager AsyncIteratorIteratorz^\s*@synchronizer\..*\n)r isfunctionismethodr#rhasattrr2r1 Exceptionwarningswarn signaturereplaceiscoroutinefunctionisasyncgenfunctionresubn)r)callable original_namedefinition_source_rrr get_signature*s,    rI) rrrCrr=synchronicity.synchronizerrtuplerr1rIrrrrs