Edit File: envelope.cpython-37.pyc
B Z~�_� � @ 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� Zedd� �Zedd� �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 | j �|� d S )N)r �append)r �itemr r r r# A 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 r. G 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 r/ O s z Envelope.get_transaction_eventc C s t | j�S )N)�iterr )r r r r �__iter__W 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 r4 [ s zEnvelope.serialize_intoc C s t �� }| �|� |�� S )N)�io�BytesIOr4 �getvalue)r �outr r r � serialized s zEnvelope.serializec C sB t �|�� �}g }x"t�|�}|d kr(P |�|� qW | ||d�S )N)r r )r �loads�readliner$ �deserialize_fromr, )�clsr5 r r r- r r r r= j s zEnvelope.deserialize_fromc C s | � t�|��S )N)r= r6 r7 )r> �bytesr r r � deserializex 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. r/ r1 r4 r: � classmethodr= r@ rA 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 r? �path)r r? rG 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� )r? rG r �open�readr r )r r5 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 rG r* r? �decode� mimetypesZ guess_type)r rG �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 zPayloadRef.__repr__)NNN)rB rC rD r rL rE rQ rA 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)r? zutf-8�filenamer" � content_type) r r r* r? r% r �encoderQ r! )r r! r r" rS rR 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 rA � 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 rU )r rP r r r r � s zItem.data_categoryc C s | j �� S )N)r! rL )r r r r rL � 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 r. � s zItem.get_eventc C s" | j dkr | jjd k r | jjS d S )Nr' )r" r! r )r r r r r/ � s zItem.get_transaction_eventc C sN t | j�}| �� }t|�|d< |�t|�� |�d� |�|� |�d� d S )N�lengthr2 )r r rL r r3 r )r r5 r r? r r r r4 � s zItem.serialize_intoc C s t �� }| �|� |�� S )N)r6 r7 r4 r8 )r r9 r r r r: s zItem.serializec C sp |� � �� }|sd S t�|�}|d }|�|�}|�d�dkrX| |tt�|�d�d�}n | ||d�}|� � |S )NrY r" )r r' )r )r r! )r< �rstripr r; rK rU r% )r> r5 �liner rY r! �rvr r r r= s zItem.deserialize_fromc C s | � t�|��S )N)r= r6 r7 )r> r? r r r r@ # s zItem.deserialize)NNNN)rB rC rD r rA rE r" r rL r. r/ r4 r: rF r= r@ r r r r r$ � s r$ )r6 r rO Zsentry_sdk._compatr Zsentry_sdk._typesr Zsentry_sdk.sessionsr Zsentry_sdk.utilsr r �typingr r r r r r r r �objectr r% r$ r r r r �<module> s" o,