diff options
author | Christopher Allan Webber <cwebber@dustycloud.org> | 2011-07-02 15:48:49 -0500 |
---|---|---|
committer | Christopher Allan Webber <cwebber@dustycloud.org> | 2011-07-02 15:48:49 -0500 |
commit | 5d5eb6c3310ac7dea78a5788919a848dd3203616 (patch) | |
tree | fafdce96bddaedd6aa011b10eca9f3707e6afb5a /mediagoblin/init/celery/from_celery.py | |
parent | 2110408f4437a9bd7d4d158139cf869518c46456 (diff) | |
parent | 421129b6bbf460057831914c25ab5694f5d21d4f (diff) | |
download | mediagoblin-5d5eb6c3310ac7dea78a5788919a848dd3203616.tar.lz mediagoblin-5d5eb6c3310ac7dea78a5788919a848dd3203616.tar.xz mediagoblin-5d5eb6c3310ac7dea78a5788919a848dd3203616.zip |
Merge remote branch 'remotes/elrond/dev/init'
Diffstat (limited to 'mediagoblin/init/celery/from_celery.py')
-rw-r--r-- | mediagoblin/init/celery/from_celery.py | 60 |
1 files changed, 60 insertions, 0 deletions
diff --git a/mediagoblin/init/celery/from_celery.py b/mediagoblin/init/celery/from_celery.py new file mode 100644 index 00000000..c053591b --- /dev/null +++ b/mediagoblin/init/celery/from_celery.py @@ -0,0 +1,60 @@ +# GNU MediaGoblin -- federated, autonomous media hosting +# Copyright (C) 2011 Free Software Foundation, Inc +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program 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 Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +import os + +from mediagoblin import app, mg_globals +from mediagoblin.init.celery import setup_celery_from_config + + +OUR_MODULENAME = __name__ + + +def setup_self(check_environ_for_conf=True, module_name=OUR_MODULENAME, + default_conf_file='mediagoblin.ini'): + """ + Transform this module into a celery config module by reading the + mediagoblin config file. Set the environment variable + MEDIAGOBLIN_CONFIG to specify where this config file is. + + By default it defaults to 'mediagoblin.ini'. + + Note that if celery_setup_elsewhere is set in your config file, + this simply won't work. + """ + if check_environ_for_conf: + mgoblin_conf_file = os.path.abspath( + os.environ.get('MEDIAGOBLIN_CONFIG', default_conf_file)) + else: + mgoblin_conf_file = default_conf_file + + if not os.path.exists(mgoblin_conf_file): + raise IOError( + "MEDIAGOBLIN_CONFIG not set or file does not exist") + + # By setting the environment variable here we should ensure that + # this is the module that gets set up. + os.environ['CELERY_CONFIG_MODULE'] = module_name + app.MediaGoblinApp(mgoblin_conf_file, setup_celery=False) + + setup_celery_from_config( + mg_globals.app_config, mg_globals.global_config, + settings_module=module_name, + set_environ=False) + + +if os.environ['CELERY_CONFIG_MODULE'] == OUR_MODULENAME: + setup_self() |