B ��Y�(�@sfdZddlmZddlZddlZddlmZddlm Z ddl mZddlm Z ddlmZdd lmZydd lmZWn ek r�ddlmZYnXddlmZdd lmZmZddlmZddlmZm Z m!Z!ddl"m#Z#ddl$m%Z%m&Z&m'Z'ddl(m)Z)ddl*m+Z+dZ,edk�r&dd�Z-ndd�Z-Gdd�de.�Z/dd�Z0e)dd��Z1Gdd �d e�Z2dS)!z� raven.contrib.django.client ~~~~~~~~~~~~~~~~~~~~~~~~~~~~ :copyright: (c) 2010-2012 by the Sentry Team, see AUTHORS for more details. :license: BSD, see LICENSE for more details. �)�absolute_importN)�VERSION)�settings)�SuspiciousOperation)�HttpRequest)�TemplateSyntaxError)�MultiValueDict)�Origin)�LoaderOrigin)�Client)�get_data_from_template�get_host)�SentryMiddleware)�string_types�binary_type� iterlists)� RouteResolver)�get_headers�get_environ� get_client_ip)�once)�breadcrumbs)�DjangoClient)�� cCs|��S)N)�is_authenticated)�request_user�r�L/opt/alt/python37/lib/python3.7/site-packages/raven/contrib/django/client.pyr*srcCs|jS)N)r)rrrrr-sc@seZdZdd�Zdd�ZdS)�_FormatConvertercCs||_g|_dS)N)� param_mapping�params)�selfr rrr�__init__3sz_FormatConverter.__init__cCs|j�|j�|��dS)Nz%s)r!�appendr �get)r"�valrrr�__getitem__7sz_FormatConverter.__getitem__N)�__name__� __module__�__qualname__r#r'rrrrr1srcCs�g}t|t�r.t|�}|r*||}|j}nd}x�|p6dD]z}|dkrP|�d�q8t|t�r�t|t�rp|�dd�}t|�dkr�|dd�d}|�d|� dd ��q8|�t |��q8W||fS) Nr�NULLzutf-8�replace�u…z'%s'�'z'')� isinstance�dictrr!r$rr�decode�lenr,�repr)�sqlr!�rv�conv�paramrrr� format_sql<s$ r8cs�yddlm}Wn tk r0ddlm}YnXy|j�|j�Wntk rVdSXdd��fdd��d ��fdd � }��fd d�}||_||_t�d�dS)z9If installed this causes Django's queries to be captured.r)� CursorWrapperNcs��fdd�}tj|d�dS)Ncs2t��\}}|r|t|�}|�|dd��dS)N�query)�message�category)r8�tuple�update)�dataZreal_sqlZreal_params)r!r4rr� processorgsz7install_sql_hook..record_sql..processor)r@)r�record)�vendor�alias�start�durationr4r!r@r)r!r4r� record_sqlfs z$install_sql_hook..record_sqlcs0t��|}x|D]}�||||||�qWdS)N)�time)rBrCrDr4� param_listrEr!)rFrr�record_many_sqlss z)install_sql_hook..record_many_sqlc sDt��}z�|||�S�|jjt|jdd�|t��|||�XdS)NrC)rG�dbrB�getattr)r"r4r!rD)�real_executerFrr�executexs z!install_sql_hook..executecs:t��}z�|||�S�|jjt|jdd�|||�XdS)NrC)rGrJrBrK)r"r4rHrD)�real_executemanyrIrr�executemany�s z%install_sql_hook..executemanyzdjango.db.backends)N) Zdjango.db.backends.utilsr9�ImportErrorZdjango.db.backends.utilrMrO�AttributeErrorrZ ignore_logger)r9rMrOr)rLrNrIrFr�install_sql_hookVs rRcsbeZdZe�d�Ze�Zdd�Zdd�Z dd�Z dd �Z�fd d�Zd�fd d� Z dd�Z�ZS)rzsentry.errors.client.djangocOs.|�dd�}tj|f|�|�|r*|��dS)NrRT)�poprr#rR)r"�args�kwargsrRrrrr#�szDjangoClient.__init__cCs t�dS)N)rR)r"rrrrR�szDjangoClient.install_sql_hookcCs�dt|j�i}t|dd�}|dkr&|Sy^t|�}|s8|S|j|d<t|d�rV|j|d<t|d�rn|��|d<nt|d�r�|j|d<Wnt k r�YnX|S)NZ ip_address�user�id�email�get_username�username) r�METArKr�pk�hasattrrXrYrZ� Exception)r"�requestZ user_inforVZ authenticatedrrr� get_user_info�s$ zDjangoClient.get_user_infocCs>i}|�|�|d<y|��}Wn>tk r\|��rcss.|]&\}}|t|�dkr |dn|fVqdS)rrN)r2)�.0�k�vrrr� �sz5DjangoClient.get_data_from_request..r_ZQUERY_STRING)�method�urlZquery_stringr?Zcookies�headers�env)r`Zbuild_absolute_urirZ is_securer �pathrh�bodyr^Z raw_post_dataZPOSTr/rr0rr[r>r%ZCOOKIESrr)r"r_�resultZuri�scheme�hostr?�environrrr�get_data_from_request�sJ z"DjangoClient.get_data_from_requestc s�tt|�j||�}x4|�|�D]&}|�d�}|s2q|�d�rd|d<qW|js�dtjkr�y6ddl m }|j��}|j pz|j}|d�d |�Wn@tk r�y|d�d tj�Wntk r�YnXYnX|S) N�modulezdjango.FZin_appzdjango.contrib.sitesr)�SiteZtags�site)�superr� build_msgZ_iter_framesr%� startswithrurZINSTALLED_APPSZdjango.contrib.sites.modelsrt�objectsZget_current�name�domain� setdefaultr^ZSITE_IDrQ) r"rTrUr?�framersrtruZ site_name)� __class__rrrw�s& zDjangoClient.build_msgNc s,d|kri|d<}n|d}|dkr4ttjdd�}t|t�}|rR|�|�|��|�d�r�|dd}t|d�s�t|t �r�tt|dd�t tf�r�t|jdt �s�t|d�r�t|dt|dd��}t|dd�}|dkr�|j�d �|�t||��tt|�j|f|�} |�r(| �r(|�d |jj�| d�|_| S)Nr?r_�exc_inforZdjango_template_source�sourcerZtemplate_debugz,Unable to get template source from exception�project)Z project_idrW)rKr�threadr/rr>rrr%r]rr=�listr�r �logger�inforrvr�captureZremoter�Zsentry) r"Z event_typer_rUr?Zis_http_request� exc_valuer��debugrn)r~rrr�s6 zDjangoClient.capturecCs|j�|j�S)N)�resolver�resolverl)r"r_rrr�get_transaction_from_request4sz)DjangoClient.get_transaction_from_request)N)r(r)r*�logging� getLoggerr�rr�r#rRr`rrrwr�r�� __classcell__rr)r~rr�s 5,r)3�__doc__� __future__rrGr�ZdjangorZDJANGO_VERSIONZdjango.confrZdjango.core.exceptionsrZdjango.httprZdjango.templaterZdjango.utils.datastructuresrZdjango.template.baser rPZdjango.template.loaderr Z raven.baserZraven.contrib.django.utilsrr Zraven.contrib.django.middlewarerZraven.utils.compatrrrZraven.contrib.django.resolverrZraven.utils.wsgirrrZraven.utilsr�ravenr�__all__r�objectrr8rRrrrrr�s: 7