__init__.py 850 B

1234567891011121314151617181920212223
  1. from celery.app import app_or_default
  2. from celery.utils import get_cls_by_name
  3. BACKEND_ALIASES = {
  4. "amqp": "celery.backends.amqp.AMQPBackend",
  5. "cache": "celery.backends.cache.CacheBackend",
  6. "redis": "celery.backends.pyredis.RedisBackend",
  7. "mongodb": "celery.backends.mongodb.MongoBackend",
  8. "tyrant": "celery.backends.tyrant.TyrantBackend",
  9. "database": "celery.backends.database.DatabaseBackend",
  10. "cassandra": "celery.backends.cassandra.CassandraBackend",
  11. }
  12. _backend_cache = {}
  13. def get_backend_cls(backend, loader=None):
  14. """Get backend class by name/alias"""
  15. loader = loader or app_or_default().loader
  16. if backend not in _backend_cache:
  17. aliases = dict(BACKEND_ALIASES, **loader.override_backends)
  18. _backend_cache[backend] = get_cls_by_name(backend, aliases)
  19. return _backend_cache[backend]