mamba/dispatchers/mmemcache.py mamba/receivers/mmemcache.py mamba/returners/mmemcache.py @@ -1,3 +1,4 @@ from base import * from local import * from sqs import * +from mmemcache import * mamba/dispatchers/__init__.py @@ -5,12 +5,10 @@ class SQSDispatcher(Dispatcher): def __init__(self, *args, **kwargs): Dispatcher.__init__(self, *args, **kwargs) - if 'queue' in self.options: - self.queue = self.options['queue'] - else: - raise Exception - self.connection = SQSConnection() + + self.queue = self.options.get('queue', 'mamba') + self.q = self.connection.create_queue(self.queue) def dispatch(self): mamba/dispatchers/sqs.py @@ -1,3 +1,4 @@ from base import * from local import * from sqs import * +from mmemcache import * mamba/receivers/__init__.py @@ -140,7 +140,7 @@ class Receiver(Daemon): if 'returner' in self.options: mod = sys.modules['mamba.returners'] cls = getattr(mod, self.options['returner']) - self.returner = cls() + self.returner = cls(options=self.options) while True: self.poll() mamba/receivers/base.py @@ -9,10 +9,9 @@ class SQSReceiver(Receiver): self.connection = SQSConnection() def poll(self): - if 'queues' not in self.options: - raise Exception + queues = self.options.get('queues', ['mamba']) - for queue in self.options['queues']: + for queue in queues: q = self.connection.create_queue(queue) m = q.read() if m: mamba/receivers/sqs.py @@ -1,2 +1,3 @@ from base import * from sqs import * +from mmemcache import * mamba/returners/__init__.py @@ -18,6 +18,5 @@ class SQSReturner(Returner): val = simplejson.loads(m.get_body()) q.delete_message(m) q.delete() - return val - return m + return val mamba/returners/sqs.py 0beaad429dc94e871ed7936e6babca33c59d3eed neilmock neilmock@gmail.com http://github.com/neilmock/mamba/commit/a99b0bcfeac20eef78ddbf77eba6c5aedd410fe0 a99b0bcfeac20eef78ddbf77eba6c5aedd410fe0 2009-10-22T17:00:38-07:00 2009-10-22T17:00:38-07:00 memcache backend, sane default options 7c24e914b289085db0c749fff50f5772f52b1fc5 neilmock neilmock@gmail.com