Edit File: gcp.cpython-37.pyc
B Z~�_^ � @ 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 d Zd Zer�d d lmZ d dlm Z d dlm Z d dlm Z d dlm Z m Z m!Z! e de def d�Z"dd� Z#G dd� de�Z$dd� Z%dd � Z&dS ) � )�datetime� timedelta)�environN)�Hub�_should_send_default_pii)� Transaction)�reraise)�AnnotatedValue�capture_internal_exceptions�event_from_exception�logger� TimeoutThread)� Integration)�_filter_headers)�MYPYg �?g @�@)�Any)�TypeVar)�Callable)�Optional)�EventProcessor�Event�Hint�F.)Zboundc s � fdd�}|S )Nc s� t j}|�t�}|d kr*� | |f|�|�S |j}t�d�}|sZt�d� � | |f|�|�S t |�}t � � }|� � ��j} t � �^ | �� | �t|||�� | �dt�d�� d } |jr�|tkr�|t } t| |�} | �� W d Q R X i } t|d�r�|j} tj| dt�dd�d �} t�d�t�d �t�d �t�d�t�d �d �|d�}|j| |d��� zfy� | |f|�|�S tk �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 )NZFUNCTION_TIMEOUT_SECzOThe configured timeout could not be fetched from Cloud Functions configuration.Z gcp_region�FUNCTION_REGION�headerszserverless.function� FUNCTION_NAME� )�op�name� 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)$r �currentZget_integration�GcpIntegration�clientr �getr �debug�intr �utcnowZ push_scoper Zclear_breadcrumbsZadd_event_processor� _make_request_event_processorZset_tag�timeout_warning�TIMEOUT_WARNING_BUFFERr �start�hasattrr r Zcontinue_from_headersZstart_transaction� Exception�sys�exc_infor �optionsZ capture_eventr �stop�flush)Zfunctionhandlerr' �args�kwargsZhubZ integrationr- Zconfigured_time� initial_timeZscopeZtimeout_threadZ waiting_timer Z transactionZsampling_contextr9 Z sentry_eventr* )�func� �L/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/integrations/gcp.py� sentry_func% sf z _wrap_func.<locals>.sentry_funcrA )r@ rC rA )r@ rB � _wrap_func# s OrD c @ s& e Zd ZdZddd�Zedd� �ZdS ) r, r( Fc C s || _ d S )N)r3 )�selfr3 rA rA rB �__init__z 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__r6 r ZwarningrG rD ZFunctionHandlerZinvoke_user_function)Z gcp_functionsZworker1rA rA rB � setup_once~ s zGcpIntegration.setup_onceN)F)�__name__� __module__� __qualname__� identifierrF � staticmethodrI rA rA rA rB r, w s r, c s � ��fdd�}|S )Nc s" t �� }|� }|jt }| �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�r��j � d�|d< t �d�r�t �j�|d< t� r�t �d�r�j|d< n(t �d��rtddddddggi�|d< || d < | S )N�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 �datar �remz!raw�xr )r r1 � microseconds�MILLIS_TO_SECONDS� setdefaultr r. �_get_google_cloud_logs_url�formatr6 rS rT �decoder r r rU r )�eventr* � final_timeZ time_diffrP rO rR )�configured_timeoutr' r? rA rB �event_processor� s6 z6_make_request_event_processor.<locals>.event_processorrA )r' r` r? ra rA )r` r' r? rB r2 � s 1r2 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. � )�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" ZregionZ timestamp_endZtimestamp_start)r r\ r r. �strftime)r_ Zhour_ago� formatstringrQ rA rA rB r[ � s r[ )'r r �osr r8 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 r4 rY �typingr r r r r r r r rD r, r2 r[ rA rA rA rB �<module> s, T7