Homec4science
Diffusion invenio-infoscience (maint-2.1) Browse

R3600/modules/bibworkflow/lib/workers/worker_celery.py
f973507cbbc7maint-2.1

Last Change
Raw File

worker_celery.py
View Options

## This file is part of Invenio.
## Copyright (C) 2012, 2013 CERN.
##
## Invenio is free software; you can redistribute it and/or
## modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation; either version 2 of the
## License, or (at your option) any later version.
##
## Invenio is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
from invenio.bibworkflow_worker_engine import (run_worker,
restart_worker,
continue_worker)
from invenio.celery import celery
@celery.task(name='invenio.bibworkflow_workers.worker_celery.run_worker')
def celery_run(workflow_name, data, **kwargs):
"""
Runs the workflow with Celery
"""
run_worker(workflow_name, data, **kwargs)
@celery.task(name='invenio.bibworkflow_workers.worker_celery.restart_worker')
def celery_restart(wid, **kwargs):
"""
Restarts the workflow with Celery
"""
restart_worker(wid, **kwargs)
@celery.task(name='invenio.bibworkflow_workers.worker_celery.continue_worker')
def celery_continue(oid, restart_point, **kwargs):
"""
Restarts the workflow with Celery
"""
continue_worker(oid, restart_point, **kwargs)
class worker_celery(object):
def run_worker(self, workflow_name, data, **kwargs):
"""
Helper function to get celery task
decorators to worker_celery
@param workflow_name: name of the workflow to be run
@type workflow_name: string
@param data: list of objects for the workflow
@type data: list
"""
return celery_run.delay(workflow_name, data, **kwargs)
def restart_worker(self, wid, **kwargs):
"""
Helper function to get celery task
decorators to worker_celery
@param wid: uuid of the workflow to be run
@type wid: string
"""
return celery_restart.delay(wid, **kwargs)
def continue_worker(self, oid, restart_point, **kwargs):
"""
Helper function to get celery task
decorators to worker_celery
@param oid: uuid of the object to be started
@type oid: string
@param restart_point: sets the start point
@type restart_point: string
"""
return celery_continue.delay(oid, restart_point, **kwargs)
c4science · Help