관리-도구
편집 파일: gcp.cpython-38.pyc
U �Mg^ � @ sJ d dl Z d dlZd dlmZ d dlmZmZmZ d dlmZ d dl Z d dl m Z d dl m Z d dlmZ d dlmZ d d lmZ d d lmZ d d lmZmZmZmZmZm Z d d l m Z d Z dZ e �r d dl 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)dd� Z*G dd� de�Z+dd � Z,d d � Z-dS ) � N)�deepcopy)�datetime� timedelta�timezone)�environ)�continue_trace)�OP)� Integration)�_filter_headers)�should_send_default_pii)� TRANSACTION_SOURCE_COMPONENT)�AnnotatedValue�capture_internal_exceptions�event_from_exception�logger� TimeoutThread�reraise)� TYPE_CHECKINGg �?g @�@)�Any)�TypeVar)�Callable)�Optional)�EventProcessor�Event�Hint�F.)�boundc s t �� �� fdd��}|S )Nc s t �� }|�t�}|d kr,� | |f|�|�S t�d�}|sVt�d� � | |f|�|�S t|�}t � t j �}t � � ���}t� �^ |�� |�t|||�� |�dt�d�� d } |jr�|tkr�|t } t| |�} | �� W 5 Q R X i } t|d�r�|j} t| tjt�dd�t tj d�} t�d�t�d �t�d �t�d�t�d �d �|d �} t j | | d��� z�z0� | |f|�|�W W �lW 5 Q R � W 5 Q R � S t!k �r� t"�#� }t$||j%ddd�d�\}}t j&||d� t'|� Y nX W 5 | �r�| � � |� � X W 5 Q R X W 5 Q R X d S )NZFUNCTION_TIMEOUT_SECzOThe configured timeout could not be fetched from Cloud Functions configuration.Z gcp_region�FUNCTION_REGION�headers� FUNCTION_NAME� )�op�name�source�origin� ENTRY_POINT�FUNCTION_IDENTITY� GCP_PROJECT)� function_name�function_entry_point�function_identity�function_region�function_project)Zgcp_env� gcp_event)Zcustom_sampling_context�gcpF)�typeZhandled)Zclient_optionsZ mechanism)�hint)(� sentry_sdkZ get_clientZget_integration�GcpIntegrationr �getr �debug�intr �nowr �utcZisolation_scoper Zclear_breadcrumbsZadd_event_processor� _make_request_event_processorZset_tag�timeout_warning�TIMEOUT_WARNING_BUFFERr �start�hasattrr r r Z FUNCTION_GCPr r$ Zstart_transaction�stop�flush� Exception�sys�exc_infor �optionsZ capture_eventr )Zfunctionhandlerr- �args�kwargs�clientZ integrationZconfigured_time� initial_time�scopeZtimeout_threadZ waiting_timer Z transactionZsampling_contextrA Z sentry_eventr0 ��func� �I/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/integrations/gcp.py� sentry_func* s� � ���� � �� �0� z _wrap_func.<locals>.sentry_func)� functools�wraps)rI rL rJ rH rK � _wrap_func( s QrO c @ s0 e Zd ZdZde� �Zd dd�Zedd� �ZdS ) r2 r. zauto.function.Fc C s || _ d S )N)r9 )�selfr9 rJ rJ rK �__init__� s zGcpIntegration.__init__c C s: dd l } t| d�s t�d� d S | j}t|jj�|j_d S )Nr � worker_v1zFGcpIntegration currently supports only Python 3.7 runtime environment.)�__main__r< r �warningrR rO ZFunctionHandlerZinvoke_user_function)Z gcp_functionsZworker1rJ rJ rK � setup_once� s ��zGcpIntegration.setup_onceN)F)�__name__� __module__� __qualname__� identifierr$ rQ � staticmethodrU rJ rJ rJ rK r2 s r2 c s � ��fdd�}|S )Nc s t �tj�}|� }|tdd� }| �di �}t�d�t�d�t�d�t�d�t�d�|� d �|d < d t|�i|d < | �d i �}d� t�d��|d < t �d�r��j |d< t �d�rj � d�|d< t �d�r�t�j�|d< t� r�t �d�r�j|d< nt �d��rt�� |d< t|�| d < | S )N� )Z milliseconds�extrar r% r& r r' )r( r) r* r+ r, � execution_duration_in_millisZ configured_timeout_in_secondszgoogle cloud functions�urlzgoogle cloud logs�requestz gcp:///{}�method� query_stringzutf-8r �data)r r6 r r7 r � setdefaultr r3 �_get_google_cloud_logs_url�formatr< r` ra �decoder r r rb r Zremoved_because_raw_datar )�eventr0 � final_timeZ time_diffr] r\ r_ ��configured_timeoutr- rF rJ rK �event_processor� s< � � z6_make_request_event_processor.<locals>.event_processorrJ )r- rj rF rk rJ ri rK r8 � s 1r8 c C sH | t dd� }d}djt�d�t�d�t�d�| �|�|�|�d�}|S ) z� Generates a Google Cloud Logs console URL based on the environment variables Arguments: final_time {datetime} -- Final time Returns: str -- Google Cloud Logs Console URL to logs. r[ )�hoursz%Y-%m-%dT%H:%M:%SZaY https://console.cloud.google.com/logs/viewer?project={project}&resource=cloud_function%2Ffunction_name%2F{function_name}%2Fregion%2F{region}&minLogLevel=0&expandAll=false×tamp={timestamp_end}&customFacets=&limitCustomFacetWidth=true&dateRangeStart={timestamp_start}&dateRangeEnd={timestamp_end}&interval=PT1H&scrollTimestamp={timestamp_end}r' r r )�projectr( �regionZ timestamp_endZtimestamp_start)r re r r3 �strftime)rh Zhour_ago� formatstringr^ rJ rJ rK rd � s ��rd ).rM r@ �copyr r r r �osr r1 Zsentry_sdk.apir Zsentry_sdk.constsr Zsentry_sdk.integrationsr Z$sentry_sdk.integrations._wsgi_commonr Zsentry_sdk.scoper Zsentry_sdk.tracingr Zsentry_sdk.utilsr r r r r r �typingr r: ZMILLIS_TO_SECONDSr r r r Zsentry_sdk._typesr r r r rO r2 r8 rd rJ rJ rJ rK �<module> s4 W7