1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
"""
kombu.transport
===============
Built-in transports.
:copyright: (c) 2009 - 2012 by Ask Solem.
:license: BSD, see LICENSE for more details.
"""
from __future__ import absolute_import
import sys
from kombu.syn import detect_environment
DEFAULT_TRANSPORT = 'amqp'
AMQP_TRANSPORT = 'kombu.transport.amqplib.Transport'
AMQP_ALIAS = 'librabbitmq'
if detect_environment() == 'default':
try:
import librabbitmq # noqa
AMQP_TRANSPORT = 'kombu.transport.librabbitmq.Transport' # noqa
AMQP_ALIAS = 'amqp' # noqa
except ImportError:
pass
def _ghettoq(name, new, alias=None):
xxx = new # stupid enclosing
def __inner():
import warnings
_new = callable(xxx) and xxx() or xxx
gtransport = 'ghettoq.taproot.%s' % name
ktransport = 'kombu.transport.%s.Transport' % _new
this = alias or name
warnings.warn("""
Ghettoq does not work with Kombu, but there is now a built-in version
of the %s transport.
You should replace %r with simply: %r
""" % (name, gtransport, this))
return ktransport
return __inner
TRANSPORT_ALIASES = {
'amqp': AMQP_TRANSPORT,
'amqplib': 'kombu.transport.amqplib.Transport',
'librabbitmq': 'kombu.transport.librabbitmq.Transport',
'pika': 'kombu.transport.pika2.Transport',
'oldpika': 'kombu.transport.pika.SyncTransport',
'memory': 'kombu.transport.memory.Transport',
'redis': 'kombu.transport.redis.Transport',
'SQS': 'kombu.transport.SQS.Transport',
'sqs': 'kombu.transport.SQS.Transport',
'beanstalk': 'kombu.transport.beanstalk.Transport',
'mongodb': 'kombu.transport.mongodb.Transport',
'couchdb': 'kombu.transport.couchdb.Transport',
'zookeeper': 'kombu.transport.zookeeper.Transport',
'django': 'kombu.transport.django.Transport',
'sqlalchemy': 'kombu.transport.sqlalchemy.Transport',
'sqla': 'kombu.transport.sqlalchemy.Transport',
'ghettoq.taproot.Redis': _ghettoq('Redis', 'redis', 'redis'),
'ghettoq.taproot.Database': _ghettoq('Database', 'django', 'django'),
'ghettoq.taproot.MongoDB': _ghettoq('MongoDB', 'mongodb'),
'ghettoq.taproot.Beanstalk': _ghettoq('Beanstalk', 'beanstalk'),
'ghettoq.taproot.CouchDB': _ghettoq('CouchDB', 'couchdb'),
'filesystem': 'kombu.transport.filesystem.Transport',
}
_transport_cache = {}
def resolve_transport(transport=None):
transport = TRANSPORT_ALIASES.get(transport, transport)
if callable(transport):
transport = transport()
transport_module_name, _, transport_cls_name = transport.rpartition('.')
if not transport_module_name:
raise KeyError('No such transport: %s' % (transport, ))
return transport_module_name, transport_cls_name
def _get_transport_cls(transport=None):
transport_module_name, transport_cls_name = resolve_transport(transport)
__import__(transport_module_name)
transport_module = sys.modules[transport_module_name]
return getattr(transport_module, transport_cls_name)
def get_transport_cls(transport=None):
"""Get transport class by name.
The transport string is the full path to a transport class, e.g.::
"kombu.transport.amqplib.Transport"
If the name does not include `"."` (is not fully qualified),
the alias table will be consulted.
"""
transport = transport or DEFAULT_TRANSPORT
if transport not in _transport_cache:
_transport_cache[transport] = _get_transport_cls(transport)
return _transport_cache[transport]
|
