U
vi # @ sh d Z ddlZddlmZ ddlmZmZmZmZm Z m
Z
m Z m Z m
Z
mZ ddlmZ G dd deZdS )z
Process and analyze your data with tasks in the InfluxDB task engine.
Use tasks (scheduled Flux queries) to input a data stream and then analyze, modify, and act on the data accordingly.
N)List)
TasksServiceTaskTaskCreateRequestTaskUpdateRequest
LabelResponse LabelMapping AddResourceMemberRequestBody RunManuallyRunLogEvent)
_Paginatedc @ s e Zd ZdZdd ZedddZdd Zd d
ZdLee ed d
dZ
e e e e edddZ
edddZe e e e edddZe dddZeeddd Zeed d d Zeedd d!Zd"d# Ze e ed$d%d&Ze e d'd(d)Ze dd*d+Zd,d- Zd.d/ Zd0d1 Z d2d3 Z d4d5 Z e d6 dd7d8Z e e e!d9d:d;Z"e e e d< d9d=d>Z#dMe e$d?d@dAZ%e e dBdCdDZ&e e dBdEdFZ'e e d< dGdHdIZ(dJdK Z)d S )NTasksApiz,Implementation for '/api/v2/tasks' endpoint.c C s || _ t|j| _dS )zInitialize defaults.N)Z_influxdb_clientr Z
api_client_service)selfinfluxdb_client r Q/opt/alt/python38/lib/python3.8/site-packages/influxdb_client/client/tasks_api.py__init__ s zTasksApi.__init__)returnc C s | j |}|S )zRetrieve a task.)r Z get_tasks_id)r task_idtaskr r r find_task_by_id s zTasksApi.find_task_by_idc K s | j jf |jS )a List all tasks up to set limit (max 500).
:key str name: only returns tasks with the specified name
:key str after: returns tasks after specified ID
:key str user: filter tasks to a specific user ID
:key str org: filter tasks to a specific organization name
:key str org_id: filter tasks to a specific organization ID
:key int limit: the number of tasks to return
:return: Tasks
)r get_taskstasksr kwargsr r r
find_tasks s zTasksApi.find_tasksc K s t | jjdd jf |S )a Iterate over all tasks with pagination.
:key str name: only returns tasks with the specified name
:key str after: returns tasks after specified ID
:key str user: filter tasks to a specific user ID
:key str org: filter tasks to a specific organization name
:key str org_id: filter tasks to a specific organization ID
:key int limit: the number of tasks in one page
:return: Tasks iterator
c S s | j S )N)r )responser r r 4 z*TasksApi.find_tasks_iter..)r
r r Z find_iterr r r r find_tasks_iter) s zTasksApi.find_tasks_iterN)r task_create_requestr c C sP |dk r| j |S |dk rDt|j|j|j|j|jd}| j|dS t ddS )zCreate a new task.N)fluxorg_idorg descriptionstatus)r" z,task or task_create_request must be not None)
r Z
post_tasksr r# r$ r% r& r' create_task
ValueError)r r r" requestr r r r( 6 s zTasksApi.create_task)namer# r$ r c C sf t d| |d|d}d}|d k r.|d7 }||7 }|d k rN|d7 }|d| d 7 }d|| |}||_|S ) Nr active)idr+ r$ r' r# zevery: zcron: "z%{}
option task = {{name: "{}", {}}})r formatr# )r+ r# everycronr$ r Z
repetitionZflux_with_optionsr r r _create_taskC s zTasksApi._create_taskc C s | |||d|j}| |S )z1Create a new task with every repetition schedule.N)r3 r- r( )r r+ r# r1 Z organizationr r r r create_task_everyV s zTasksApi.create_task_every)r+ r# r2 r$ r c C s | j ||||dd}| |S )z0Create a new task with cron repetition schedule.N)r+ r# r2 r$ r1 )r3 r( )r r+ r# r2 r$ r r r r create_task_cron[ s zTasksApi.create_task_cronr c C s |dk r| j j|dS dS )zDelete a task.Nr6 )r Zdelete_tasks_idr r r r r delete_task` s zTasksApi.delete_task)r r c C s0 t |j|j|j|j|j|jd}| j|j|dS )Update a task.)r# r& r1 r2 r' offsetr task_update_request) r r# r& r1 r2 r' r: update_task_requestr- )r r reqr r r update_taske s
zTasksApi.update_task)r< r c C s | j j||dS )r9 r; )r Zpatch_tasks_id)r r r< r r r r= l s z TasksApi.update_task_requestc C sX t d|j|j|j|jdd}| |}|jrT| |j}|jD ]}| |j|j q>|S )z
Clone a task.r r, )r- r+ r$ r% r# r' )
r r+ r$ r% r# r( r-
get_labelslabels add_label)r r ZclonedcreatedrA labelr r r
clone_taskp s
zTasksApi.clone_taskc C s | j j|dS )zList all labels for a task.r6 )r Zget_tasks_id_labelsr7 r r r r@ { s zTasksApi.get_labels)label_idr r c C s t |d}| jj||dS )zAdd a label to a task.)rF )r
label_mapping)r r Zpost_tasks_id_labels)r rF r rG r r r rB s
zTasksApi.add_label)rF r c C s | j j||dS )zDelete a label from a task.)r rF )r Zdelete_tasks_id_labels_id)r rF r r r r delete_label s zTasksApi.delete_labelc C s | j j|djS )zList all task members.r6 )r Zget_tasks_id_membersusersr7 r r r get_members s zTasksApi.get_membersc C s t |d}| jj||dS )zAdd a member to a task.r- r Z add_resource_member_request_body)r r Zpost_tasks_id_members)r member_idr userr r r
add_member s
zTasksApi.add_memberc C s | j j||dS )z Remove a member from a task.Zuser_idr )r Zdelete_tasks_id_members_id)r rM r r r r
delete_member s zTasksApi.delete_memberc C s | j j|djS )zList all owners of a task.r6 )r Zget_tasks_id_ownersrI r7 r r r
get_owners s zTasksApi.get_ownersc C s t |d}| jj||dS )zAdd an owner to a task.rK rL )r r Zpost_tasks_id_owners)r owner_idr rN r r r add_owner s
zTasksApi.add_ownerc C s | j j||dS )z Remove an owner from a task.rP )r Zdelete_tasks_id_owners_id)r rS r r r r delete_owner s zTasksApi.delete_ownerr c K s | j jf d|i|jS )a~
Retrieve list of run records for a task.
:param task_id: task id
:key str after: returns runs after specified ID
:key int limit: the number of runs to return
:key datetime after_time: filter runs to those scheduled after this time, RFC3339
:key datetime before_time: filter runs to those scheduled before this time, RFC3339
r )r Zget_tasks_id_runsruns)r r r r r r get_runs s
zTasksApi.get_runs)r run_idr c C s | j j||dS )z
Get run record for specific task and run id.
:param task_id: task id
:param run_id: run id
:return: Run for specified task and run id
r rX )r Zget_tasks_id_runs_idr r rX r r r get_run s zTasksApi.get_runr c C s | j j||djS )z Retrieve all logs for a run.rY )r Zget_tasks_id_runs_id_logseventsrZ r r r get_run_logs s zTasksApi.get_run_logs)r
scheduled_forc C s t |d}| jj||dS )z
Manually start a run of the task now overriding the current schedule.
:param task_id:
:param scheduled_for: planned execution
)r^ )r run_manually)r
r Zpost_tasks_id_runs)r r r^ rr r r r_ s
zTasksApi.run_manuallyrY c C s | j j||dS )zb
Retry a task run.
:param task_id: task id
:param run_id: run id
rY )r Zpost_tasks_id_runs_id_retryrZ r r r retry_run s zTasksApi.retry_runc C s | j j||dS )za
Cancel a currently running run.
:param task_id:
:param run_id:
rY )r Zdelete_tasks_id_runs_idrZ r r r
cancel_run s zTasksApi.cancel_run)r r c C s | j j|djS )zP
Retrieve all logs for a task.
:param task_id: task id
r6 )r Zget_tasks_id_logsr\ r7 r r r get_logs s zTasksApi.get_logsc C s | j |dS )zList all tasks by user.)rN )r )r Z task_user_idr r r find_tasks_by_user s zTasksApi.find_tasks_by_user)NN)N)*__name__
__module__ __qualname____doc__r r r r r! r r( staticmethodstrr3 r4 r5 r8 r? r r= rE r@ r rB rH rJ rO rQ rR rT rU r rW r r[ r] datetimer_ ra rb rc rd r r r r r s>
r )rh rk typingr r r r r r r r r r
r r Z influxdb_client.client._pagesr
objectr r r r r s
0