Mercurial > kallithea
view rhodecode/lib/celerypylons/loader.py @ 2095:17c9393e9645 beta
docs
author | Marcin Kuzminski <marcin@python-works.com> |
---|---|
date | Fri, 02 Mar 2012 22:14:03 +0200 |
parents | 3a7f5b1a19dd |
children | ffd45b185016 |
line wrap: on
line source
from celery.loaders.base import BaseLoader from pylons import config to_pylons = lambda x: x.replace('_', '.').lower() to_celery = lambda x: x.replace('.', '_').upper() LIST_PARAMS = """CELERY_IMPORTS ADMINS ROUTES""".split() class PylonsSettingsProxy(object): """Pylons Settings Proxy Proxies settings from pylons.config """ def __getattr__(self, key): pylons_key = to_pylons(key) try: value = config[pylons_key] if key in LIST_PARAMS:return value.split() return self.type_converter(value) except KeyError: raise AttributeError(pylons_key) def get(self, key): try: return self.__getattr__(key) except AttributeError: return None def __getitem__(self, key): try: return self.__getattr__(key) except AttributeError: raise KeyError() def __setattr__(self, key, value): pylons_key = to_pylons(key) config[pylons_key] = value def __setitem__(self, key, value): self.__setattr__(key, value) def type_converter(self, value): #cast to int if value.isdigit(): return int(value) #cast to bool if value.lower() in ['true', 'false']: return value.lower() == 'true' return value class PylonsLoader(BaseLoader): """Pylons celery loader Maps the celery config onto pylons.config """ def read_configuration(self): self.configured = True return PylonsSettingsProxy() def on_worker_init(self): """ Import task modules. """ self.import_default_modules()