o
    hR                      @   sH  d dl Z d dlZd dlmZ d dlmZmZmZ d dlmZ d dl	Z	d dl
mZ d dlmZ d dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZmZmZmZmZmZ d dlmZ dZdZ erd dlm!Z! d dlm"Z" d dlm#Z# d dlm$Z$ d dl%m&Z&m'Z'm(Z( e"de#de!f dZ)dd Z*G dd deZ+dd Z,dd Z-dS )    N)deepcopy)datetime	timedeltatimezone)environ)continue_trace)OP)Integration)_filter_headers)should_send_default_pii)TransactionSource)AnnotatedValuecapture_internal_exceptionsevent_from_exceptionloggerTimeoutThreadreraise)TYPE_CHECKINGg      ?g     @@)Any)TypeVar)Callable)Optional)EventProcessorEventHintF.)boundc                    s   t   fdd}|S )Nc                    s  t  }|t}|d u r | |g|R i |S td}|s1td  | |g|R i |S t|}t	
tj}t  }t 4 |  |t||| |dtd d }	|jrq|tkrq|t }
t|
|}	|	  W d    n1 s{w   Y  i }t|dr|j}t|tjtddtjtjd}tdtd	td
tdtdd|d}t j||do zZz& | |g|R i |W W |	r|	   |!  W  d    W  d    S  t"y	   t#$ }t%||j&dddd\}}t j'||d t(|  Y nw W |	r|	   |!  n|	r|	   |!  w W d    n1 s.w   Y  W d    d S W d    d S 1 sGw   Y  d S )NFUNCTION_TIMEOUT_SECzOThe configured timeout could not be fetched from Cloud Functions configuration.
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)gcp_env	gcp_event)custom_sampling_contextgcpF)typehandled)client_options	mechanism)hint))
sentry_sdk
get_clientget_integrationGcpIntegrationr   getr   debugintr   nowr   utcisolation_scoper   clear_breadcrumbsadd_event_processor_make_request_event_processorset_tagtimeout_warningTIMEOUT_WARNING_BUFFERr   starthasattrr    r   r   FUNCTION_GCPr   	COMPONENTr&   start_transactionstopflush	Exceptionsysexc_infor   optionscapture_eventr   )functionhandlerr0   argskwargsclientintegrationconfigured_timeinitial_timescopetimeout_threadwaiting_timer    transactionsampling_contextrQ   sentry_eventr7   func ]/var/www/html/pro-man-master/venv/lib/python3.10/site-packages/sentry_sdk/integrations/gcp.pysentry_func*   s   




	
/

 *$z_wrap_func.<locals>.sentry_func)	functoolswraps)rb   re   rc   ra   rd   
_wrap_func(   s   Qrh   c                   @   s0   e Zd ZdZde Zd	ddZedd ZdS )
r;   r2   zauto.function.Fc                 C   s
   || _ d S )N)rF   )selfrF   rc   rc   rd   __init__   s   
zGcpIntegration.__init__c                  C   s:   dd l } t| dstd d S | j}t|jj|j_d S )Nr   	worker_v1zFGcpIntegration currently supports only Python 3.7 runtime environment.)__main__rI   r   warningrk   rh   FunctionHandlerinvoke_user_function)gcp_functionsworker1rc   rc   rd   
setup_once   s   
zGcpIntegration.setup_onceN)F)__name__
__module____qualname__
identifierr&   rj   staticmethodrr   rc   rc   rc   rd   r;      s    

r;   c                    s    fdd}|S )Nc                    s  t tj}| }|tdd }| di }tdtdtdtdtd| d	|d
< dt|i|d< | di }d	td|d< t
drTj|d< t
drajd|d< t
drmtj|d< t r{t
drzj|d< nt
drt |d< t|| d< | S )N   )millisecondsextrar!   r'   r(   r   r)   )r*   r+   r,   r-   r.   execution_duration_in_millisconfigured_timeout_in_secondszgoogle cloud functionsurlzgoogle cloud logsrequestz	gcp:///{}methodquery_stringzutf-8r    data)r   r?   r   r@   r   
setdefaultr   r<   _get_google_cloud_logs_urlformatrI   r   r   decoder
   r    r   r   r   removed_because_raw_datar   )eventr7   
final_time	time_diffr{   rz   r~   configured_timeoutr0   rZ   rc   rd   event_processor   s<   







z6_make_request_event_processor.<locals>.event_processorrc   )r0   r   rZ   r   rc   r   rd   rD      s   1rD   c                 C   sH   | t dd }d}djt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.
    rx   )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&timestamp={timestamp_end}&customFacets=&limitCustomFacetWidth=true&dateRangeStart={timestamp_start}&dateRangeEnd={timestamp_end}&interval=PT1H&scrollTimestamp={timestamp_end}r)   r!   r   )projectr*   regiontimestamp_endtimestamp_start)r   r   r   r<   strftime)r   hour_agoformatstringr}   rc   rc   rd   r      s   	r   ).rf   rP   copyr   r   r   r   osr   r8   sentry_sdk.apir   sentry_sdk.constsr   sentry_sdk.integrationsr	   $sentry_sdk.integrations._wsgi_commonr
   sentry_sdk.scoper   sentry_sdk.tracingr   sentry_sdk.utilsr   r   r   r   r   r   typingr   rG   MILLIS_TO_SECONDSr   r   r   r   sentry_sdk._typesr   r   r   r   rh   r;   rD   r   rc   rc   rc   rd   <module>   s6     	W7