관리-도구
편집 파일: envelope.cpython-38.pyc
U �Mg�' � @ s� d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm Z e r�d dlm Z d dlm Z d dlm Z d dlm Z d d lmZ d d lmZ d d lmZmZ d d � ZG dd� d�ZG dd� d�ZG dd� d�ZdS )� N)�Session)� json_dumps�capture_internal_exceptions)� TYPE_CHECKING)�Any)�Optional)�Union)�Dict)�List)�Iterator)�Event�EventDataCategoryc C s t | t�r| �dd�} t�| �S )N�utf-8�replace)� isinstance�bytes�decode�json�loads)�data� r �A/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/envelope.py� parse_json s r c @ s� e Zd ZdZd'dd�Zedd� �Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zdd � Zd d � Zd d � Zed!d"� �Zed#d$� �Zd%d&� ZdS )(�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 k rt |�}|pi | _|d kr(g }nt|�}|| _d S �N)�dict�headers�list�items)�selfr r r r r �__init__$ s zEnvelope.__init__c C s$ dt | j�d�dd� | jD ��f S )Nzenvelope with %s items (%s)z, c s s | ] }|j V qd S r )� data_category)�.0�xr r r � <genexpr>8 s z'Envelope.description.<locals>.<genexpr>)�lenr �join�r r r r � description3 s �zEnvelope.descriptionc C s | � tt|d�dd�� d S )N�r �event��payload�type��add_item�Item� PayloadRef)r r* r r r � add_event; s zEnvelope.add_eventc C s | � tt|d�dd�� d S )Nr) � transactionr+ r. )r r3 r r r �add_transactionA s zEnvelope.add_transactionc C s | � tt|d�dd�� d S )Nr) �profiler+ r. )r r5 r r r � add_profileG s zEnvelope.add_profilec C s | � tt|d�dd�� d S )Nr) � profile_chunkr+ r. )r r7 r r r �add_profile_chunkM s �zEnvelope.add_profile_chunkc C s | � tt|d�dd�� d S )Nr) �check_inr+ r. )r Zcheckinr r r � add_checkinU s zEnvelope.add_checkinc C s. t |t�r|�� }| �tt|d�dd�� d S )Nr) �sessionr+ )r r �to_jsonr/ r0 r1 )r r; r r r � add_session[ s zEnvelope.add_sessionc C s | � tt|d�dd�� d S )Nr) �sessionsr+ r. )r r> r r r � add_sessionsc s zEnvelope.add_sessionsc C s | j �|� d S r )r �append)r �itemr r r r/ i s zEnvelope.add_itemc C s( | j D ] }|�� }|d k r| S qd S r )r � get_event)r r r* r r r rB o s zEnvelope.get_eventc C s( | j D ] }|�� }|d k r| S qd S r )r �get_transaction_event)r rA r* r r r rC w s z Envelope.get_transaction_eventc C s t | j�S r )�iterr r' r r r �__iter__ s zEnvelope.__iter__c C s4 |� t| j�� |� d� | jD ]}|�|� q d S )N� )�writer r r �serialize_into)r �frA r r r rH � s zEnvelope.serialize_intoc C s t �� }| �|� |�� S r ��io�BytesIOrH �getvalue�r �outr r r � serialize� s zEnvelope.serializec C s< t |�� �}g }t�|�}|d kr$q0|�|� q| ||d�S )N�r r )r �readliner0 �deserialize_fromr@ )�clsrI r r rA r r r rS � s zEnvelope.deserialize_fromc C s | � t�|��S r �rS rK rL �rT r r r r � deserialize� s zEnvelope.deserializec C s d| j | jf S )Nz <Envelope headers=%r items=%r>rQ r' r r r �__repr__� s zEnvelope.__repr__)NN)�__name__� __module__� __qualname__�__doc__r �propertyr( r2 r4 r6 r8 r: r= r? r/ rB rC rE rH rP � classmethodrS rW rX r r r r r s0 � r c @ s2 e Zd Zd dd�Zdd� Zedd� �Zdd � ZdS ) r1 Nc C s || _ || _|| _d S r )r r �path)r r r_ r r r r r � s zPayloadRef.__init__c C sj | j d kr`| jd k rJt� �( t| jd��}|�� | _ W 5 Q R X W 5 Q R X n| jd k r`t| j�| _ | j phdS )N�rb� )r r_ r �open�readr r )r rI r r r � get_bytes� s zPayloadRef.get_bytesc C sN | j d k rdS | jd k rJ| j}t|t�r4|�dd�}t�|�d }|rJ|S dS )Nzapplication/jsonr r r zapplication/octet-stream)r r_ r r r � mimetypes� guess_type)r r_ �tyr r r �inferred_content_type� s z PayloadRef.inferred_content_typec C s d| j f S )Nz <Payload %r>)rh r' r r r rX � s zPayloadRef.__repr__)NNN)rY rZ r[ r rd r] rh rX r r r r r1 � s � r1 c @ sv e Zd Zddd�Zdd� Zedd� �Zedd � �Zd d � Zd d � Z dd� Z dd� Z dd� Z e dd� �Ze dd� �ZdS )r0 Nc C s� |d k rt |�}n |d kr i }|| _t|t�r:t|d�}n t|t�rVt|�d�d�}n|}|d k rj||d< |d k rz||d< |d k r�||d< nd|kr�|j|d< || _d S )N)r r �filenamer- � content_type) r r r r r1 �str�encoderh r, )r r, r r- rj ri r r r r � s& z Item.__init__c C s d| j | j| jf S )Nz-<Item headers=%r payload=%r data_category=%r>)r r, r! r' r r r rX � s �z Item.__repr__c C s | j �d�S )Nr- �r �getr' r r r r- � s z Item.typec C s� | j �d�}|dks |dkr dS |dkr,dS |dkr8dS |dkrDdS |dkrPd S |d kr\d S |d krhd S |d krtd S |dkr�dS dS d S )Nr- r; r> � attachmentr3 r* �errorZ client_reportZinternalr5 r7 ZstatsdZ metric_bucketr9 Zmonitor�defaultrm )r rg r r r r! s( zItem.data_categoryc C s | j �� S r )r, rd r' r r r rd s zItem.get_bytesc C s" | j dkr | jjdk r | jjS dS )z9 Returns an error event if there is one. r* N�r- r, r r' r r r rB s zItem.get_eventc C s" | j dkr | jjd k r | jjS d S )Nr3 rr r' r r r rC ) s zItem.get_transaction_eventc C sN t | j�}| �� }t|�|d< |�t|�� |�d� |�|� |�d� d S )N�lengthrF )r r rd r% rG r )r rI r r r r r rH / s zItem.serialize_intoc C s t �� }| �|� |�� S r rJ rN r r r rP ; s zItem.serializec C s� |� � �� }|sd S t|�}|�d�}|d k rB|�|�}|� � n|� � �d�}|�d�dkrv| |tt|�d�d�}n | ||d�}|S )Nrs rF r- )r* r3 Zmetric_bucketsr) )r r, )rR �rstripr rn rc r1 )rT rI �liner rs r, �rvr r r rS A s zItem.deserialize_fromc C s | � t�|��S r rU rV r r r rW X s zItem.deserialize)NNNN)rY rZ r[ r rX r] r- r! rd rB rC rH rP r^ rS rW r r r r r0 � s& � r0 )rK r re Zsentry_sdk.sessionr Zsentry_sdk.utilsr r �typingr r r r r r r Zsentry_sdk._typesr r r r r1 r0 r r r r �<module> s$ *