Edit File: aws_lambda.cpython-37.pyc
B Z~�_9 � @ s< d dl m Z mZ d dlmZ d dlZd dlmZmZ d dlm Z d dl m Z d dl m Z mZmZmZmZ d dlmZ d d lmZ d d lmZ er�d d lmZ d d lmZ d d lmZ d dlm Z d dlm Z m Z m Z ededef d�Z dZ!dZ"dd� Z#dd� Z$dd� Z%G dd � d e�Z&d d � Z'd d � Z(d!d"� Z)d#d$� Z*dS )%� )�datetime� timedelta)�environN)�Hub�_should_send_default_pii)� Transaction)�reraise)�AnnotatedValue�capture_internal_exceptions�event_from_exception�logger� TimeoutThread)� Integration)�_filter_headers)�MYPY)�Any)�TypeVar)�Callable)�Optional)�EventProcessor�Event�Hint�F.)Zboundi� g @�@c s � fdd�}|S )Nc s� t j}|�t�}|d kr"� | |�S |j}t� �^ |�� �}|�� W d Q R X t� � }|r�t |�r�t ||j ddd�d�\}}|j ||d� W d Q R X � | |�S )N� aws_lambdaF)�type�handled)�client_options� mechanism)�hint)r �current�get_integration�AwsLambdaIntegration�clientr Zconfigure_scope�clear_breadcrumbs�sys�exc_info�allr �options� capture_event) �args�kwargs�hub� integrationr" �scoper% � sentry_eventr )� init_error� �S/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/aws_lambda.py�sentry_init_error% s z+_wrap_init_error.<locals>.sentry_init_errorr0 )r/ r2 r0 )r/ r1 �_wrap_init_error# s r3 c s � fdd�}|S )Nc s� t | t�r | d }t| �}n| }d}t |t�s2i }tj}|�t�}|d kr\� | |f|�|�S |j}|� � } |� � ��X} t � �� | � � | � t||| �� | �d|j�d�d � |dkr�| �dd� | �d|� d } |j�r| tk�r| t t } t| | t �} | �� W d Q R X |�d i �} tj| d |jd �}|j|| |d �d ��~ zfy� | |f|�|�S t k �r� t � � }t ||j ddd�d�\}}|j!||d� t"|� Y nX W d | �r�| �#� X W d Q R X W d Q R X d S )Nr � Z aws_region�:� Z batch_requestT� batch_size�headerszserverless.function)�op�name)� aws_event� aws_context)Zcustom_sampling_contextr F)r r )r r )r )$� isinstance�list�len�dictr r r r! r" � get_remaining_time_in_millisZ push_scoper r# Zadd_event_processor� _make_request_event_processorZset_tag�invoked_function_arn�split�timeout_warning�TIMEOUT_WARNING_BUFFER�MILLIS_TO_SECONDSr �start�getr Zcontinue_from_headers� function_nameZstart_transaction� Exceptionr$ r% r r' r( r �stop)r; r< r) r* Z request_datar7 r+ r, r" Zconfigured_timer- Ztimeout_threadZ waiting_timer8 Z transactionr% r. r )�handlerr0 r1 �sentry_handlerD sh z%_wrap_handler.<locals>.sentry_handlerr0 )rM rN r0 )rM r1 � _wrap_handlerB s _rO c C s6 t � �& tj} | �t�}|d k r(| �� W d Q R X d S )N)r r r r r! �flush)r+ r, r0 r0 r1 � _drain_queue� s rQ c @ s& e Zd ZdZddd�Zedd� �ZdS ) r! r Fc C s || _ d S )N)rE )�selfrE r0 r0 r1 �__init__� s z AwsLambdaIntegration.__init__c s� t � } | st�d� d S t| d�s0t�d� d S t| d�}|r�| j� � fdd�}|| _| j��fdd�}|| _| j��fd d �}|| _nPt| jj �| j_ | j� � fd d�}|| _d d � }|| jj �| j_ || jj �| j_ d S )NzfNot running in AWS Lambda environment, AwsLambdaIntegration disabled (could not find bootstrap module)�handle_event_requestzjNot running in AWS Lambda environment, AwsLambdaIntegration disabled (could not find handle_event_request)�handle_http_requestc s t | �} � | f|�|�S )N)rO )�request_handlerr) r* )�old_handle_event_requestr0 r1 �sentry_handle_event_request� s zDAwsLambdaIntegration.setup_once.<locals>.sentry_handle_event_requestc s t | �} � | f|�|�S )N)rO )rV r) r* )�old_handle_http_requestr0 r1 �sentry_handle_http_request� s zCAwsLambdaIntegration.setup_once.<locals>.sentry_handle_http_requestc s t � � | |�S )N)rQ )r) r* )� old_to_jsonr0 r1 �sentry_to_json� s z7AwsLambdaIntegration.setup_once.<locals>.sentry_to_jsonc s t |�}� | |f|�|�S )N)rO )Zlambda_runtime_clientrV r) r* )rW r0 r1 rX � s c s � fdd�}|S )Nc s t � � | |�S )N)rQ )r) r* )�fr0 r1 �inner s zKAwsLambdaIntegration.setup_once.<locals>._wrap_post_function.<locals>.innerr0 )r] r^ r0 )r] r1 �_wrap_post_function� s z<AwsLambdaIntegration.setup_once.<locals>._wrap_post_function) �get_lambda_bootstrapr Zwarning�hasattrrT rU Zto_jsonr3 ZLambdaRuntimeClientZpost_init_errorZpost_invocation_resultZpost_invocation_error)Zlambda_bootstrapZpre_37rX rZ r\ r_ r0 )rW rY r[ r1 � setup_once� s< z AwsLambdaIntegration.setup_onceN)F)�__name__� __module__� __qualname__� identifierrS � staticmethodrb r0 r0 r0 r1 r! � s r! c C s0 dt jkrt jd S dt jkr(t jd S d S d S )NZ bootstrap�__main__)r$ �modulesr0 r0 r0 r1 r` s r` c s t �� }|f� ��fdd� }|S )Nc sb � � � }�| }| �di �}� j� j� j� j||d�|d< t� |�� j� jd�|d< | � di �}d�krt�d |d< t �� �|d < d �kr��d |d < d �kr�t �d �|d < t � �r,| �d i �}�� di �� d�}|d k r�|�d|� �� di �� d�} | d k �r|�d| � d�k�rV�� dd�|d< n*�� dd ��rVt ddddddggi�|d< || d< | S )N�extra)rJ �function_versionrC �aws_request_idZ execution_duration_in_millisZremaining_time_in_millis�lambda)�url� log_group� log_streamzcloudwatch logs�requestZ httpMethod�methodrn ZqueryStringParametersZ query_stringr8 �userZidentityZuserArn�idZsourceIpZ ip_address�body� �data�remz!raw�xr )rA � setdefaultrJ rk rC rl �_get_cloudwatch_logs_url�log_group_name�log_stream_namerI �_get_urlr r r ) r. r � start_timeZremaining_time_in_milisZ exec_durationrj rq Z user_infort Zip)r< r; �configured_timeoutr0 r1 �event_processor/ sD z6_make_request_event_processor.<locals>.event_processor)r �utcnow)r; r< r� r r� r0 )r< r; r� r1 rB + s 9rB c C sV | � dd �}| � di �}|� dd �}|� dd �}|rJ|rJ|rJd�|||�S d�|j�S )N�pathr8 ZHostzX-Forwarded-Protoz {}://{}{}zawslambda:///{})rI �formatrJ )r; r< r� r8 �host�protor0 r0 r1 r~ k s r~ c C sJ d}dj t�d�| j| j|tdd� �|�t�� tdd� �|�d�}|S )z� Generates a CloudWatchLogs console URL based on the context object Arguments: aws_context {Any} -- context from lambda handler Returns: str -- AWS Console URL to logs. z%Y-%m-%dT%H:%M:%SZz�https://console.aws.amazon.com/cloudwatch/home?region={region}#logEventViewer:group={log_group};stream={log_stream};start={start_time};end={end_time}Z AWS_REGIONr4 )�seconds� )Zregionro rp r Zend_time) r� r rI r| r} r �strftimer r� )r< r � formatstringrn r0 r0 r1 r{ v s r{ )+r r �osr r$ Zsentry_sdk.hubr r Zsentry_sdk.tracingr Zsentry_sdk._compatr Zsentry_sdk.utilsr r r r r Zsentry_sdk.integrationsr Z$sentry_sdk.integrations._wsgi_commonr Zsentry_sdk._typesr �typingr r r r r r r r rF rG r3 rO rQ r! r` rB r~ r{ r0 r0 r0 r1 �<module> s4 d c@