File: //opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/__pycache__/envelope.cpython-37.pyc
B
T ab! � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dl m
Z
m Z er�d dl m
Z
d dl mZ d dl mZ d d l mZ d d
l mZ d d l mZ d d lmZmZ G d
d� de�ZG dd� de�ZG dd� de�ZdS )� N)� text_type)�MYPY)�Session)�
json_dumps�capture_internal_exceptions)�Any)�Optional)�Union)�Dict)�List)�Iterator)�Event�EventDataCategoryc @ s� e Zd Zd dd�Zedd� �Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zedd� �Zed d � �Zd d � ZdS )!�EnvelopeNc C s: |d k rt |�}|pi | _|d kr(g }nt|�}|| _d S )N)�dict�headers�list�items)�selfr r � r �D/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/envelope.py�__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 )N)�
data_category)�.0�xr r r � <genexpr>* s z'Envelope.description.<locals>.<genexpr>)�lenr �join)r r r r � description% s zEnvelope.descriptionc C s | � tt|d�dd�� d S )N)�json�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 )N)r � transaction)r! r" )r# r$ r% )r r' r r r �add_transaction3 s zEnvelope.add_transactionc C s. t |t�r|�� }| �tt|d�dd�� d S )N)r �session)r! r" )�
isinstancer Zto_jsonr# r$ r% )r r) r r r � add_session9 s
zEnvelope.add_sessionc C s | � tt|d�dd�� d S )N)r �sessions)r! r" )r# r$ r% )r r, r r r � add_sessionsA s zEnvelope.add_sessionsc C s | j �|� d S )N)r �append)r �itemr r r r# G s zEnvelope.add_itemc C s( x"| j D ]}|�� }|d k r|S qW d S )N)r � get_event)r r r r r r r0 M s
zEnvelope.get_eventc C s( x"| j D ]}|�� }|d k r|S qW d S )N)r �get_transaction_event)r r/ r r r r r1 U s
z Envelope.get_transaction_eventc C s
t | j�S )N)�iterr )r r r r �__iter__] s zEnvelope.__iter__c C s8 |� t| j�� |� d� x| jD ]}|�|� q"W d S )N�
)�writer r r �serialize_into)r �fr/ r r r r6 a s
zEnvelope.serialize_intoc C s t �� }| �|� |�� S )N)�io�BytesIOr6 �getvalue)r �outr r r � serializej s
zEnvelope.serializec C sB t �|�� �}g }x"t�|�}|d kr(P |�|� qW | ||d�S )N)r r )r �loads�readliner$ �deserialize_fromr. )�clsr7 r r r/ r r r r? p s
zEnvelope.deserialize_fromc C s | � t�|��S )N)r? r8 r9 )r@ �bytesr r r � deserialize~ s zEnvelope.deserializec C s d| j | jf S )Nz <Envelope headers=%r items=%r>)r r )r r r r �__repr__� s zEnvelope.__repr__)NN)�__name__�
__module__� __qualname__r �propertyr r& r( r+ r- r# r0 r1 r3 r6 r< � classmethodr? rB rC r r r r r s
r c @ s2 e Zd Zd
dd�Zdd� Zedd� �Zdd � ZdS ) r% Nc C s || _ || _|| _d S )N)r rA �path)r rA rI r r r r r � s zPayloadRef.__init__c
C sn | j d krh| jd k rJt� �( t| jd��}|�� | _ W d Q R X W d Q R X n | jd k rbt| j�| _ nd| _ | j S )N�rb� )rA rI r �open�readr r )r r7 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/jsonzutf-8�replacer zapplication/octet-stream)r rI r* rA �decode� mimetypesZ
guess_type)r rI �tyr r r �inferred_content_type� s
z PayloadRef.inferred_content_typec C s d| j f S )Nz <Payload %r>)rS )r r r r rC � s zPayloadRef.__repr__)NNN)rD rE rF r rN rG rS rC r r r r r% � s
r% c @ sv e Zd Zddd�Zdd� Zedd� �Zedd � �Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z e
dd� �Ze
dd� �ZdS )r$ 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)rA zutf-8�filenamer" � content_type) r r r* rA r% r �encoderS r! )r r! r r" rU rT 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 rC � s z
Item.__repr__c C s | j �d�S )Nr" )r �get)r r r r r" � s z Item.typec C sD | j �d�}|dkrdS |dkr$dS |dkr0dS |dkr<dS dS d S )Nr" r) �
attachmentr' r �error�default)r rW )r rR r r r r � s zItem.data_categoryc C s
| j �� S )N)r! rN )r r r r rN � 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 r0 � s zItem.get_eventc C s" | j dkr | jjd k r | jjS d S )Nr' )r" r! r )r r r r r1 � s zItem.get_transaction_eventc C sN t | j�}| �� }t|�|d< |�t|�� |�d� |�|� |�d� d S )N�lengthr4 )r r rN r r5 r )r r7 r rA r r r r6 s
zItem.serialize_intoc C s t �� }| �|� |�� S )N)r8 r9 r6 r: )r r; r r r r< s
zItem.serializec C sp |� � �� }|sd S t�|�}|d }|�|�}|�d�dkrX| |tt�|�d�d�}n | ||d�}|� � |S )Nr[ r" )r r' )r )r r! )r> �rstripr r= rM rW r% )r@ r7 �liner r[ r! �rvr r r r? s
zItem.deserialize_fromc C s | � t�|��S )N)r? r8 r9 )r@ rA r r r rB ) s zItem.deserialize)NNNN)rD rE rF r rC rG r" r rN r0 r1 r6 r< rH r? rB r r r r r$ � s
r$ )r8 r rQ Zsentry_sdk._compatr Zsentry_sdk._typesr Zsentry_sdk.sessionr Zsentry_sdk.utilsr r �typingr r r r
r r r
r �objectr r% r$ r r r r �<module> s" u,