관리-도구
편집 파일: aws_lambda.cpython-37.pyc
B T a�; � @ 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.)�boundi� 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_errorr1 )r0 r3 r1 )r0 r2 �_wrap_init_error# s r4 c s � fdd�}|S )Nc s� t | t�r | d }t| �}n| }d}t |t�s2i }tj}|�t�}|d kr\� | |f|�|�S |j}|� � } |� � ��d} d } t � �� | � � | � t||| �� | �d|j�d�d � |dkr�| �dd� | �d|� |j�r| tk�r| t t } t| | t �} | �� W d Q R X |�d �} | d k�r*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_datar8 r, r- r# Zconfigured_timer. Ztimeout_threadZ waiting_timer9 Z transactionr& r/ r )�handlerr1 r2 �sentry_handlerD sl z%_wrap_handler.<locals>.sentry_handlerr1 )rN rO r1 )rN r2 � _wrap_handlerB s brP 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- r1 r1 r2 � _drain_queue� s rR c @ s& e Zd ZdZddd�Zedd� �ZdS ) r" r Fc C s || _ d S )N)rF )�selfrF r1 r1 r2 �__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)rP )�request_handlerr* r+ )�old_handle_event_requestr1 r2 �sentry_handle_event_request� s zDAwsLambdaIntegration.setup_once.<locals>.sentry_handle_event_requestc s t | �} � | f|�|�S )N)rP )rW r* r+ )�old_handle_http_requestr1 r2 �sentry_handle_http_request� s zCAwsLambdaIntegration.setup_once.<locals>.sentry_handle_http_requestc s t � � | |�S )N)rR )r* r+ )� old_to_jsonr1 r2 �sentry_to_json� s z7AwsLambdaIntegration.setup_once.<locals>.sentry_to_jsonc s t |�}� | |f|�|�S )N)rP )Zlambda_runtime_clientrW r* r+ )rX r1 r2 rY � s c s � fdd�}|S )Nc s t � � | |�S )N)rR )r* r+ )�fr1 r2 �inner s zKAwsLambdaIntegration.setup_once.<locals>._wrap_post_function.<locals>.innerr1 )r^ r_ r1 )r^ r2 �_wrap_post_function s z<AwsLambdaIntegration.setup_once.<locals>._wrap_post_function) �get_lambda_bootstrapr �warning�hasattrrU rV Zto_jsonr4 ZLambdaRuntimeClientZpost_init_errorZpost_invocation_resultZpost_invocation_error)Zlambda_bootstrapZpre_37rY r[ r] r` r1 )rX rZ r\ r2 � setup_once� s< z AwsLambdaIntegration.setup_onceN)F)�__name__� __module__� __qualname__� identifierrT � staticmethodrd r1 r1 r1 r2 r" � s r" c C sL dt jkrt jd S dt jkrDtt jd d�r:t jd jS t jd S d S d S )N� bootstrap�__main__)r% �modulesrc rj r1 r1 r1 r2 ra s ra c s t �� }|f� ��fdd� }|S )Nc sh � � � }�| }| �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 � �r2| �d i �}�� d�}|d kr�i }|� d�} | d k r�|�d| � |� d�} | d k �r|�d| � d�k�r\�� dd�|d< n*�� dd ��r\t ddddddggi�|d< || d< | S )N�extra)rK �function_versionrD �aws_request_idZ execution_duration_in_millisZremaining_time_in_millis�lambda)�url� log_group� log_streamzcloudwatch logs�requestZ httpMethod�methodrq ZqueryStringParametersZ query_stringr9 �user�identityZuserArn�idZsourceIpZ ip_address�body� �data�remz!raw�xr )rB � setdefaultrK rn rD ro �_get_cloudwatch_logs_url�log_group_name�log_stream_namerJ �_get_urlr r r ) r/ r � start_timeZremaining_time_in_milisZ exec_durationrm rt Z user_inforw rx Zip)r= r< �configured_timeoutr1 r2 �event_processor8 sJ z6_make_request_event_processor.<locals>.event_processor)r �utcnow)r< r= r� r� r� r1 )r= r< r� r2 rC 4 s =rC c C s` | � dd �}| � d�}|d kr"i }|� dd �}|� dd �}|rT|rT|rTd�|||�S d�|j�S )N�pathr9 ZHostzX-Forwarded-Protoz {}://{}{}zawslambda:///{})rJ �formatrK )r< r= r� r9 �host�protor1 r1 r2 r� x s r� c C s` d}t �dd�}dj|�d�r"dnd|| j| j|tdd � �|�t� � td d � �|�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 AWS_REGIONrz z�https://console.{domain}/cloudwatch/home?region={region}#logEventViewer:group={log_group};stream={log_stream};start={start_time};end={end_time}zcn-z amazonaws.cnzaws.amazon.comr5 )�seconds� )�domain�regionrr rs r� �end_time) r rJ r� � startswithr� r� r �strftimer r� )r= r� � formatstringr� rq r1 r1 r2 r � 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 rG rH r4 rP rR r" ra rC r� r r1 r1 r1 r2 �<module> s4 g c D