관리-도구
편집 파일: context_managers.cpython-38.pyc
U .?�f' � @ s� d dl mZ d dlmZ d dlmZmZmZmZm Z m Z m Z m Z m Z ddlmZ e rtddlmZ e dedef d �ZG d d � d �ZG d d � d �ZG dd� d�ZdS )� )� default_timer)� TracebackType) �Any�Callable�Literal�Optional�Tuple�Type� TYPE_CHECKING�TypeVar�Union� ��decorate)�Counter�F.)�boundc @ s| e Zd Zdeee eee df f dd�dd�Zdd�dd �Ze ee e e e e e d d �d d �Z ddd�dd�Z dS )�ExceptionCounterr .N)�counter� exception�returnc C s || _ || _d S �N)�_counter� _exception)�selfr r � r �P/opt/hc_python/lib/python3.8/site-packages/prometheus_client/context_managers.py�__init__ s zExceptionCounter.__init__)r c C s d S r r �r r r r � __enter__ s zExceptionCounter.__enter__F)�typ�value� tracebackr c C s t || j�r| j�� dS )NF)� isinstancer r �inc�r r r! r" r r r �__exit__ s zExceptionCounter.__exit__r ��fr c s � fdd�}t ||�S )Nc s& � � | ||�W 5 Q R � S Q R X d S r r ��func�args�kwargsr r r �wrapped s z*ExceptionCounter.__call__.<locals>.wrappedr �r r( r- r r r �__call__ s zExceptionCounter.__call__)�__name__� __module__� __qualname__r r � BaseExceptionr r r r r r r& r/ r r r r r s *(r c @ s4 e Zd Zdd� Zdd� Zdd� Zddd�d d �Zd S ) �InprogressTrackerc C s || _ d S r )�_gauge)r Zgauger r r r % s zInprogressTracker.__init__c C s | j �� d S r )r5 r$ r r r r r ( s zInprogressTracker.__enter__c C s | j �� d S r )r5 �decr% r r r r& + s zInprogressTracker.__exit__r r' c s � fdd�}t ||�S )Nc s& � � | ||�W 5 Q R � S Q R X d S r r r) r r r r- / s z+InprogressTracker.__call__.<locals>.wrappedr r. r r r r/ . s zInprogressTracker.__call__N)r0 r1 r2 r r r& r/ r r r r r4 $ s r4 c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d d �d d�ZdS )�Timerc C s || _ || _d S r )�_metric�_callback_name)r ZmetricZ callback_namer r r r 7 s zTimer.__init__c C s | � | j| j�S r )� __class__r8 r9 r r r r � _new_timer; s zTimer._new_timerc C s t � | _| S r )r �_startr r r r r > s zTimer.__enter__c C s, t t� | j d�}t| j| j�}||� d S )Nr )�maxr r< �getattrr8 r9 )r r r! r" �duration�callbackr r r r& B s zTimer.__exit__c O s | j j||�| _ d S r )r8 �labels)r r+ �kwr r r rA H s z Timer.labelsr r' c s � fdd�}t ||�S )Nc s* � � � � | ||�W 5 Q R � S Q R X d S r )r; r) r r r r- L s z Timer.__call__.<locals>.wrappedr r. r r r r/ K s zTimer.__call__N) r0 r1 r2 r r; r r& rA r/ r r r r r7 6 s r7 N)Ztimeitr �typesr �typingr r r r r r r r r � decoratorr � r r r r4 r7 r r r r �<module> s ,