File: //home/arjun/projects/env/lib/python3.10/site-packages/boto/manage/__pycache__/task.cpython-310.pyc
o
we} � @ sx d dl Z d dlmZmZmZ d dlmZ d dlZd dlZd dl Z d dl
m Z dd� Z G dd� de�Z
G d d
� d
e�ZdS ) � N)�StringProperty�DateTimeProperty�IntegerProperty)�Model)�StringIOc C s, | dkrd S t | �dk st | �dkrt�d S )N�*r � )�int�
ValueError)�val� r �I/home/arjun/projects/env/lib/python3.10/site-packages/boto/manage/task.py�
check_hour s
�r c s� e Zd ZdZe� Zededd�Zedd�Ze � Z
e � Z e� Z
e� Zedd� �Zd� fd d
� Zd d � Zd
d� Zddd�Zdd� Z� ZS )�Taska�
A scheduled, repeating task that can be executed by any participating servers.
The scheduling is similar to cron jobs. Each task has an hour attribute.
The allowable values for hour are [0-23|*].
To keep the operation reasonably efficient and not cause excessive polling,
the minimum granularity of a Task is hourly. Some examples:
hour='*' - the task would be executed each hour
hour='3' - the task would be executed at 3AM GMT each day.
Tr )�required� validator�default)r c C s | � � D ]}|�|� qd S �N)�all�start)�cls�
queue_name�taskr r r
� start_all9 s �zTask.start_allNc s@ t t| �j|fi |�� | jdk| _| jdk| _tj�� | _d S )Nr ) �superr �__init__�hour�hourly�daily�datetime�utcnow�now)�self�id�kw�� __class__r r
r > s z
Task.__init__c C s� t j�d| j| j| jf � | jr| jsdS | jr>| js>t| j �| jj kr'dS t
t| j �| jj | jj t| j � �d d S | j| j }| jrS|j dkrNdS d|j S t| j �| jj kre|j dkrcdS dS t
t| j �| jj | jj t| j � �d d S )ag
Determine how long until the next scheduled time for a Task.
Returns the number of seconds until the next scheduled time or zero
if the task needs to be run immediately.
If it's an hourly task and it's never been run, run it now.
If it's a daily task and it's never been run and the hour is right, run it now.
z!checking Task[%s]-now=%s, last=%sr �< i � ipC )
�boto�log�info�namer! �
last_executedr r r r �max�seconds�days)r" �deltar r r
�checkD s" .
.z
Task.checkc C s t j�d| j| jf � t� }tj| jdtjtjtjd�}d}|}|� � d u rZt j�d||f � ||krK||7 }t j�d| j|f � |rK|�
|� t � d� |d7 }|� � d u s'|�
� }|�|d � |�|d � t j�d | j|�� f � | j| _|j| _|�� dd
� | _d S ) NzTask[%s] - running:%sT)�shell�stdin�stdout�stderr� znsecs=%s, timeout=%sz(Task[%s] - setting timeout to %d secondsr r( zTask[%s] - output: %si� )r) r* r+ r, �commandr �
subprocess�Popen�PIPE�poll�change_visibility�time�sleep� communicate�write�getvaluer! r- �
returncode� last_status� last_output)r" �msg�vtimeout�log_fp�process�nsecs�current_timeout�tr r r
�_runf s0 �
� z Task._runr' c C s� | � � }tj�d| j|f � |dkrM| �||� |j}|�| j�}|� |�}|j| _
| � � tj�d| j|jf � |� � tj�d| j|jf � d S tj�d| � |�
|� d S )NzTask[%s] - delay=%s secondsr z Task[%s] - new message id=%sz Task[%s] - deleted message %sznew_vtimeout: %d)r2 r) r* r+ r, rM �queue� new_messager# rA �
message_id�put�deleter= )r" rF rG �delayrN �new_msgr r r
�run~ s
zTask.runc C s^ t j�d| j|f � t �d|�}|�| j�}|�|�}|j| _| � � t j�d| j � d S )Nz"Task[%s] - starting with queue: %s�sqszTask[%s] - start successful)
r) r* r+ r, �lookuprO r# rA rP rQ )r" r rN rF r r r
r � s
z
Task.startr )r' )�__name__�
__module__� __qualname__�__doc__r r, r r r8 r r- r rD rE rP � classmethodr r r2 rM rU r �
__classcell__r r r% r
r # s
"
r c @ s e Zd Zdd� Zddd�ZdS )�
TaskPollerc C s t �� | _| j�|�| _d S r )r) � connect_sqsrV rW rN )r" r r r r
r � s
zTaskPoller.__init__r' c C s| | j �|�}|r8t�|�� �}|r7|jr|j|jkr.tj� d|j
|jf � |� ||� ntj� d|j
� nt �
|� q)NTzTask[%s] - read message %sz-Task[%s] - found extraneous message, ignoring)rN �readr � get_by_id�get_bodyrP r# r) r* r+ r, rU r> r? )r" �waitrG �mr r r r
r< � s �
�zTaskPoller.pollN)r' r' )rX rY rZ r r< r r r r
r^ � s r^ )r) �boto.sdb.db.propertyr r r �boto.sdb.db.modelr r r9 r> � boto.compatr r r �objectr^ r r r r
�<module> s u