在supervisord中设置环境变量时出现Python Key Error

Question

我在supervisord中设置了一个环境变量:

[program:worker]
directory = /srv/app/
command=celery -A tasks worker -Q default -l info -n default_worker.%%h
environment=BROKER="amqp://admin:password@xxxxx:5672//"

在我的celeryconfig.py中,然后尝试读取这样的变量.

BROKER = os.environ['BROKER']

但我仍然得到下面的错误,为什么？

  File "/usr/local/lib/python2.7/dist-packages/celery/loaders/base.py", line 106, in import_module
    return importlib.import_module(module, package=package)
  File "/usr/lib/python2.7/importlib/__init__.py", line 37, in import_module
    __import__(name)
  File "/srv/app/celeryconfig.py", line 6, in <module>
    BROKER = os.environ['BROKER']
  File "/usr/lib/python2.7/UserDict.py", line 23, in __getitem__
    raise KeyError(key)
KeyError: 'BROKER

根据评论中的建议,有一个envs的文件转储:

{
    'SUPERVISOR_GROUP_NAME': 'celery_default_worker',
    'TERM': 'linux',
    'SUPERVISOR_SERVER_URL': 'unix: ///var/run/supervisor.sock',
    'UPSTART_INSTANCE': '',
    'RUNLEVEL': '2',
    'UPSTART_EVENTS': 'runlevel',
    'PREVLEVEL': 'N',
    'SUPERVISOR_PROCESS_NAME': 'celery_default_worker',
    'UPSTART_JOB': 'rc',
    'PWD': '/',
    'SUPERVISOR_ENABLED': '1',
    'runlevel': '2',
    'PATH': '/usr/local/sbin: /usr/local/bin: /sbin: /bin: /usr/sbin: /usr/bin',
    'previous': 'N'
}

Answer 1

它看起来像一个已知的bug supervisord:

http://github.com/Supervisor/supervisor/issues/91(已解决)

http://github.com/Supervisor/supervisor/pull/550 (待定)

在这种情况下,将环境规范移动到全局范围(对于supervisord进程本身)可能是一种可接受的解决方法.

最后,如果所有其他方法都失败了,请将celeryshell脚本包装为接受此特定环境变量作为命令行参数.

Answer 2

这个答案很可能不是原因，请检查/sf/answers/2018041371/以获取有关相关supervisord bug 的信息。

我可以部分重现您的错误。当 celery 在主管中运行时，我没有看到错误。当我尝试从主管外部的环境运行任务时，我在未设置环境变量的情况下看到了错误BROKER。celeryconfig.py由 celery 和任何试图执行任务的东西执行。

我不确定这个问题是否正是您遇到的问题，如果您可以分享您如何执行任务以及何时引发异常，它可能会有所帮助。

例如，如果我尝试运行任务，ipython则会生成与您的错误匹配的错误。

In [1]: from tasks import add
In [2]: add.delay(2,3)
...
     21         if hasattr(self.__class__, "__missing__"):
     22             return self.__class__.__missing__(self, key)
---> 23         raise KeyError(key)
     24     def __setitem__(self, key, item): self.data[key] = item
     25     def __delitem__(self, key): del self.data[key]

KeyError: 'BROKER'

在本地加载celeryconfig.py，以便建立与 celery 代理和后端的连接。如果不设置环境变量，我无法执行任务BROKER。

如果我在执行任务之前设置环境变量，则相同的代码适用于我。

In [3]: import os
In [4]: os.environ["BROKER"] = "broker is set"
In [5]: add.delay(2,3)
Out[5]: <AsyncResult: 0f3xxxx-87fa-48d7-9258-173bdd2052ca>

这是我使用的文件，以防有帮助。

supervisor.conf:supervisord -c supervisor.conf

[unix_http_server]
file=/tmp/supervisor.sock

[supervisord]
loglevel = info
nodaemon = true
identifier = supervisor

[supervisorctl]
serverurl=unix:///tmp/supervisor.sock

[rpcinterface:supervisor]
supervisor.rpcinterface_factory = supervisor.rpcinterface:make_main_rpcinterface

[program:worker]
command=/app/srv/main-env/bin/celery -A tasks worker -Q default -l info -n default_worker.%%h
environment=BROKER="amqp://admin:password@xxxxx:5672//"
directory=/app/srv/
numprocs=1
stdout_logfile=/app/srv/worker.log
stderr_logfile=/app/srv/worker.log
autostart=true
autorestart=true
startsecs=10
stopwaitsecs = 600
killasgroup=true
priority=998

celeryconfig.py:

import os


BROKER = os.environ['BROKER']

tasks.py:

from celery import Celery

app = Celery(
    'tasks',
    backend='amqp',
    broker='amqp://admin:password@xxxxx:5672//')
app.config_from_object('celeryconfig')


@app.task
def add(x, y):
        return x + y