Fix queue urls for other regions. Closes #411.

This commit is contained in:
Steve Pulec 2015-09-19 09:18:16 -04:00
commit f72613cc47
3 changed files with 20 additions and 7 deletions

View file

@ -106,9 +106,10 @@ class Queue(object):
'VisibilityTimeout',
'WaitTimeSeconds']
def __init__(self, name, visibility_timeout, wait_time_seconds):
def __init__(self, name, visibility_timeout, wait_time_seconds, region):
self.name = name
self.visibility_timeout = visibility_timeout or 30
self.region = region
# wait_time_seconds will be set to immediate return messages
self.wait_time_seconds = wait_time_seconds or 0
@ -179,6 +180,10 @@ class Queue(object):
result[attribute] = getattr(self, camelcase_to_underscores(attribute))
return result
@property
def url(self):
return "http://sqs.{0}.amazonaws.com/123456789012/{1}".format(self.region, self.name)
@property
def messages(self):
return [message for message in self._messages if message.visible and not message.delayed]
@ -196,14 +201,20 @@ class Queue(object):
class SQSBackend(BaseBackend):
def __init__(self):
def __init__(self, region_name):
self.region_name = region_name
self.queues = {}
super(SQSBackend, self).__init__()
def reset(self):
region_name = self.region_name
self.__dict__ = {}
self.__init__(region_name)
def create_queue(self, name, visibility_timeout, wait_time_seconds):
queue = self.queues.get(name)
if queue is None:
queue = Queue(name, visibility_timeout, wait_time_seconds)
queue = Queue(name, visibility_timeout, wait_time_seconds, self.region_name)
self.queues[name] = queue
return queue
@ -314,4 +325,4 @@ class SQSBackend(BaseBackend):
sqs_backends = {}
for region in boto.sqs.regions():
sqs_backends[region.name] = SQSBackend()
sqs_backends[region.name] = SQSBackend(region.name)