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 e�Zd
d� ZG dd� de�Zd dd�Zd dd�Zd dd�Z dd� Z dd� Z dd � Z dZ edk�r@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)�
TracebackType�CodeType)�missing�
internal_code)�TemplateSyntaxError)� iteritems�reraise�PY2)�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 operation�args�kwargs)�proxyr r �operation_handlerF s
z+make_frame_proxy.<locals>.operation_handler)r r
r )�framer( r )r' r �make_frame_proxyB s
r* c @ sP e Zd ZdZdd� Zddd�Zddd �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_value�framesr )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� )� traceback�format_exceptionr, r- r. �join�rstrip)r r/ �linesr r r �render_as_texta s z!ProcessedTraceback.render_as_textFc C s, ddl m} d|| |d�| j� jdd�f S )z<Return a unicode string with the traceback as rendered HTML.r )�render_tracebackz%s
<!--
%s
-->)�fullzutf-8�replace)Zjinja2.debugrendererr7 r6 �decode)r r8 r7 r r r �render_as_htmlg s
z!ProcessedTraceback.render_as_htmlc C s t | jt�S )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 |t�r$t||�} d}nd}t| |�S )z7Creates a processed traceback object from the exc_info.r � )r< r �translate_syntax_error�translate_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||| j�S )z4Rewrites a syntax error to please traceback systems.TNz <unknown>)�sourceZ
translated� __class__�filename�
fake_exc_info�lineno)�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 |s�t
| d | d | d � t| d | d |�S )z�If 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 }x�t| �D ]x\}}|jd� s.|tkrLq.y |jdd�\}}}t|�}W n tk
r� w.Y nX |j |d�d }||k r.||f||< q.W x6t|�D ]*\}\}}|tkr�|j |d � q�|||< q�W |S ) N�contextZl_�_rL rA r ���)rV )
rO Zget_all�copyr �
startswithr �split�int�
ValueError�pop) Z real_localsZctx�localsZlocal_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 kr�d
} n |
jd �r�d |
d
d� } nd} t r�t
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_stacksize�co_flags�co_code� co_consts�co_names� co_varnames�co_firstlineno� co_lnotab�co_kwonlyargcountr rd �sysr> r )
r> rH rJ r, r- r r] �globals�code�locationZfunction�eZnew_tbr r r rI � sN
rI c s ddl �ddlm� tr4t�jd�r,�j} q:�j} n�j} G dd� d�j �}d| fd�j
|�fg|_ tt d �r�G d
d� d�j �}d �j
|�fd �j
|�fd| fd�j
|�fg|_ G d
d� d|��d�j
��fd�j
|�fd�jfd�jfg�_ � ��fdd�}|S )z�This 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.<locals>._PyObjectN)r r r r r r r � _PyObject> s rz � ob_refcnt�ob_typeZ
getobjectsc @ s e Zd ZdS )z"_init_ugly_crap.<locals>._PyObjectN)r r r r r r r rz G s Z_ob_nextZ_ob_prevc @ s e Zd ZdS )z#_init_ugly_crap.<locals>._TracebackN)r r r r r r r �
_TracebackP s r} r r �tb_lastirP c s� t | � �o|dkpt |� �s$td���jt| ��}| jdk rZ�jt| j��}| jd8 _|dkrr�j��� |_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_address�idr r{ �POINTERZpointer)r r �obj�old)r r} �ctypesr r r Y s
z$_init_ugly_crap.<locals>.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 �<module> s@
4
, FD