o
    hx(                     @   s   d dl Z d dlZd dlZd dlmZ d dlmZmZ d dl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)Session)
json_dumpscapture_internal_exceptions)TYPE_CHECKING)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   U/var/www/html/pro-man-master/venv/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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>8   s    z'Envelope.description.<locals>.<genexpr>)lenr   joinr   r   r   r   description3   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_transactionA   r4   zEnvelope.add_transactionc                 C   r)   )Nr*   profiler,   r/   )r   r7   r   r   r   add_profileG   r4   zEnvelope.add_profilec              
   C   s*   |  tt|ddd|ddid d S )Nr*   profile_chunkplatformpython)r-   r.   r   )r0   r1   r2   get)r   r9   r   r   r   add_profile_chunkM   s   zEnvelope.add_profile_chunkc                 C   r)   )Nr*   check_inr,   r/   )r   checkinr   r   r   add_checkinY   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   rA   r   r   r   add_session_   s   
zEnvelope.add_sessionc                 C   r)   )Nr*   sessionsr,   r/   )r   rD   r   r   r   add_sessionsg   r4   zEnvelope.add_sessionsc                 C   s   | j | d S r   )r   append)r   itemr   r   r   r0   m   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   rI   s      
zEnvelope.get_eventc                 C   rH   r   )r   get_transaction_event)r   rG   r+   r   r   r   rK   {   rJ   zEnvelope.get_transaction_eventc                 C   s
   t | jS r   )iterr   r'   r   r   r   __iter__      
zEnvelope.__iter__c                 C   s4   | t| j | d | jD ]}|| qd S )N   
)writer   r   r   serialize_into)r   frG   r   r   r   rQ      s
   

zEnvelope.serialize_intoc                 C      t  }| | | S r   ioBytesIOrQ   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_fromrF   )clsrR   r   r   rG   r   r   r   r_      s   

zEnvelope.deserialize_fromc                 C      |  t|S r   r_   rU   rV   r`   r   r   r   r   deserialize      zEnvelope.deserializec                 C   s   d| j | jf S )Nz<Envelope headers=%r items=%r>r]   r'   r   r   r   __repr__   s   zEnvelope.__repr__)NN)__name__
__module____qualname____doc__r    propertyr(   r3   r6   r8   r=   r@   rC   rE   r0   rI   rK   rM   rQ   rZ   classmethodr_   rd   rf   r   r   r   r   r      s2    

	

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   rm   r   r   r   r   r       s   
zPayloadRef.__init__c              	   C   s   | j d u rD| jd ur9t # t| jd}| | _ W d    n1 s$w   Y  W d    n1 s3w   Y  n| jd urDt| j| _ | j pHdS )Nrb    )r   rm   r   openreadr   r   )r   rR   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   rm   r   r   r   	mimetypes
guess_type)r   rm   tyr   r   r   inferred_content_type   s   


z PayloadRef.inferred_content_typec                 C   s   d| j f S )Nz<Payload %r>)rv   r'   r   r   r   rf      s   zPayloadRef.__repr__)NNN)rg   rh   ri   r    rr   rk   rv   rf   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encoderv   r-   )r   r-   r   r.   rx   rw   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   rf      s
   zItem.__repr__c                 C   s   | j dS )Nr.   r   r<   r'   r   r   r   r.     s   z	Item.typec                 C   s   | j d}|dk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r.dS |dkr4dS |dkr:dS |dkr@dS |dkrFdS dS )Nr.   rA   rD   
attachmentr5   r+   errorloglog_itemclient_reportinternalr7   r9   statsdmetric_bucketr>   monitordefaultr{   )r   ru   r   r   r   r!     s,   zItem.data_categoryc                 C   s
   | j  S r   )r-   rr   r'   r   r   r   rr   "  rN   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   rI   &  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   rK   /  s   zItem.get_transaction_eventc                 C   sN   t | j}|  }t||d< |t| |d || |d d S )NlengthrO   )r   r   rr   r%   rP   r   )r   rR   r   r   r   r   r   rQ   5  s   


zItem.serialize_intoc                 C   rS   r   rT   rX   r   r   r   rZ   A  r[   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   rO   r.   )r+   r5   metric_bucketsr*   )r   r-   )r^   rstripr   r<   rq   r2   )r`   rR   liner   r   r-   rvr   r   r   r_   G  s   


zItem.deserialize_fromc                 C   ra   r   rb   rc   r   r   r   rd   ^  re   zItem.deserialize)NNNN)rg   rh   ri   r    rf   rk   r.   r!   rr   rI   rK   rQ   rZ   rl   r_   rd   r   r   r   r   r1      s(    


	
r1   )rU   r   rs   sentry_sdk.sessionr   sentry_sdk.utilsr   r   typingr   r   r   r   r	   r
   r   sentry_sdk._typesr   r   r   r   r2   r1   r   r   r   r   <module>   s&     *