Carrot compat interface now working with celery

This commit is contained in:
Ask Solem 2010-07-20 16:14:45 +02:00
parent 58e0c08146
commit 90eaf27f85
1 changed files with 11 additions and 10 deletions

View File

@ -18,7 +18,7 @@ def entry_to_binding(queue, **options):
e_auto_delete = options.get("exchange_auto_delete") or \
options.get("auto_delete")
q_durable = options.get("queue_durable") or options.get("durable")
q_auto_delete = options.get("queue_auto_delete") or
q_auto_delete = options.get("queue_auto_delete") or \
options.get("auto_delete")
e_arguments = options.get("exchange_arguments")
q_arguments = options.get("queue_arguments")
@ -29,7 +29,7 @@ def entry_to_binding(queue, **options):
routing_key=options.get("routing_key"),
durable=e_durable,
auto_delete=e_auto_delete,
arguments=e_arguments,
arguments=e_arguments)
return entity.Binding(queue,
exchange=exchange,
routing_key=binding_key,
@ -72,7 +72,7 @@ class Publisher(messaging.Producer):
super(Publisher, self).__init__(self.backend, self.exchange,
**kwargs)
def send(*args, **kwargs):
def send(self, *args, **kwargs):
return self.publish(*args, **kwargs)
def close(self):
@ -183,7 +183,7 @@ class _CSet(messaging.Consumer):
super(_CSet, self).__init__(self.backend, *args, **kwargs)
def iterconsume(self):
raise iterconsume(self.connection, self)
return iterconsume(self.connection, self)
def add_consumer_from_dict(self, queue, **options):
self.bindings.append(entry_to_binding(queue, **options))
@ -197,12 +197,13 @@ class _CSet(messaging.Consumer):
def ConsumerSet(connection, from_dict=None, consumers=None,
callbacks=None, **options):
callbacks=None, **kwargs):
bindings = []
for consumer in consumers:
bindings.extend(consumer.bindings)
for queue, options in from_dict.items():
bindings.append(entry_to_binding(queue, **options))
if consumers:
map(bindings.extend, consumer.bindings)
if from_dict:
for queue_name, queue_options in from_dict.items():
bindings.append(entry_to_binding(queue_name, **queue_options))
return _CSet(connection, bindings, **options)
return _CSet(connection, bindings, **kwargs)