File: //usr/local/lib/python3.9/site-packages/kombu/asynchronous/__pycache__/timer.cpython-39.pyc
a
X>h
� @ s* d Z ddlmZ ddlZddlZddlmZ ddlmZ ddlm Z ddl
m Z ddl
m
Z dd l
mZ dd
lmZ dd lmZ dd lmZ ejd
kr�ddlmZ n ddlmZ er�ddlmZ dZ ee �Z dZ e� ded��Z!e"ed�Z#edd�Z$ed�e fdd�Z%e G dd� d��Z&G dd� d�Z'dS ) z"Timer scheduling Python callbacks.� )� annotationsN)�
namedtuple)�datetime)�total_ordering)� monotonic)�time)�
TYPE_CHECKING)�proxy)�wraps)�
get_logger)� � )�ZoneInfo)�
TracebackType)�Entry�Timer� to_timestamp� �UTC�pypy_version_info� scheduled)�eta�priority�entryc C sF t | t�rB| jdu r | j|d�} t� |� }t| t �� | d�S | S )z^Convert datetime to timestamp.
If d' is already a timestamp, then that will be used.
N)�tzinfor )�
isinstancer r �replace�_time�max�EPOCH�
total_seconds)�dZdefault_timezoner �diff� r# �B/usr/local/lib/python3.9/site-packages/kombu/asynchronous/timer.pyr & s
r c @ s\ e Zd ZdZesdZddd�Zdd� Zdd � Zd
d � Z d d
� Z
e dd� �Z e j
dd� �Z dS )r zSchedule Entry.)�fun�args�kwargs�tref�canceled� _last_run� __weakref__Nc C s4 || _ |p g | _|pi | _t| �| _d | _d| _d S )NF)r% r& r' � weakrefproxyr( r* r) )�selfr% r& r' r# r# r$ �__init__= s
zEntry.__init__c C s | j | ji | j��S �N)r% r& r' �r- r# r# r$ �__call__E s zEntry.__call__c C s$ z d| j _W n ty Y n0 d S �NT)r( r) �ReferenceErrorr0 r# r# r$ �cancelH s z Entry.cancelc C s d� | jj| j| j�S )Nz <TimerEntry: {}(*{!r}, **{!r}))�formatr% �__name__r& r' r0 r# r# r$ �__repr__N s �zEntry.__repr__c C s t | �t |�k S r/ )�id)r- �otherr# r# r$ �__lt__S s z Entry.__lt__c C s | j S r/ �r) r0 r# r# r$ � cancelledV s zEntry.cancelledc C s
|| _ d S r/ r; )r- �valuer# r# r$ r< Z s )NN)r6 �
__module__� __qualname__�__doc__�IS_PYPY� __slots__r. r1 r4 r7 r: �propertyr< �setterr# r# r# r$ r 3 s
r c @ s� e Zd ZdZeZdZd0dd�Zdd� Zddd d
d �d d
�Zd1dd�Z d2dd�Z
d3dd�Z dde fdd�Z
de fdd�Zejfdd�Zd d � Zd d � Zd d!� Zee ejejfd"d#�Zd$d%� Zd&d'� Zd(d)� Zd*d+� Ze ejfd,d-��Z e d.d/� �Z dS )4r zAsync timer implementation.Nc K s$ t |pt�| _|p| j| _g | _d S r/ )�float�DEFAULT_MAX_INTERVAL� max_interval�on_error�_queue)r- rG rH r' r# r# r$ r. f s zTimer.__init__c C s | S r/ r# r0 r# r# r$ � __enter__k s zTimer.__enter__ztype[BaseException] | NonezBaseException | NonezTracebackType | None�None)�exc_type�exc_val�exc_tb�returnc C s | � � d S r/ )�stop)r- rL rM rN r# r# r$ �__exit__n s zTimer.__exit__r# r c C s$ |si n|}| � | �|||�||�S r/ )�enter_atr )r- r r% r& r' r r# r# r$ �call_atv s z
Timer.call_atc C s$ |si n|}| � || �|||�|�S r/ )� enter_afterr )r- �secsr% r& r' r r# r# r$ �
call_afterz s zTimer.call_afterc sP |si n|}�� � ||��t� �� ����fdd��}|�_d �_������S )Nc s� �j t� }}|r |�j n�}z�|rr|�krr|�_ � | i |��W �jsp�j }|r^�|| n�}��|��� S W �jsԈj }|r��|| n�}��|��� n0�js҈j }|r��|| n�}��|��� 0 d S r/ )r* r r) rT )r& r' �last�nowZlsince�next�r% r rU r- r( r# r$ � _reschedules� s& ��z+Timer.call_repeatedly.<locals>._reschedules)r r
r% r* rT )r- rU r% r& r' r r[ r# rZ r$ �call_repeatedly~ s
zTimer.call_repeatedlyc
C sf |du r|� }t |t�rXz t|�}W n2 tyV } z| �|�s@� W Y d}~dS d}~0 0 | �|||�S )z�Enter function into the scheduler.
Arguments:
---------
entry (~kombu.asynchronous.timer.Entry): Item to enter.
eta (datetime.datetime): Scheduled time.
priority (int): Unused.
N)r r r � Exception� handle_error�_enter)r- r r r r �excr# r# r$ rR � s
zTimer.enter_atc C s | � ||� t|� |�S r/ )rR rE )r- rU r r r r# r# r$ rT � s zTimer.enter_afterc C s || j t|||�� |S r/ )rI r )r- r r r �pushr# r# r$ r_ � s z Timer._enterc
C sN z
|� W n> t yH } z&| �|�s4tjd|dd� W Y d }~n
d }~0 0 d S )NzError in timer: %rT)�exc_info)r] r^ �logger�error)r- r r` r# r# r$ � apply_entry� s
zTimer.apply_entryc C s | j r| � |� dS d S r2 )rH )r- rb r# r# r$ r^ � s
zTimer.handle_errorc C s d S r/ r# r0 r# r# r$ rP � s z
Timer.stopc c s� | j }| j}|r~|d }|� |d }} || k rF|| | |�dfV q�||�}
|
|u rr|d } | js d| fV q q�|||
� q dV q dS )z�Iterate over schedule.
This iterator yields a tuple of ``(wait_seconds, entry)``,
where if entry is :const:`None` the caller should wait
for ``wait_seconds`` until it polls the schedule again.
r Nr )NN)rG rI r) ) r- �minZnowfun�popra rG �queueZeventArX r ZeventBr r# r# r$ �__iter__� s
zTimer.__iter__c C s g | j d d �< d S r/ )rI r0 r# r# r$ �clear� s z Timer.clearc C s |� � d S r/ )r4 )r- r( r# r# r$ r4 � s z Timer.cancelc C s
t | j�S r/ )�lenrI r0 r# r# r$ �__len__� s z
Timer.__len__c C s dS r2 r# r0 r# r# r$ � __nonzero__� s zTimer.__nonzero__c s& t | j�}� fdd�|gt|� D �S )z%Snapshot of underlying datastructure.c s g | ] }� |��qS r# r# )�.0�v��_popr# r$ �
<listcomp>� � z Timer.queue.<locals>.<listcomp>)�listrI rk )r- rq �eventsr# rp r$ rh � s
z Timer.queuec C s | S r/ r# r0 r# r# r$ �schedule� s zTimer.schedule)NN)r# Nr )r# Nr )r# Nr ) r6 r> r? r@ r rH r. rJ rQ rS rV r\ r rR rT �heapq�heappushr_ re r^ rP rf �heappopri rj r4 rl rm rC rh rv r# r# r# r$ r _ s4
�
r )(r@ �
__future__r rw �sys� collectionsr r � functoolsr r r r �typingr �weakrefr r, Z
vine.utilsr
Z kombu.logr � version_infoZzoneinfor Zbackports.zoneinfo�typesr �__all__r6 rc rF �
fromtimestampr �hasattrrA r r r r r# r# r# r$ �<module> s6
+