File: //opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/__pycache__/sessions.cpython-37.pyc
B
T a� � @ s� d dl Z d dlZd dlmZmZ d dlmZ d dlZd dlm Z d dl
m Z d dl m
Z
d dlmZ e
r�d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ d d
lmZ ddd�Zeddd��ZdZdZdd� ZG dd� de �Z dS )� N)�Thread�Lock)�contextmanager)�Envelope)�Session)�MYPY)�format_timestamp)�Callable)�Optional)�Any)�Dict)�List)� Generatorc C s> | dkrt jj} | jj}|dkr:| jr.| jjni }|d }|S )z<Utility function to find out if session tracking is enabled.N�auto_session_tracking)�
sentry_sdk�Hub�currentZscopeZ _force_auto_session_trackingZclient�options)�hub� should_trackZclient_options� r �D/opt/alt/python37/lib/python3.7/site-packages/sentry_sdk/sessions.py� is_auto_session_tracking_enabled s r � applicationc c sF | dkrt jj} t| �}|r(| j|d� z
dV W d|r@| �� X dS )z8Starts and stops a session automatically around a block.N)� session_mode)r r r r Z
start_sessionZ end_session)r r r r r r r ! s
r )�exited�abnormal�crashed�d c C s t |�t| �� �d�S )N)�attrsZ
aggregates)�dict�list�values)Zaggregate_statesr r r r �make_aggregate_envelope5 s r# c @ sF e Zd Zddd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dS )�SessionFlusher�< c C s> || _ || _g | _i | _d | _t� | _t� | _d | _d| _ d S )NT)
� capture_func�flush_interval�pending_sessions�pending_aggregates�_threadr � _thread_lock�_aggregate_lock�_thread_for_pid�_running)�selfr& r' r r r �__init__; s zSessionFlusher.__init__c C s� | j }g | _ | j� | j}i | _W d Q R X t� }x4|D ],}t|j�tkrX| �|� t� }|�|� q6W xB|�� D ]6\}}t|j�tkr�| �|� t� }|� t
||�� qpW t|j�dkr�| �|� d S )Nr ) r( r, r) r �len�items�MAX_ENVELOPE_ITEMSr& � add_sessionZ add_sessionsr# )r/ r( r) Zenvelope�sessionr �statesr r r �flushK s$
zSessionFlusher.flushc s� � j t�� kr � jd k r d S � j�V � j t�� kr@� jd k r@d S � fdd�}t|d�}d|_|�� |� _t�� � _ W d Q R X d S )Nc s* x$� j r$t�� j� � j r� �� qW d S )N)r. �time�sleepr' r7 r )r/ r r r* n s z/SessionFlusher._ensure_running.<locals>._thread)�targetT)r- �os�getpidr* r+ r �daemon�start)r/ r* �threadr )r/ r �_ensure_runningf s
z SessionFlusher._ensure_runningc C s� | j �� |jdd�}tt|�� ��}|j}| j�|i �}|�|i �}d|krZt|j�|d< |j dkrz|�
dd�d |d< nT|j dkr�|�
dd�d |d< n4|j dkr�|�
dd�d |d< n|�
d d�d |d < W d Q R X d S )
NF)Zwith_user_info�startedr r � r Zerroredr ) r, Zget_json_attrs�tuple�sortedr2 Ztruncated_startedr) �
setdefaultr �status�get�errors)r/ r5 r Z primary_keyZ
secondary_keyr6 �stater r r �add_aggregate_session| s
z$SessionFlusher.add_aggregate_sessionc C s2 |j dkr| �|� n| j�|�� � | �� d S )NZrequest)r rJ r( �appendZto_jsonr@ )r/ r5 r r r r4 � s
zSessionFlusher.add_sessionc C s
d| _ d S )NF)r. )r/ r r r �kill� s zSessionFlusher.killc C s | � � d S )N)rL )r/ r r r �__del__� s zSessionFlusher.__del__N)r% )
�__name__�
__module__� __qualname__r0 r7 r@ rJ r4 rL rM r r r r r$ : s
r$ )N)Nr ) r; r8 � threadingr r �
contextlibr r Zsentry_sdk.enveloper Zsentry_sdk.sessionr Zsentry_sdk._typesr Zsentry_sdk.utilsr �typingr r
r r r
r r r ZTERMINAL_SESSION_STATESr3 r# �objectr$ r r r r �<module> s,