OwlCyberSecurity - MANAGER
Edit File: aws_lambda.cpython-38.opt-1.pyc
U ��\f � @ s� d dl Z d dlmZmZ d dlmZ d dlmZmZm Z m Z d dl m Z d dl mZ dd� Zd d � ZG d d � d e �Zd d� Zdd� ZdS )� N)�Hub�_should_send_default_pii)�reraise)�AnnotatedValue�capture_internal_exceptions�event_from_exception�logger)� Integration)�_filter_headersc s � fdd�}|S )Nc s� t j}|�t�}|d kr*� | |f|�|�S |�� ��}t� �& |�� |j|_|� t | |�� W 5 Q R X z � | |f|�|�W W 5 Q R � S t k r� t � � }t||jjddd�d�\} }|j| |d� t|� Y nX W 5 Q R X d S )N� aws_lambdaF)�typeZhandled)Zclient_optionsZ mechanism)�hint)r �current�get_integration�AwsLambdaIntegrationZ push_scoper Zclear_breadcrumbs� function_nameZ transactionZadd_event_processor� _make_request_event_processor� Exception�sys�exc_infor �client�optionsZ capture_eventr ) �event�context�args�kwargs�hub� integrationZscoper r ��handler� �S/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/integrations/aws_lambda.py�sentry_handler s( � z%_wrap_handler.<locals>.sentry_handlerr )r r" r r r! � _wrap_handler s r# c C s8 t � �( tj} | �t�}|d k r*| j�� W 5 Q R X d S �N)r r r r r r �flush)r r r r r! � _drain_queue+ s r& c @ s e Zd ZdZedd� �ZdS )r r c s� dd l } d}t| d�s<zdd l} d}W n tk r: Y nX t| d�sTt�d� d S |r�| j� � fdd�}|| _| j��fd d �}|| _| j��fd d �}|| _n@| j� � fd d�}|| _dd� }|| j j �| j _ || j j �| j _ d S )Nr T�handle_http_requestF�handle_event_requestzDNot running in AWS Lambda environment, AwsLambdaIntegration disabledc s t | �} � | f|�|�S r$ �r# ��request_handlerr r ��old_handle_event_requestr r! �sentry_handle_event_requestP s zDAwsLambdaIntegration.setup_once.<locals>.sentry_handle_event_requestc s t | �} � | f|�|�S r$ r) r* )�old_handle_http_requestr r! �sentry_handle_http_requestX s zCAwsLambdaIntegration.setup_once.<locals>.sentry_handle_http_requestc s t � � | |�S r$ �r& �r r )� old_to_jsonr r! �sentry_to_jsonc s z7AwsLambdaIntegration.setup_once.<locals>.sentry_to_jsonc s t |�}� | |f|�|�S r$ r) )Zlambda_runtime_clientr+ r r r, r r! r. k s ���c s � fdd�}|S )Nc s t � � | |�S r$ r1 r2 ��fr r! �innery s zKAwsLambdaIntegration.setup_once.<locals>._wrap_post_function.<locals>.innerr )r6 r7 r r5 r! �_wrap_post_functionx s z<AwsLambdaIntegration.setup_once.<locals>._wrap_post_function) �__main__�hasattrZ bootstrap� ImportErrorr Zwarningr( r' Zto_jsonZLambdaRuntimeClientZpost_invocation_resultZpost_invocation_error)Zlambda_bootstrapZpre_37r. r0 r4 r8 r )r- r/ r3 r! � setup_once8 sB � ��z AwsLambdaIntegration.setup_onceN)�__name__� __module__� __qualname__Z identifier� staticmethodr< r r r r! r 5 s r c s � �fdd�}|S )Nc s | � di �}� �� � j� j� j� jd�|d< | � di �}d�krL�d |d< t�� �|d< d�krn�d |d < d �kr�t�d �|d < ��d d �r�t d d ddddggi�|d< t � �r| � di �}��di ��d�}|d k r�||d< ��di ��d�}|d k �r||d< | S )N�extra)Zremaining_time_in_millisr �function_version�invoked_function_arn�aws_request_id�lambda�requestZ httpMethod�methodZurlZqueryStringParametersZ query_string�headersZbody� Zremz!raw�xr �data�userZidentityZuserArn�idZsourceIpZ ip_address) � setdefaultZ get_remaining_time_in_millisr rB rC rD �_get_urlr �getr r )r r rA rF Z user_inforM Zip�� aws_context� aws_eventr r! �event_processor� s4 � z6_make_request_event_processor.<locals>.event_processorr )rS rR rT r rQ r! r � s )r c C sV | � dd �}| � di �}|� dd �}|� dd �}|rJ|rJ|rJd�|||�S d�|j�S )N�pathrH ZHostzX-Forwarded-Protoz {}://{}{}zawslambda:///{})rP �formatr )r r rU rH �host�protor r r! rO � s rO )r Zsentry_sdk.hubr r Zsentry_sdk._compatr Zsentry_sdk.utilsr r r r Zsentry_sdk.integrationsr Z$sentry_sdk.integrations._wsgi_commonr r# r& r r rO r r r r! �<module> s R-