o
    #if&                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm	Z	 erNd 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 dd ZG dd dZG dd dZG dd dZdS )    N)TYPE_CHECKING)Session)
json_dumpscapture_internal_exceptions)Any)Optional)Union)Dict)List)Iterator)EventEventDataCategoryc                 C   s    t | tr| dd} t| S )Nutf-8replace)
isinstancebytesdecodejsonloads)data r   L/var/www/html/corbot_env/lib/python3.10/site-packages/sentry_sdk/envelope.py
parse_json   s   

r   c                   @   s   e Zd ZdZ		d%ddZedd Zdd Zd	d
 Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zedd  Zed!d" Zd#d$ ZdS )&Envelopea.  
    Represents a Sentry Envelope. The calling code is responsible for adhering to the constraints
    documented in the Sentry docs: https://develop.sentry.dev/sdk/envelopes/#data-model. In particular,
    each envelope may have at most one Item with type "event" or "transaction" (but not both).
    Nc                 C   s:   |d urt |}|pi | _|d u rg }nt|}|| _d S N)dictheaderslistitems)selfr   r   r   r   r   __init__#   s   

zEnvelope.__init__c                 C   s$   dt | jddd | jD f S )Nzenvelope with %s items (%s)z, c                 s   s    | ]}|j V  qd S r   )data_category).0xr   r   r   	<genexpr>7   s    z'Envelope.description.<locals>.<genexpr>)lenr   joinr   r   r   r   description2   s   zEnvelope.descriptionc                 C      |  tt|ddd d S )Nr   eventpayloadtypeadd_itemItem
PayloadRef)r   r+   r   r   r   	add_event:      zEnvelope.add_eventc                 C   r)   )Nr*   transactionr,   r/   )r   r5   r   r   r   add_transaction@   r4   zEnvelope.add_transactionc                 C   r)   )Nr*   profiler,   r/   )r   r7   r   r   r   add_profileF   r4   zEnvelope.add_profilec                 C   r)   )Nr*   check_inr,   r/   )r   checkinr   r   r   add_checkinL   r4   zEnvelope.add_checkinc                 C   s.   t |tr	| }| tt|ddd d S )Nr*   sessionr,   )r   r   to_jsonr0   r1   r2   )r   r<   r   r   r   add_sessionR   s   
zEnvelope.add_sessionc                 C   r)   )Nr*   sessionsr,   r/   )r   r?   r   r   r   add_sessionsZ   r4   zEnvelope.add_sessionsc                 C   s   | j | d S r   )r   append)r   itemr   r   r   r0   `   s   zEnvelope.add_itemc                 C   (   | j D ]}| }|d ur|  S qd S r   )r   	get_event)r   r   r+   r   r   r   rD   f      
zEnvelope.get_eventc                 C   rC   r   )r   get_transaction_event)r   rB   r+   r   r   r   rF   n   rE   zEnvelope.get_transaction_eventc                 C   s
   t | jS r   )iterr   r'   r   r   r   __iter__v      
zEnvelope.__iter__c                 C   s4   | t| j | d | jD ]}|| qd S )N   
)writer   r   r   serialize_into)r   frB   r   r   r   rL   z   s
   

zEnvelope.serialize_intoc                 C      t  }| | | S r   ioBytesIOrL   getvaluer   outr   r   r   	serialize      
zEnvelope.serializec                 C   s>   t | }g }	 t|}|d u rn|| q	| ||dS )N   r   r   )r   readliner1   deserialize_fromrA   )clsrM   r   r   rB   r   r   r   rZ      s   

zEnvelope.deserialize_fromc                 C      |  t|S r   rZ   rP   rQ   r[   r   r   r   r   deserialize      zEnvelope.deserializec                 C   s   d| j | jf S )Nz<Envelope headers=%r items=%r>rX   r'   r   r   r   __repr__   s   zEnvelope.__repr__)NN)__name__
__module____qualname____doc__r    propertyr(   r3   r6   r8   r;   r>   r@   r0   rD   rF   rH   rL   rU   classmethodrZ   r_   ra   r   r   r   r   r      s0    

	

r   c                   @   s8   e Zd Z			d
ddZdd Zedd Zdd	 ZdS )r2   Nc                 C   s   || _ || _|| _d S r   )r   r   path)r   r   rh   r   r   r   r   r       s   
zPayloadRef.__init__c              	   C   s   | j d u rW| jd urFt . t| jd}| | _ W d    n1 s$w   Y  W d    | j S W d    | j S 1 s>w   Y  | j S | jd urTt| j| _ | j S d| _ | j S )Nrb    )r   rh   r   openreadr   r   )r   rM   r   r   r   	get_bytes   s$   




zPayloadRef.get_bytesc                 C   sN   | j d urdS | jd ur%| j}t|tr|dd}t|d }|r%|S dS )Nzapplication/jsonr   r   r   zapplication/octet-stream)r   rh   r   r   r   	mimetypes
guess_type)r   rh   tyr   r   r   inferred_content_type   s   


z PayloadRef.inferred_content_typec                 C   s   d| j f S )Nz<Payload %r>)rq   r'   r   r   r   ra      s   zPayloadRef.__repr__)NNN)rb   rc   rd   r    rm   rf   rq   ra   r   r   r   r   r2      s    

r2   c                   @   s~   e Zd Z				dddZdd Zedd Zedd	 Zd
d Zdd Z	dd Z
dd Zdd Zedd Zedd ZdS )r1   Nc                 C   s   |d ur	t |}n|d u ri }|| _t|trt|d}nt|tr+t|dd}n|}|d ur5||d< |d ur=||d< |d urF||d< n	d|vrO|j|d< || _d S )N)r   r   filenamer.   content_type)	r   r   r   r   r2   strencoderq   r-   )r   r-   r   r.   rs   rr   r   r   r   r       s&   





zItem.__init__c                 C   s   d| j | j| jf S )Nz-<Item headers=%r payload=%r data_category=%r>)r   r-   r!   r'   r   r   r   ra      s
   zItem.__repr__c                 C   s   | j dS )Nr.   r   getr'   r   r   r   r.      s   z	Item.typec                 C   sp   | j d}|dkrdS |dkrdS |dkrdS |dkrdS |dkr$dS |d	kr*d	S |d
kr0dS |dkr6dS dS )Nr.   r<   
attachmentr5   r+   errorclient_reportinternalr7   statsdmetric_bucketr9   monitordefaultrv   )r   rp   r   r   r   r!      s$   zItem.data_categoryc                 C   s
   | j  S r   )r-   rm   r'   r   r   r   rm     rI   zItem.get_bytesc                 C   s"   | j dkr| jjdur| jjS dS )z9
        Returns an error event if there is one.
        r+   Nr.   r-   r   r'   r   r   r   rD     s   zItem.get_eventc                 C   s"   | j dkr| jjd ur| jjS d S )Nr5   r   r'   r   r   r   rF      s   zItem.get_transaction_eventc                 C   sN   t | j}|  }t||d< |t| |d || |d d S )NlengthrJ   )r   r   rm   r%   rK   r   )r   rM   r   r   r   r   r   rL   &  s   


zItem.serialize_intoc                 C   rN   r   rO   rS   r   r   r   rU   2  rV   zItem.serializec                 C   s   |   }|s
d S t|}|d}|d ur!||}|   n|  d}|ddv r<| |tt|dd}|S | ||d}|S )Nr   rJ   r.   )r+   r5   metric_bucketsr*   )r   r-   )rY   rstripr   rw   rl   r2   )r[   rM   liner   r   r-   rvr   r   r   rZ   8  s   


zItem.deserialize_fromc                 C   r\   r   r]   r^   r   r   r   r_   O  r`   zItem.deserialize)NNNN)rb   rc   rd   r    ra   rf   r.   r!   rm   rD   rF   rL   rU   rg   rZ   r_   r   r   r   r   r1      s(    


	
r1   )rP   r   rn   sentry_sdk._typesr   sentry_sdk.sessionr   sentry_sdk.utilsr   r   typingr   r   r   r	   r
   r   r   r   r   r   r2   r1   r   r   r   r   <module>   s&     ,