OwlCyberSecurity - MANAGER
Edit File: context_managers.cpython-312.pyc
� B��g' � � � d dl mZ d dlmZ d dlmZmZmZmZm Z m Z m Z m Z m Z ddlmZ e rddlmZ e dedef � � Z G d � d � Z G d � d � Z G d� d� Zy)� )� default_timer)� TracebackType) �Any�Callable�Literal�Optional�Tuple�Type� TYPE_CHECKING�TypeVar�Union� ��decorate)�Counter�F.)�boundc � � e Zd Zdddeee eee df f ddfd�Zdd�Zd e ee d e e d e e de d fd �Z dd�Z y)�ExceptionCounter�counterr � exception.�returnNc � � || _ || _ y �N)�_counter� _exception)�selfr r s �Q/opt/hc_python/lib/python3.12/site-packages/prometheus_client/context_managers.py�__init__zExceptionCounter.__init__ s � � �� �#��� c � � y r � �r s r � __enter__zExceptionCounter.__enter__ s � � r �typ�value� tracebackFc �d � t || j � r| j j � y)NF)� isinstancer r �inc�r r% r& r' s r �__exit__zExceptionCounter.__exit__ s# � � �e�T�_�_� -� �M�M� � � �r c �& � � � fd�}t ||� S )Nc �D �� �5 | |i |��cd d d � S # 1 sw Y y xY wr r" ��func�args�kwargsr s �r �wrappedz*ExceptionCounter.__call__.<locals>.wrapped � �� ���T�,�V�,� ���� �� r �r �fr3 s ` r �__call__zExceptionCounter.__call__ � �� � -� ��7�#�#r )r N�r7 r r r )�__name__� __module__� __qualname__r r � BaseExceptionr r r$ r r r r, r8 r" r r r r s� � �$� � $�e�D� �<O�QV�W[�\i�Wj�lo�Wo�Qp�<p�6q� $�vz� $� ��H�T�-�%8� 9� �(�=�BY� �fn�o|�f}� � CJ� KP� CQ� � $r r c �& � e Zd Zd� Zd� Zd� Zdd�Zy)�InprogressTrackerc � � || _ y r )�_gauge)r �gauges r r zInprogressTracker.__init__% s � ��� r c �8 � | j j � y r )rB r* r# s r r$ zInprogressTracker.__enter__( � � � � � ���r c �8 � | j j � y r )rB �decr+ s r r, zInprogressTracker.__exit__+ rE r c �& � � � fd�}t ||� S )Nc �D �� �5 | |i |��cd d d � S # 1 sw Y y xY wr r" r/ s �r r3 z+InprogressTracker.__call__.<locals>.wrapped/ r4 r5 r r6 s ` r r8 zInprogressTracker.__call__. r9 r Nr: )r; r< r= r r$ r, r8 r" r r r@ r@ $ s � � ���$r r@ c �2 � e Zd Zd� Zd� Zd� Zd� Zd� Zdd�Zy) �Timerc � � || _ || _ y r )�_metric�_callback_name)r �metric� callback_names r r zTimer.__init__7 s � � �� � +��r c �N � | j | j | j � S r )� __class__rM rN r# s r � _new_timerzTimer._new_timer; s � ��~�~�d�l�l�D�,?�,?�@�@r c �$ � t � | _ | S r )r �_startr# s r r$ zTimer.__enter__> s � �#�o�� �� r c � � t t � | j z d� }t | j | j � } ||� y )Nr )�maxr rU �getattrrM rN )r r% r& r' �duration�callbacks r r, zTimer.__exit__B s7 � ��}�����4�a�8���4�<�<��)<�)<�=����r c �F � | j j |i |��| _ y r )rM �labels)r r1 �kws r r\ z Timer.labelsH s � �*�t�|�|�*�*�D�7�B�7�� r c �& � � � fd�}t ||� S )Nc �` �� �j � 5 | |i |��cd d d � S # 1 sw Y y xY wr )rS r/ s �r r3 z Timer.__call__.<locals>.wrappedL s( �� � � � �"��T�,�V�,� #�"�"�s �$�-r r6 s ` r r8 zTimer.__call__K s �� � -� ��7�#�#r Nr: ) r; r< r= r rS r$ r, r\ r8 r" r r rK rK 6 s! � �,�A��� 8�$r rK N)�timeitr �typesr �typingr r r r r r r r r � decoratorr � r r r r@ rK r" r r �<module>re sX �� � �� � � �����8�C��H�-�.�A�$� $�*$� $�$ $� $r