loader.py
70 lines
| 1.7 KiB
| text/x-python
|
PythonLexer
|
r4116 | # -*- coding: utf-8 -*- | ||
r776 | 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] | ||||
r1002 | if key in LIST_PARAMS:return value.split() | |||
r776 | return self.type_converter(value) | |||
except KeyError: | ||||
raise AttributeError(pylons_key) | ||||
r1002 | 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() | ||||
r776 | def __setattr__(self, key, value): | |||
pylons_key = to_pylons(key) | ||||
config[pylons_key] = value | ||||
r1002 | def __setitem__(self, key, value): | |||
self.__setattr__(key, value) | ||||
r776 | ||||
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() | ||||