cdZdZ ddlZddlZddlZddlmZddlZddl m Z  ddl m Z dZ dZd ZGd d eZdZdZy#e$rdZYCwxYw#e$r d Z d ZY/wxYw)zCustomized version of pdb's default debugger. - sets up a history file - uses ipython if available to colorize lines of code - overrides list command to search for current block instead of using 5 lines of context zrestructuredtext enN)Pdb)StringIO) PyColorizecXtj}t}|j||g}t |j j D]>\}}||z}||k(r|j|dd|(|j|dd|@dj|S)zUcolorize and annotate source with linenos (as in pdb's list command) z>4z -> z  ) rParserrformat enumerategetvalue splitlinesappendjoin) source start_lineno curlinenoparseroutput annotatedindexlinelinenos 9/usr/lib/python3/dist-packages/logilab/common/debugger.pycolorizer<s""$ ff% $V__%6%A%A%CD ;KE4\)F"  F2;fTF!;<  F2;d4&!9:  ; yy##ctj}t}|j|||j S)zcolorize given source)rrrr r )rrrs rcolorize_sourcerLs2""$ ff%  rc|S)zfallback colorize function)rrrs rrr3s rc|SNr)rs rrr7s rcXtj|\}}dj||fS)aReturn the text of the source code for an object. The argument may be a module, class, method, function, traceback, frame, or code object. The source code is returned as a single string. An IOError is raised if the source code cannot be retrieved.)inspectgetsourcelinesr)objlineslnums r getsourcer(Ts+ ((-KE4 775>4 rcVeZdZdZd dZdZdZdZdZdZ d Z d Z d Z e Z d ZeZy)Debuggerzcustom debugger - sets up a history file - uses ipython if available to colorize lines of code - overrides list command to search for current block instead of using 5 lines of context Nctj||j|r%|j|j}|j||_t j jd|_y)Nz ~/.pdbhist) r__init__resettb_next_tcbkospath expanduser _histfile)selftcbks rr,zDebugger.__init__hsV T ,,*||,,* ++L9rcpt! tj|jyy#t$rYywxYw)z/if readline is available, read pdb history fileN)readlineread_history_filer3OSErrorr4s rsetup_history_filezDebugger.setup_history_fileqs:   **4>>:   s ) 55cd|j|jj|jy)zstarts the interactive modeN) interactionr/tb_framer:s rstartzDebugger.start{s  ,,djj9rcR|jtj|||y)zsetup hook: set up history fileN)r;rsetup)r4framer5s rrAzDebugger.setups ! $t$rcxttj|jtj|y)z,quit hook: save commands in the history fileN)r7write_history_filer3rset_quitr:s rrEzDebugger.set_quits&    ' ' 7 Trct|jj}|j|jjd|vr|j ||S|Dcgc]}|j |s|c}Scc}w)z7provide variable names completion for the ``p`` command.)dictcurframe f_globalsupdatef_locals attr_matches startswith)r4textr begin_idxend_idx namespacevarnames r complete_pzDebugger.complete_psi001 //0 $;$$T95 5'0MGG4F4Ft4LMMMs A=6A=cxddl}|jd|}|sy|jdd\}}t||}t |}t |dr/|j d||j|jz}g} t|} |D]'} | d| |k(s | dk7s| j |d| )| S) aimplementation coming from rlcompleter.Completer.attr_matches Compute matches when text contains a dot. Assuming the text is of the form NAME.NAME....[NAME], and is evaluatable in self.namespace, it will be evaluated and its attributes (as revealed by dir()) are used as possible completions. (For class instances, class members are also considered.) WARNING: this can still invoke arbitrary C code, if an object with a __getattr__ hook is evaluated. rNz(\w+(\.\w+)*)\.(\w*) __class__ __builtins__rG) rematchgroupevaldirhasattrr get_class_membersrXlen) r4rOrRrZmexprattrobjectwordsmatchesnwords rrMzDebugger.attr_matchess  HH,d 3 WWQ] ddI&F  6; ' LL %D2263C3CDDE I 1DBQx4DN$:$q/0 1rc~t|}t|dr%|jD]}||j|z}|S)z8implementation coming from rlcompleter.get_class_members __bases__)r^r_rkr`)r4klassretbases rr`zDebugger.get_class_memberssB%j 5+ & 9D22488 9 rcFd|_|sS t|j\}}tt dj |||jj ytj||y#t$rYyt$rtj||YywxYw)znoverrides default list command to display the surrounding block instead of 5 lines of context listr"N) lastcmdr(rIprintrrf_linenoKeyboardInterruptr9rdo_list)r4argrrs rruzDebugger.do_lists  ''0'?$ hrwwv dmm>T>TUV KKc " %  ' D#& 'sAA44 B ?B B c|jjj}|jj}d|d|}t j |y)z:opens source file corresponding to the current stack levelzemacsclient --no-wait + N)rIf_code co_filenamersr0system)r4rvfilenamercmds rdo_openzDebugger.do_opensC==''33'''xq ; #rr )__name__ __module__ __qualname____doc__r,r;r?rArErTrMr`rudo_lr~do_orrrr*r*_sH::%  N># D Drr*cVttj}|jy)zuse our custom debuggerN)r*syslast_tracebackr?)dbgs rpmrs 3%% &CIIKrcntjtjjyr )r* set_tracer _getframef_backrrrrrs J//0r)r __docformat__r7 ImportErrorr0rpdbrr#logilab.common.compatrIPythonrrrr(r*rrrrrrs$ &   *"!"$ ! tstn 1sH  s"AAA A AA