관리-도구
편집 파일: sessions.cpython-38.pyc
U �Mg�# � @ s* d dl Z d dlZd dlZd dlmZmZ d dlmZ d dlZd dl m Z d dl m Z d dl mZ d dlmZ er�d dlmZ d d lmZ d d lmZ d d lmZ d d lmZ d d lmZ d dlmZ d"dd�Zed#dd��Zdd� Zdd� Zed$dd��Z ed%dd��Z d Z d Z d d � Z G d d!� d!�Z!dS )&� N)�Thread�Lock)�contextmanager)�Envelope)�Session)�format_timestamp)� TYPE_CHECKING)�Any)�Callable)�Dict)� Generator)�List)�Optional)�Unionc C sR t jdtdd� | dkr tjj} | jj}|dkrN| jr>| jj ni }|� dd�}|S )zHDEPRECATED: Utility function to find out if session tracking is enabled.�nThis function is deprecated and will be removed in the next major release. There is no public API replacement.� �� stacklevelN�auto_session_trackingF) �warnings�warn�DeprecationWarning� sentry_sdk�Hub�current�scope� _force_auto_session_tracking�client�options�get)�hub� should_track�client_options� r# �A/opt/hc_python/lib/python3.8/site-packages/sentry_sdk/sessions.py� is_auto_session_tracking_enabled s � r% � applicationc c sv t jdtdd� | dkr tjj} t �� � t �dt� t| �}W 5 Q R X |rX| j |d� z dV W 5 |rp| � � X dS )zgDEPRECATED: Use track_session instead Starts and stops a session automatically around a block. zeThis function is deprecated and will be removed in the next major release. Use track_session instead.r r N�ignore�� session_mode) r r r r r r �catch_warnings� simplefilterr% � start_session� end_session)r r) r! r# r# r$ r 0 s � r c C s t jdtdd� t| �S )zR DEPRECATED: Utility function to find out if session tracking is enabled. r r r )r r r �!_is_auto_session_tracking_enabled)r r# r# r$ �&is_auto_session_tracking_enabled_scopeK s �r/ c C s( | j }|dkr$t�� j}|�dd�}|S )zF Utility function to find out if session tracking is enabled. Nr F)r r Z get_clientr r )r r! r" r# r# r$ r. \ s r. c c s2 t jdtdd� t| |d�� dV W 5 Q R X dS )z�DEPRECATED: This function is a deprecated alias for track_session. Starts and stops a session automatically around a block. zdThis function is a deprecated alias for track_session and will be removed in the next major release.r r r( N)r r r � track_session)r r) r# r# r$ �auto_session_tracking_scopej s �r1 c c s6 t | �}|r| j|d� z dV W 5 |r0| �� X dS )z� Start a new session in the provided scope, assuming session tracking is enabled. This is a no-op context manager if session tracking is not enabled. r( N)r. r, r- )r r) r! r# r# r$ r0 { s r0 )�exited�abnormal�crashed�d c C s t |�t| �� �d�S )N)�attrsZ aggregates)�dict�list�values)Zaggregate_statesr6 r# r# r$ �make_aggregate_envelope� 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 5 Q R X t� }|D ],}t|j�tkrV| �|� t� }|�|� q4|�� D ]6\}}t|j�tkr�| �|� t� }|� t ||�� qjt|j�dkr�| �|� d S )Nr ) r? rC r@ r �len�items�MAX_ENVELOPE_ITEMSr= � add_sessionZ add_sessionsr: )rF r? r@ �envelope�sessionr6 �statesr# r# r$ �flush� s$ zSessionFlusher.flushc s� � j t�� kr � jdk r dS � j�� � j t�� krJ� jdk rJW 5 Q R � dS � fdd�}t|d�}d|_z |�� W n& tk r� d� _ Y W 5 Q R � dS X |� _t�� � _ W 5 Q R X dS )a% Check that we have an active thread to run in, or create one if not. Note that this might fail (e.g. in Python 3.12 it's not possible to spawn new threads at interpreter shutdown). In that case self._running will be False after running this function. Nc s& � j r"t�� j� � j r � �� q d S �N)rE �time�sleepr> rO r# �rF r# r$ rA � s z/SessionFlusher._ensure_running.<locals>._thread)�targetTF) rD �os�getpidrA rB r �daemon�start� RuntimeErrorrE )rF rA �threadr# rS r$ �_ensure_running� 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 5 Q R X d S ) NF)Zwith_user_info�startedr4 r � r3 Zerroredr2 ) rC Zget_json_attrs�tuple�sortedrI Ztruncated_startedr@ � setdefaultr �statusr �errors)rF rM r6 Z primary_keyZ secondary_keyrN �stater# r# r$ �add_aggregate_session� s z$SessionFlusher.add_aggregate_sessionc C s2 |j dkr| �|� n| j�|�� � | �� d S )N�request)r) rd r? �append�to_jsonr[ )rF rM r# r# r$ rK s zSessionFlusher.add_sessionc C s d| _ d S )NF)rE rS r# r# r$ �kill s zSessionFlusher.killc C s | � � d S rP )rh rS r# r# r$ �__del__ s zSessionFlusher.__del__N)r< ) �__name__� __module__� __qualname__rG rO r[ rd rK rh ri r# r# r# r$ r; � s � % r; )N)Nr& )r& )r& )"rU rQ r � threadingr r � contextlibr r Zsentry_sdk.enveloper Zsentry_sdk.sessionr Zsentry_sdk.utilsr �typingr r r r r r r r r% r r/ r. r1 r0 ZTERMINAL_SESSION_STATESrJ r: r; r# r# r# r$ �<module> s<