U ��\f�@s�ddlZddlmZmZddlmZddlmZmZm Z m Z ddlmZddl mZdd�Zd d �ZGdd�de�Zd d�Zdd�ZdS)�N)�Hub�_should_send_default_pii)�reraise)�AnnotatedValue�capture_internal_exceptions�event_from_exception�logger)�Integration)�_filter_headerscs�fdd�}|S)Nc s�tj}|�t�}|dkr*�||f|�|�S|��}t��&|��|j|_|� t ||��W5QRXz �||f|�|�WW5QR�Stk r�t� �}t||jjddd�d�\}}|j||d�t|�YnXW5QRXdS)N� aws_lambdaF)�typeZhandled)Zclient_optionsZ mechanism)�hint)r�current�get_integration�AwsLambdaIntegrationZ push_scoperZclear_breadcrumbs� function_nameZtransactionZadd_event_processor�_make_request_event_processor� Exception�sys�exc_infor�client�optionsZ capture_eventr) �event�context�args�kwargs�hub�integrationZscoperr ��handler��S/opt/alt/python38/lib/python3.8/site-packages/sentry_sdk/integrations/aws_lambda.py�sentry_handlers( � z%_wrap_handler..sentry_handlerr )rr"r rr!� _wrap_handlersr#c Cs8t��(tj}|�t�}|dk r*|j��W5QRXdS�N)rrrrrr�flush)rrr r r!�_drain_queue+s r&c@seZdZdZedd��ZdS)rrcs�ddl}d}t|d�s.sentry_handle_event_requestcst|�}�|f|�|�Sr$r)r*)�old_handle_http_requestr r!�sentry_handle_http_requestXszCAwsLambdaIntegration.setup_once..sentry_handle_http_requestcst��||�Sr$�r&�rr)�old_to_jsonr r!�sentry_to_jsoncsz7AwsLambdaIntegration.setup_once..sentry_to_jsoncst|�}�||f|�|�Sr$r))Zlambda_runtime_clientr+rrr,r r!r.ks��cs�fdd�}|S)Ncst��||�Sr$r1r2��fr r!�inneryszKAwsLambdaIntegration.setup_once.._wrap_post_function..innerr )r6r7r r5r!�_wrap_post_functionxsz._wrap_post_function)�__main__�hasattrZ bootstrap�ImportErrorrZwarningr(r'Zto_jsonZLambdaRuntimeClientZpost_invocation_resultZpost_invocation_error)Zlambda_bootstrapZpre_37r.r0r4r8r )r-r/r3r!� setup_once8sB ��zAwsLambdaIntegration.setup_onceN)�__name__� __module__�__qualname__� identifier�staticmethodr<r r r r!r5srcs��fdd�}|S)Ncs|�di�}��j�j�j�jd�|d<|�di�}d�krL�d|d<t��|d<d�krn�d|d <d �kr�t�d �|d <��dd�r�t dd ddddggi�|d<t ��r|�di�}��di��d�}|dk r�||d<��di��d�}|dk �r||d<|S)N�extra)Zremaining_time_in_millisr�function_version�invoked_function_arn�aws_request_id�lambda�requestZ httpMethod�method�urlZqueryStringParametersZquery_string�headers�body��remz!raw�xr�data�userZidentityZuserArn�idZsourceIpZ ip_address)� setdefaultZget_remaining_time_in_millisrrCrDrE�_get_urlr �getrr)rr rBrGZ user_inforQZip��aws_context� aws_eventr r!�event_processor�s4� z6_make_request_event_processor..event_processorr )rWrVrXr rUr!r�s)rcCsV|�dd�}|�di�}|�dd�}|�dd�}|rJ|rJ|rJd�|||�Sd�|j�S)N�pathrJZHostzX-Forwarded-Protoz {}://{}{}zawslambda:///{})rT�formatr)rrrYrJ�host�protor r r!rS�srS)rZsentry_sdk.hubrrZsentry_sdk._compatrZsentry_sdk.utilsrrrrZsentry_sdk.integrationsr Z$sentry_sdk.integrations._wsgi_commonr r#r&rrrSr r r r!�s R-