3
aG\
/ @ sD d Z ddlZddlZddlmZmZ ddlmZmZ ddl m
Z
ddl m Z m
Z
mZ yddlmZ W n ek
r| dZY nX y ed W n. ek
r d ZY n ek
r d
ZY nX G d d d eZd
d ZG dd deZd ddZd ddZd ddZ dd Z dd Z dd Z dZ edkr@y
e Z W n Y nX [ dS ) aQ
jinja2.debug
~~~~~~~~~~~~
Implements the debug interface for Jinja. This module does some pretty
ugly stuff with the Python traceback system in order to achieve tracebacks
with correct line numbers, locals and contents.
:copyright: (c) 2017 by the Jinja Team.
:license: BSD, see LICENSE for more details.
N)
TracebackTypeCodeType)missing
internal_code)TemplateSyntaxError) iteritemsreraisePY2)tproxyzraise TypeError, 'foo'z raise __jinja_exception__[1]z4raise __jinja_exception__[0], __jinja_exception__[1]c @ s@ e Zd ZdZdd Zedd Zdd Zedd Zd
d Z d S )
TracebackFrameProxyzProxies a traceback frame.c C s || _ d | _d S )N)tb_tb_next)selfr r /usr/lib/python3.6/debug.py__init__' s z TracebackFrameProxy.__init__c C s | j S )N)r
)r r r r tb_next+ s zTracebackFrameProxy.tb_nextc C sB t d k r8yt | j|r|jp d W n tk
r6 Y nX || _d S )N) tb_set_nextr Exceptionr
)r nextr r r set_next/ s z TracebackFrameProxy.set_nextc C s d| j jjkS )N__jinja_template__)r tb_frame f_globals)r r r r is_jinja_frame: s z"TracebackFrameProxy.is_jinja_framec C s t | j|S )N)getattrr )r namer r r __getattr__> s z TracebackFrameProxy.__getattr__N)
__name__
__module__ __qualname____doc__r propertyr r r r r r r r r $ s r c s* t | td kr S fdd}tt|S )Nc s@ | dkrt |d S | dkr, j|| nt | ||S d S )N__getattribute__r r __setattr__)r# r )r r$ )Z operationargskwargs)proxyr r operation_handlerF s
z+make_frame_proxy..operation_handler)r r
r )framer( r )r' r make_frame_proxyB s
r* c @ sP e Zd ZdZdd ZdddZddd Zed
d Zed d
Z edd Z
dS )ProcessedTracebackz?Holds a Jinja preprocessed traceback for printing or reraising.c C sJ || _ || _|| _d }x$| jD ]}|d k r4|j| |}q W |jd d S )N)exc_type exc_valueframesr )r r, r- r. Zprev_tbr r r r r S s
zProcessedTraceback.__init__Nc C s* t j| j| j| jd |d}dj|j S )z#Return a string with the traceback.r )limit ) tracebackformat_exceptionr, r- r. joinrstrip)r r/ linesr r r render_as_texta s z!ProcessedTraceback.render_as_textFc C s, ddl m} d|| |d| j jddf S )z)fullzutf-8replace)Zjinja2.debugrendererr7 r6 decode)r r8 r7 r r r render_as_htmlg s
z!ProcessedTraceback.render_as_htmlc C s t | jtS )z*`True` if this is a template syntax error.)
isinstancer- r )r r r r is_template_syntax_erroro s z+ProcessedTraceback.is_template_syntax_errorc C s | j | j| jd fS )z;Exception info tuple with a proxy around the frame objects.r )r, r- r. )r r r r exc_infot s zProcessedTraceback.exc_infoc C s* | j d }t|tk r |j}| j| j|fS )z'Standard python exc_info for re-raisingr )r. typer r r, r- )r r r r r standard_exc_infoy s
z$ProcessedTraceback.standard_exc_info)N)F) r r r r! r r6 r; r" r= r> r@ r r r r r+ P s
r+ c C s2 | \}}}t |tr$t||} d}nd}t| |S )z7Creates a processed traceback object from the exc_info.r )r< r translate_syntax_errortranslate_exception)r> Z source_hintr, r- r initial_skipr r r make_traceback s
rE c C s8 || _ d| _| j| df}| j}|dkr*d}t||| jS )z4Rewrites a syntax error to please traceback systems.TNz )sourceZ
translated __class__filename
fake_exc_infolineno)errorrF r> rH r r r rB s rB c C s | d }g }x t |D ]}|dk r|j}qW |}xz|dk r|jjtkrN|j}q2|j}|jjjd}|dk r|j|j}t | dd |f |j
|d }|j t | |}q2W |st
| d | d | d t| d | d |S )zIf passed an exc_info it will automatically rewrite the exceptions
all the way down to the correct line numbers and frames.
Nr r rA )ranger r f_coder r getZget_corresponding_lineno tb_linenorI rH appendr* r r+ ) r> rD r r. xZ
initial_tbr templaterJ r r r rC s*
rC c C s | j d}|r |j j }ni }i }xt| D ]x\}}|jd s.|tkrLq.y |jdd\}}}t|}W n tk
r w.Y nX |j |dd }||k r.||f||< q.W x6t|D ]*\}\}}|tkr|j |d q|||< qW |S ) NcontextZl__rL rA r )rV )
rO Zget_allcopyr
startswithr splitint
ValueErrorpop) Z real_localsZctxlocalsZlocal_overridesr valuerU depthZ cur_depthr r r get_jinja_locals s*
r` c
C s | \}}}|dk r,t |jj}|jdd ni }||| dd dd}td|d t |d}y|dkrnd} n8|jjj}
|
d krd
} n |
jd rd |
d
d } nd} t rt
d|j |j |j
|j|j|j|j|| |j|jf f }n8t
d|j|j |j |j
|j|j|j|j|| |j|jf f }W n$ tk
r> } zW Y dd} ~ X nX yt||| W n tj } | d j} Y nX | dd | f S )z!Helper for `translate_exception`.N__jinja_exception__rL )r __file__ra r
rA execrS rootztop-level template codeZblock_z
block "%s" r )r` r f_localsr\ compile raise_helperrN co_namerX r r
co_nlocals co_stacksizeco_flagsco_code co_constsco_names co_varnamesco_firstlineno co_lnotabco_kwonlyargcountr rd sysr> r )
r> rH rJ r, r- r r] globalscodelocationZfunctioneZnew_tbr r r rI sN
rI c s ddl ddlm tr4tjdr,j} q:j} nj} G dd dj }d| fdj
|fg|_ tt d rG d
d dj }d j
|fd j
|fd| fdj
|fg|_ G d
d d|dj
fdj
|fdjfdjfg_ fdd}|S )zThis function implements a few ugly things so that we can patch the
traceback objects. The function returned allows resetting `tb_next` on
any python traceback object. Do not attempt to use this on non cpython
interpreters
r N)r ZPy_InitModule4_64c @ s e Zd ZdS )z"_init_ugly_crap.._PyObjectN)r r r r r r r _PyObject> s rz ob_refcntob_typeZ
getobjectsc @ s e Zd ZdS )z"_init_ugly_crap.._PyObjectN)r r r r r r r rz G s Z_ob_nextZ_ob_prevc @ s e Zd ZdS )z#_init_ugly_crap.._TracebackN)r r r r r r r
_TracebackP s r} r r tb_lastirP c s t | o|dkpt | s$tdjt| }| jdk rZjt| j}| jd8 _|dkrrj |_n(jt|}| jd7 _j||_dS )z0Set the tb_next attribute of a traceback object.Nz/tb_set_next arguments must be traceback objectsrA )r< TypeErrorZ from_addressidr r{ POINTERZpointer)r r objold)r r} ctypesr r r Y s
z$_init_ugly_crap..tb_set_next)
r typesr r hasattrZ pythonapiZc_int64Zc_intZ c_ssize_tZ Structurer Z_fields_ru )Z _Py_ssize_trz r r )r r} r r _init_ugly_crap* s.
r )N)N)r )!r! ru r1 r r r Z jinja2.utilsr r Zjinja2.exceptionsr Zjinja2._compatr r r Z__pypy__r
ImportErrorrd SyntaxErrorri r objectr r* r+ rE rB rC r` rI r r r r r r s@
4
, FD