U ��\h�@sddlmZddlZddlmZmZddlmZmZddl m Z ddlmZddl mZzddlZWnek r|dZYnXddlmZmZmZmZdd lmZm Z m!Z!m"Z"Gd d�de �Z#dd �Z$dd�Z%dd�Z&Gdd�de�Z'dd�Z(dd�Z)dd�Z*dS)�)�absolute_importN)�Hub�_should_send_default_pii)�capture_internal_exceptions�event_from_exception)�Integration)�SentryWsgiMiddleware)�RequestExtractor)�Request�Flask�_request_ctx_stack�_app_ctx_stack)�appcontext_pushed�appcontext_tearing_down�got_request_exception�request_startedc@s*eZdZdZdZddd�Zedd��ZdS) �FlaskIntegration�flaskN�endpointcCs&d}||krtd||f��||_dS)N)r�urlz7Invalid value for transaction_style: %s (must be in %s))� ValueError�transaction_style)�selfrZTRANSACTION_STYLE_VALUES�r�N/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/integrations/flask.py�__init__(s��zFlaskIntegration.__init__csDt�t�t�t�t�t�t�t�t j ��fdd�}|t _ dS)Ncs4tj�t�dkr��||�St��fdd��||�S)Ncs��f|�|�S�Nr)�a�kw)�old_apprrr�A�zNFlaskIntegration.setup_once..sentry_patched_wsgi_app..)r�current�get_integrationrr)r�environZstart_response�r�rr�sentry_patched_wsgi_app<s�z.sentry_patched_wsgi_app)r�connect�_push_appctxr�_pop_appctxr�_request_startedr�_capture_exceptionr�__call__)r'rr%r� setup_once2s zFlaskIntegration.setup_once)r)�__name__� __module__�__qualname__� identifierrr�staticmethodr.rrrrr#s rc OsJtj}|�t�dk rF|��}|��|tj_|� ��}d|_ W5QRXdS)Nr)rr"r#rZ push_scope� __enter__r �top�sentry_sdk_scope_manager�configure_scope�_name)�args�kwargs�hub� scope_manager�scoperrrr)Hs r)cOs(ttjdd�}|dk r$|�ddd�dS)Nr6)�getattrr r5�__exit__)r9r:r<rrrr*Usr*c Ks\tj}|�t�}|dkrdSt�tjj�}t jj }|��}|�t |||��W5QRXdSr)rr"r#r�weakref�refrr5�requestr �appr7Zadd_event_processor�_make_request_event_processor)�senderr:r;�integration�weak_requestrCr=rrrr+\s ��r+c@sLeZdZdd�Zdd�Zdd�Zdd�Zd d �Zdd�Zd d�Z dd�Z dS)�FlaskRequestExtractorcCs|jjSr)rBr$r&rrr�envnszFlaskRequestExtractor.envcCs|jjSr)rB�cookiesr&rrrrJrszFlaskRequestExtractor.cookiescCs|jjSr)rB�datar&rrr�raw_datavszFlaskRequestExtractor.raw_datacCs|jjSr)rB�formr&rrrrMzszFlaskRequestExtractor.formcCs|jjSr)rB�filesr&rrrrN~szFlaskRequestExtractor.filescCs|jjSr)rB�is_jsonr&rrrrO�szFlaskRequestExtractor.is_jsoncCs |j��Sr)rBZget_jsonr&rrr�json�szFlaskRequestExtractor.jsoncCs|jSr)Zcontent_length)r�filerrr�size_of_file�sz"FlaskRequestExtractor.size_of_fileN)r/r0r1rIrJrLrMrNrOrPrRrrrrrHmsrHcs��fdd�}|S)Nc s��}|dkr|Sz2�jdkr,|jj|d<n�jdkrB|jj|d<Wntk rXYnXt��t|��|�W5QRXt�r�t��t |�W5QRX|S)NrZtransactionr) rZurl_rulerZrule� ExceptionrrHZextract_into_eventr�_add_user_to_event)�event�hintrB�rFrGrr�inner�s z,_make_request_event_processor..innerr)rCrGrFrXrrWrrD�srDcKsFtj}|�t�dkrdSt||jjddd�d�\}}|j||d�dS)NrF)�typeZhandled)Zclient_optionsZ mechanism)rV)rr"r#rrZclient�optionsZ capture_event)rE� exceptionr:r;rUrVrrrr,�s� r,c Cs�tdkrdStj}|dkrdSt��|�di�}z|��|d<Wntk rVYnXz|j|d<|d<Wntk r�YnXz|j|d<Wntk r�YnXW5QRXdS)N�user�id�email�username) �flask_loginZcurrent_userr� setdefaultZget_id�AttributeErrorr^rSr_)rUr\Z user_inforrrrT�s& rT)+� __future__rr@Zsentry_sdk.hubrrZsentry_sdk.utilsrrZsentry_sdk.integrationsrZsentry_sdk.integrations.wsgirZ$sentry_sdk.integrations._wsgi_commonr Z_ScopedResponseZAny�DictZImmutableTypeConversionDictZImmutableMultiDictZFileStorage�Union�Callabler`�ImportErrorrr rrr Z flask.signalsrrrrrr)r*r+rHrDr,rTrrrr�s( %