Edit File: chalice.cpython-37.pyc
B Z~�_� � @ s d dl Z d dlmZ d dlmZ d dlmZmZ d dlm Z d dl m Z m Z d dl mZ d dlmZ d dlZd d lmZmZ d d lmZ er�d d lmZ d d lmZ d d lmZ ededef d�Zyd dlm Z W n e k r� ed��Y nX G dd� de�Zdd� Z G dd� de�Z dS )� N)�reraise)�Hub)� Integration� DidNotEnable)� _make_request_event_processor)�capture_internal_exceptions�event_from_exception)�MYPY)�wraps)�Chalice�ChaliceViewError)�EventSourceHandler)�Any)�TypeVar)�Callable�F.)Zbound)� __version__zChalice is not installedc @ s e Zd Zdd� ZdS )r c C s� t j}|j}|�� ��}t� � |�� }|�t|||�� W d Q R X yt� | ||�S t k r� t � � }t ||jddd�d�\}}|j||d� |�� t|� Y nX W d Q R X d S )N�chaliceF)�type�handled)�client_options� mechanism)�hint)r �current�client� push_scoper � get_remaining_time_in_millis�add_event_processorr �ChaliceEventSourceHandler�__call__� Exception�sys�exc_infor �options� capture_event�flushr ) �self�event�context�hubr �scope�configured_timer" r � r, �P/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/chalice.pyr s$ zEventSourceHandler.__call__N)�__name__� __module__� __qualname__r r, r, r, r- r s r c s t ��� �fdd��}|S )Nc s� t j}|j}|�� ��}t� �4 � j�� }� jj|_|� t � j � � � j|�� W d Q R X y �f | �S t k r� } zJt|t�r~� t�� }t||jddd�d�\}}|j||d� |�� � W d d }~X Y nX W d Q R X d S )Nr F)r r )r r )r )r r r r r Zlambda_contextr Z function_nameZ transactionr r Zcurrent_requestZto_dictr � isinstancer r! r" r r# r$ r% ) � function_argsr) r r* r+ �excr" r' r )�app� view_functionr, r- �wrapped_view_function; s0 z:_get_view_function_response.<locals>.wrapped_view_function)r )r4 r5 r2 r6 r, )r4 r5 r- �_get_view_function_response9 s r7 c @ s e Zd ZdZedd� �ZdS )�ChaliceIntegrationr c s� y t ttt�d�d d� ��} W n& ttfk rF td�t���Y nX | dk rXt j � nddl m } |j � � fdd�}| dk r�|t _ n||_ t tj_ d S ) N�.� z Unparsable Chalice version: {})� � r )�RestAPIEventHandlerc s t | ||�}� | ||�S )N)r7 )r4 r5 r2 r6 )� old_get_view_function_responser, r- �sentry_event_responsep s z<ChaliceIntegration.setup_once.<locals>.sentry_event_response)�tuple�map�int�CHALICE_VERSION�split� ValueError� TypeErrorr �formatr r7 � chalice.appr= r r r4 )�versionr= r? r, )r> r- � setup_once` s z ChaliceIntegration.setup_onceN)r. r/ r0 � identifier� staticmethodrJ r, r, r, r- r8 ] s r8 )!r! Zsentry_sdk._compatr Zsentry_sdk.hubr Zsentry_sdk.integrationsr r Z"sentry_sdk.integrations.aws_lambdar Zsentry_sdk.utilsr r Zsentry_sdk._typesr Zsentry_sdk._functoolsr r r r rH r r �typingr r r r r rC � ImportErrorr7 r8 r, r, r, r- �<module> s, $