Question

我正在重構我的代碼以使用芹菜工人。

之前我曾經使用argparse來傳遞命令行參數。

例如

if __name__ == "__main__": 
    parser = argparse.ArgumentParser(description='Node') 
    parser.add_argument('--environment', action="store", default='local', help="env e.g. production of development") 
    environment = arg_options.environment 

但現在我得到這個錯誤。

celery -A tasks worker --loglevel=info --environment local 
celery: error: no such option: --environment 

如何添加？

如果我不需要，我不想使用環境變量。

e.g export environment=development 

Answer 1

芹菜工人不執行您的__main__。

如果你想添加額外的命令行選項，你可以使用app.user_options， 但要注意，它使用optparse模塊，不​​。

看到文檔本節更多：

http://docs.celeryproject.org/en/latest/userguide/extending.html#preload-options

Answer 2

通過查看源代碼，我已經找到了一種方法來處理這個問題。

在celery.py附加：

from celery import bootsteps 
from celery.bin import Option 

.... 
app.user_options['worker'].add(
    Option('--server', dest='api_server', default='127.0.0.1', 
      help='API server to use.') 
) 

app.conf['API_SERVER'] = '127.0.0.1' 

class ConfigBootstep(bootsteps.Step): 
    def __init__(self, worker, api_server=None, **options): 
     app.conf['API_SERVER'] = api_server 

app.steps['worker'].add(ConfigBootstep) 

然後在文件牽着你的任務補充一點：

from celery import current_app 
... 
@shared_task() 
def roboscope(): 
    API_SERVER = current_app.conf.get('API_SERVER', '127.0.0.1') 

我試圖解決它的時候是在模塊導入使API_SERVER模塊全局變量但它不起作用，因爲它爲時尚早。由於我的任務非常緊張，因此運行這麼多次並不會造成任何負面影響。