o ;i^@sTdZddlZddlZddlZddlZddlZddlmZddlmZm Z m Z ddl m Z ddl mZddlmZdd lmZeeefZeeeefefZd ed ed eeeffd dZd ededed dfddZde eded efddZde ed efddZd e eede ede efddZ de ej!fddZ"dZ#Gd d!d!Z$dS)"zHelper functions related to operating on exceptions, warnings, and traceback objects. Functions related to *displaying* tracebacks should go in `modal/cli/_traceback.py` so that Rich is not a dependency of the container Client. N) TracebackType)AnyIterableOptional)config)api_pb2) Traceback) ServerWarningexctask_idreturncCst|j}|}|dur(|jjj}|drnd|d||jj_|j}|dus |}t|di}t |jD] }|j ||j |j f<q8||fS)a6Given an exception, extract a serializable traceback (with task ID markers included), and a line cache that maps (filename, lineno) to line contents. The latter is used to show a helpful traceback to the user, even if they don't have packages installed locally that are referenced in the traceback.N:__line_cache__)TBLibTraceback __traceback__tb_framef_code co_filename startswithtb_nextto_dictgetattr traceback extract_tblinefilenamelineno)r r tbcurfiletb_dict line_cacheframer$D/home/ubuntu/.local/lib/python3.10/site-packages/modal/_traceback.pyextract_tracebacks    r&r!r"cCsJt|}|durd|jjjvrn|j}|dus ||_t|d|dS)Nz /pkg/modal/r) r from_dict as_tracebackrrrrrsetattr)r r!r"rr$r$r%append_modal_tb<sr*r user_sourcecCsdtdtfdd}|}|dur/|jdur(||jjjjr"|jj|_nn|jdus|j}|dus|}|drP|durO|jjj|krO|j}|durO|jjj|ksAn|duri|jjjdkri|j}|duri|jjjdks[|duro|}|S)zRReturn a traceback that does not contain modal entrypoint or synchronicity frames.rr cSsd|vpd|vS)Nz/site-packages/synchronicity/zmodal/_utils/deprecationr$)rr$r$r% skip_frameQsz1reduce_traceback_to_user_code..skip_frameNz.pyz)strboolrrrrendswithco_name)rr+r,tb_rootr$r$r%reduce_traceback_to_user_codeMs,    r2cCs0|durtd|jjjrdS|j}|dusdS)zYInspect the traceback stack to determine whether an error was raised locally or remotely.Nz^:TF)rematchrrrr)rr$r$r%traceback_contains_remote_callls r5valuecCsJt|||tjdkr!|dur#t|dg}t|dtjddSdSdS)zQAdd backwards compatibility for printing exceptions with "notes" for Python<3.11.) N __notes__ )sepr )rprint_exceptionsys version_inforprintstderr)r r6rnotesr$r$r%r<us  r<server_warningscCs |D] }t|jtddqdS)zIssue a warning originating from the server with empty metadata about local origin. When using the Modal CLI, these warnings should get caught and coerced into Rich panels. zrN)warnings warn_explicitmessager )rBwarningr$r$r%print_server_warnings}srGzoInternal Modal traceback frames are suppressed for readability. Use MODAL_TRACEBACK=1 to show a full traceback.c @sDeZdZddZdeeedeedeedej dfdd Z d S) suppress_tb_framecCsdS)Nr$)selfr$r$r% __enter__szsuppress_tb_frame.__enter__exc_typer rr FcCsdt|t }tds|dus|rdS|dur|jn|}||t|dg}t|vr0|tdS)NrFr9) isinstance Exceptionrgetrwith_tracebackrtraceback_suppression_noteappend)rIrKr ris_base_exceptionfinal_tbrAr$r$r%__exit__s    zsuppress_tb_frame.__exit__N) __name__ __module__ __qualname__rJrtype BaseExceptionrtypingLiteralrTr$r$r$r%rHs rH)%__doc__r3r=rrZrCtypesrrrr modal.configr modal_protor _vendor.tblibr r exceptionr dictr- TBDictTypetuple LineCacheTyperYr&r*r2r.r5rXr<WarningrGrPrHr$r$r$r%s.      "&