__init__.py

# -*- coding: utf-8 -*-
from __future__ import absolute_import

import sys

from .. import current_app
from ..local import Proxy
from ..utils import get_cls_by_name
from ..utils.functional import memoize

UNKNOWN_BACKEND = """\
Unknown result backend: {0!r}.  Did you spell that correctly? ({1!r})\
"""

BACKEND_ALIASES = {
    "amqp": "celery.backends.amqp:AMQPBackend",
    "cache": "celery.backends.cache:CacheBackend",
    "redis": "celery.backends.redis:RedisBackend",
    "mongodb": "celery.backends.mongodb:MongoBackend",
    "tyrant": "celery.backends.tyrant:TyrantBackend",
    "database": "celery.backends.database:DatabaseBackend",
    "cassandra": "celery.backends.cassandra:CassandraBackend",
    "disabled": "celery.backends.base:DisabledBackend",
}


@memoize(100)
def get_backend_cls(backend=None, loader=None):
    """Get backend class by name/alias"""
    backend = backend or "disabled"
    loader = loader or current_app.loader
    aliases = dict(BACKEND_ALIASES, **loader.override_backends)
    try:
        return get_cls_by_name(backend, aliases)
    except ValueError as exc:
        raise ValueError, ValueError(UNKNOWN_BACKEND.format(
                    backend, exc)), sys.exc_info()[2]


# deprecate this.
default_backend = Proxy(lambda: current_app.backend)