我想在运行时注册的任务中使用芹菜的最佳方法是哪种。我的工作流程如下:
答案 0 :(得分:2)
我完成的方式是基于“插件”概念,基于click package has with custom subcommands所具有的相同想法。
应用程序结构(基于python 3):
.
├── dynamic_tasks.py
├── run.py
└── tasks
└── get_rate.py
芹菜任务 dynamic_tasks.py 的定义如下:
import os
import celery
app = celery.Celery('dynamic_tasks', broker='amqp://guest@192.168.169.1/', backend='rpc://')
PLUGIN_FOLDER = os.path.join(os.path.dirname(__file__), 'tasks')
def _absolutepath(filename):
""" Return the absolute path to the filename"""
return os.path.join(PLUGIN_FOLDER, filename)
@app.task
def tasks(funcname, *args, **kwargs):
try:
funcname = funcname.replace('-', '_')
funcname += '.py'
func = _absolutepath(funcname)
ns = {}
with open(func) as f:
code = compile(f.read(), func, 'exec')
eval(code, ns, ns)
return ns['task'](*args, **kwargs)
except IOError as e:
# Manage IOError
raise e
可插入任务示例 tasks / get_rate.py :
""" This task get the currency rate between a pair of currencies """
import urllib.request
URL = 'http://finance.yahoo.com/d/quotes.csv?s={}=X&f=p'
def task(pair='EURSEK', url_tmplt=URL):
with urllib.request.urlopen(url_tmplt.format(pair)) as res:
body = res.read()
return (pair, float(body.strip()))
而且,简单地说,从 run.py 运行示例:
from dynamic_tasks import tasks
print(tasks.delay('get_rate', 'EURSEK').get())
<强> EDITED 强> 由于芹菜在不同的机器上运行,因此不可能依赖本地文件系统。我的新方法是将函数发送为字符串执行:
@app.task
def dynamic_tasks(funcname, funccode, *args, **kwargs):
try:
ns = {}
code = compile(funccode, funcname, 'exec')
eval(code, ns, ns)
logger.info('execute %r with args %r, %r', funcname, args, kwargs)
return ns['task'](*args, **kwargs)
except IOError:
logger.error("Error loading the dynamic function from text %s", funcname)