ukiddlmZddlZddlZddlZddlZddlmZmZddl m Z e jZ GddejZ d dZe j ded y) ) annotationsN)AnyIO)coreceZdZdZdZ d dfd ZdZdZdZdZ ddZ d Z d Z d Z d ZeZd ZeZdZeZdZexZZdZexZxZZdZexZZdZxZS) CliDebuggerzA text-based debugger.z(jdb) ctt|||||du|_||_d|_||_d|_y)N)stdinstdout completekeyrz Entering jdb:)super__init__ use_rawinputframes frame_index thread_idintro)selfrrr r r __class__s Y/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/jax/_src/debugger/cli_debugger.pyrzCliDebugger.__init__sB G5[I DDKDDN DJc4|j|jSN)rrrs r current_framezCliDebugger.current_frame(s ;;t'' ((rci}|j|j}|j|j|j|jt |i|Sr)rrupdateglobalslocalseval)rexprenv curr_frames revaluatezCliDebugger.evaluate+sN CT--.JJJz!!"JJz  ! b# rc tt|j||jy#|j YyxYw)zEvaluates an expression.fileNprintreprr$r _error_messagerargs rdefaultzCliDebugger.default2s6 Ds# $4;;7  /2Acg}|jd|jdddD]}|jd|jd|j|j|jdL|j |j}|jd|j tdj||jy) Nz Traceback:z File "z", line z   r&) appendrfilenamelinenooffsetsourcestripr)joinr )r backtraceframelines rprint_backtracezCliDebugger.print_backtrace9sI \"TrT"0%..!1%,,HI  *+||ELL)4 ~./ 0 $))I T[[1rc|j|j}g}|jd|jd|jdt |j D]w\}}|jJ|jdz |z |cxkr|j|zks=n@||jk(r|jd|d|jd|ytdj||jy) Nz> ()z-> r2r3r&) rrr4r5r6 enumerater8r7r)r:r )r num_linesr#contextir=s r print_contextzCliDebugger.print_contextEsT--.JG NNR ++,Aj.?.?-@BCZ../(4    ** *   a ) +q (   i ' ( !! ! ..4v ' ..4v '( $))G 4;;/rctjdd}tj|dj }t d||j y)Nr1z***r&)sysexc_info tracebackformat_exception_onlyr9r)r )rrKmsgs rr+zCliDebugger._error_messageSsB||~bq!H  ) )8 4R 8 > > @C %4;;'rc tt|j||jy#|j YyxYw)zEp expression Evaluates and prints the value of an expression r&Nr(r,s rdo_pzCliDebugger.do_pXs6 Ds# $4;;7 r/c ttj|j||jy#|j YyxYw)zMpp expression Evaluates and pretty-prints the value of an expression r&N)r)pprintpformatr$r r+r,s rdo_ppzCliDebugger.do_ppas: FNN4==- .T[[A s 9<Ac|jt|jdz k(rtd|jn|xjdz c_|j y)z$u(p) Move up a stack frame. rBzAt topmost frame.r&N)rlenrr)r rGr_s rdo_upzCliDebugger.do_upjsL 3t{{+a// dkk2 !rc|jdk(rtd|jn|xjdzc_|jy)z(d(own) Move down a stack frame. rzAt bottommost frame.r&rBN)rr)r rGrWs rdo_downzCliDebugger.do_downus> 1 "5 !rc(|jdy)z6l(ist) List source code for the current file. )rDN)rGrWs rdo_listzCliDebugger.do_lists #rcy)z7c(ont(inue)) Continue the program's execution. TrWs r do_continuezCliDebugger.do_continues rc.tjdy)zOq(uit) (exit) Quit the debugger. The program is given an exit command. rN)rJexitrWs rdo_quitzCliDebugger.do_quitsHHQKrc$|jy)zvw(here) Prints a stack trace with the most recent frame on the bottom. 'bt' is an alias for this command. N)r>rWs rdo_wherezCliDebugger.do_wheres  rc~ |jy#t$rtdtjYnwxYw=)Nz--KeyboardInterrupt--r&)cmdloopKeyboardInterruptr)rJr rs rrunzCliDebugger.runs9 8  8 %CJJ78 s $;;)NNtab)rlist[DebuggerFrame]r IO[str] | Noner rmr str)rI) __name__ __module__ __qualname____doc__promptrrr$r.r>rGr+rPrTrYdo_ur[do_dr^do_lrado_cdo_contrddo_qdo_EOFdo_exitrfdo_wdo_btrj __classcell__)rs@rrrs &>B! !,:!!) 2 0(  $ $$ $ $ $#$#' $8rrc :t||fi|jyr)rrj)rrkwargss r run_debuggerrs fi*6*..0rclir1)rrlrz int | Nonerr) __future__rcmdrRrJrLtypingrrjax._src.debuggerr debugger_core DebuggerFrameCmdrrregister_debuggerr`rrrs]# 3++ I8#''I8V11 |R8r