관리-도구
편집 파일: aws_lambda.cpython-38.pyc
U �Mg�E � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZmZ d dl m Z d dl Z d dl m Z d dlmZ d dlmZ d dlmZ d d lmZmZmZmZmZmZmZ d d l m Z d d l m Z d d l m!Z! e!�r*d d l m"Z" d dl m#Z# d dl m$Z$ d dl m%Z% d dl&m'Z'm(Z(m)Z) e#de$de"f d�Z*dZ+dZ,dd� Z-dd� Z.dd � Z/G d d � d e �Z0d d � Z1d!d"� Z2d#d$� Z3d%d&� Z4d'd(� Z5d)d*� Z6dS )+� N)�deepcopy)�datetime� timedelta�timezone)�environ)�continue_trace)�OP)�should_send_default_pii)� TRANSACTION_SOURCE_COMPONENT)�AnnotatedValue�capture_internal_exceptions�ensure_integration_enabled�event_from_exception�logger� TimeoutThread�reraise)� Integration)�_filter_headers)� TYPE_CHECKING)�Any)�TypeVar)�Callable)�Optional)�EventProcessor�Event�Hint�F.)�boundi� g @�@c s t t� �� fdd��}|S )Nc s� t �� }t� �l t �� �� t�� }|rZt|�rZt||j ddd�d�\}}t j ||d� n t t � | d ��}t � |� W 5 Q R X � | |�S )N� aws_lambdaF��typeZhandled�Zclient_options� mechanism��hint� )� sentry_sdk� get_clientr Zget_isolation_scope�clear_breadcrumbs�sys�exc_info�allr �options� capture_event�_event_from_error_json�json�loads)�args�kwargs�clientr* � sentry_eventr$ �� init_error� �P/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/aws_lambda.py�sentry_init_error- s � z+_wrap_init_error.<locals>.sentry_init_error)r �AwsLambdaIntegration)r6 r9 r7 r5 r8 �_wrap_init_error+ s r; c s t �� �� fdd��}|S )Nc s t �� }|�t�}|d kr,� | |f|�|�S t| t�rTt| �dkrT| d }t| �}n| }d}t|t�sji }|�� }t � � ���} d } t � �� | � � | � t |||�� | �d|j�d�d � |dkr�| �dd� | �d|� |j�r|tk�r|t t } t| |t �} | �� W 5 Q R X |�d i �} t| t��s6i } t| tj|jttj d �} t j | | |d �d ��� z�z0� | |f|�|�W W �lW 5 Q R � W 5 Q R � S t k �r� t �!� }t"||j#d dd�d�\}}t j$||d� t%|� Y nX W 5 | �r�| � � X W 5 Q R X W 5 Q R X d S )Nr% r Z aws_region�:� Z batch_requestT� batch_size�headers)�op�name�source�origin)� aws_event� aws_context)Zcustom_sampling_contextr Fr r! r# )&r& r' �get_integrationr: � isinstance�list�len�dict� get_remaining_time_in_millisZisolation_scoper r( Zadd_event_processor� _make_request_event_processorZset_tag�invoked_function_arn�split�timeout_warning�TIMEOUT_WARNING_BUFFER�MILLIS_TO_SECONDSr �start�getr r Z FUNCTION_AWS� function_namer rC Zstart_transaction�stop� Exceptionr) r* r r, r- r )rD rE r1 r2 r3 � integrationZ request_datar> Zconfigured_time�scopeZtimeout_threadZ waiting_timer? Z transactionr* r4 r$ ��handlerr7 r8 �sentry_handlerJ s� �� � ���� ���0� z%_wrap_handler.<locals>.sentry_handler)� functools�wraps)rZ r[ r7 rY r8 � _wrap_handlerH s gr^ c C s8 t � �( t�� } | �t�}|d k r*| �� W 5 Q R X d S �N)r r& r' rF r: �flush)r3 rW r7 r7 r8 � _drain_queue� s ra c @ s0 e Zd ZdZde� �Zd dd�Zedd� �ZdS ) r: r zauto.function.Fc C s || _ d S r_ )rO )�selfrO r7 r7 r8 �__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 r_ �r^ ��request_handlerr1 r2 ��old_handle_event_requestr7 r8 �sentry_handle_event_request� s zDAwsLambdaIntegration.setup_once.<locals>.sentry_handle_event_requestc s t | �} � | f|�|�S r_ rf rg )�old_handle_http_requestr7 r8 �sentry_handle_http_request� s zCAwsLambdaIntegration.setup_once.<locals>.sentry_handle_http_requestc s t � � | |�S r_ �ra �r1 r2 )� old_to_jsonr7 r8 �sentry_to_json� s z7AwsLambdaIntegration.setup_once.<locals>.sentry_to_jsonc s t |�}� | |f|�|�S r_ rf )Zlambda_runtime_clientrh r1 r2 ri r7 r8 rk s ���c s � fdd�}|S )Nc s t � � | |�S r_ rn ro ��fr7 r8 �inner s zKAwsLambdaIntegration.setup_once.<locals>._wrap_post_function.<locals>.innerr7 )rs rt r7 rr r8 �_wrap_post_function s z<AwsLambdaIntegration.setup_once.<locals>._wrap_post_function) �get_lambda_bootstrapr �warning�hasattrrd re �to_jsonr; ZLambdaRuntimeClientZpost_init_errorZpost_invocation_resultZpost_invocation_error)Zlambda_bootstrapZpre_37rk rm rq ru r7 )rj rl rp r8 � setup_once� sJ � � � ����z AwsLambdaIntegration.setup_onceN)F)�__name__� __module__� __qualname__� identifierrC rc � staticmethodrz r7 r7 r7 r8 r: � s r: c C sb dt jkrt jd S dt jkrZt jd } t| d�rFt| jd�rF| jjS t| d�rV| jS | S d S d S )N� bootstrap�__main__�awslambdaricmain)r) �modulesrx r� r� )�moduler7 r7 r8 rv $ s � rv c s$ t �tj�}|f� ��fdd� }|S )Nc s\ � � � }�| }| �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�rL�� dd�|d< n�� dd ��rLt �� |d< t|�| d< | S )N�extra)rT �function_versionrM �aws_request_idZ execution_duration_in_millisZremaining_time_in_millis�lambda)�url� log_group� log_streamzcloudwatch logs�requestZ httpMethod�methodr� ZqueryStringParametersZ query_stringr? �user�identityZuserArn�idZsourceIp� ip_address�body� �data)rK � setdefaultrT r� rM r� �_get_cloudwatch_logs_url�log_group_name�log_stream_namerS �_get_urlr r r Zremoved_because_raw_datar ) r4 r$ � start_timeZremaining_time_in_milisZ exec_durationr� r� Z user_infor� r� �ip�rE rD �configured_timeoutr7 r8 �event_processorM sN � � z6_make_request_event_processor.<locals>.event_processor)r �nowr �utc)rD rE r� r� r� r7 r� r8 rL I s =rL c C s` | � dd �}| � d�}|d kr"i }|� dd �}|� dd �}|rT|rT|rTd�|||�S d�|j�S )N�pathr? �HostzX-Forwarded-Protoz {}://{}{}zawslambda:///{})rS �formatrT )rD rE r� r? �host�protor7 r7 r8 r� � s r� c C sd d}t �dd�}dj|�d�r"dnd|| j| j|tdd � �|�t� t j �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_REGIONr� 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.comr% )�seconds� )�domain�regionr� r� r� Zend_time) r rS r� � startswithr� r� r �strftimer r� r r� )rE r� � formatstringr� r� r7 r7 r8 r� � s ���r� c C sV g }| D ]H}t �d|�� �}|r|�� \}}}t|�}|�|||d d d d d�� q|S )Nz File "(.+)", line (\d+), in (.+))�filename�function�lineno�varsZ pre_contextZ context_lineZ post_context)�re�match�strip�groups�int�append)Z formatted_tb�frames�framer� � file_name� line_number� func_namer7 r7 r8 �_parse_formatted_traceback� s" �� r� c C s@ dd| � d�| � d�dt| � dg ��iddd �d �gid �}|S ) a9 Converts the error JSON from AWS Lambda into a Sentry error event. This is not a full fletched event, but better than nothing. This is an example of where AWS creates the error JSON: https://github.com/aws/aws-lambda-python-runtime-interface-client/blob/2.2.1/awslambdaric/bootstrap.py#L479 �error�valuesZ errorTypeZ errorMessager� Z stackTracer Fr )r �valueZ stacktracer" )�level� exception)rS r� )Z error_json�eventr7 r7 r8 r. � s" �������r. )7r\ r/ r� r) �copyr r r r �osr r& Zsentry_sdk.apir Zsentry_sdk.constsr Zsentry_sdk.scoper Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r r r r Zsentry_sdk.integrationsr Z$sentry_sdk.integrations._wsgi_commonr �typingr r r r r Zsentry_sdk._typesr r r r rP rQ r; r^ ra r: rv rL r� r� r� r. r7 r7 r7 r8 �<module> sD $ m d%D