def set_timeout(self, visibility_timeout, callback=None):
return self.set_attribute(
'VisibilityTimeout', visibility_timeout,
transform(
self._on_timeout_set, callback,
)
)
def write(self, message, delay_seconds=None, callback=None):
return self.connection.send_message(
self,
message.get_body_encoded(),
delay_seconds,
callback=transform(self._on_message_sent, callback, message),
)
def read(self, visibility_timeout=None, wait_time_seconds=None,
callback=None):
return self.get_messages(
1, visibility_timeout,
wait_time_seconds=wait_time_seconds,
callback=transform(list_first, callback),
)
def count(self, page_size=10, vtimeout=10, callback=None,
_attr='ApproximateNumberOfMessages'):
return self.get_attributes(
_attr, callback=transform(
self._coerce_field_value, callback, _attr, int,
),
)
def get_status(self, action, params,
path='/', parent=None, verb='GET', callback=None):
return self.make_request(
action, params, path, verb,
callback=transform(
self._on_status_ready, callback, parent or self,
),
)
def _get_async(self, queue, count=1, callback=None):
q = self._new_queue(queue)
return self._get_from_sqs(
q,
count=count,
connection=self.asynsqs,
callback=transform(self._on_messages_ready, callback, q, queue),
)
def get_timeout(self, callback=None, _attr='VisibilityTimeout'):
return self.get_attributes(
_attr,
transform(
self._coerce_field_value,
callback,
_attr,
int,
),
)
def read(self,
visibility_timeout=None,
wait_time_seconds=None,
callback=None):
return self.get_messages(
1,
visibility_timeout,
wait_time_seconds=wait_time_seconds,
callback=transform(list_first, callback),
)
def test_transform(self):
callback = Mock()
def filter_key_value(key, filter_, mapping):
return filter_(mapping[key])
x = transform(filter_key_value, promise(callback), 'Value', int)
x({'Value': 303})
callback.assert_called_with(303)
with self.assertRaises(KeyError):
x({})
def count(self,
page_size=10,
vtimeout=10,
callback=None,
_attr='ApproximateNumberOfMessages'):
return self.get_attributes(
_attr,
callback=transform(
self._coerce_field_value,
callback,
_attr,
int,
),
)
def get_status(self,
action,
params,
path='/',
parent=None,
verb='GET',
callback=None):
return self.make_request(
action,
params,
path,
verb,
callback=transform(
self._on_status_ready,
callback,
parent or self,
),
)
def _get_async(self, queue, count=1, callback=None):
q = self._new_queue(queue)
return self._get_from_mqs(
q, count=count, mqs_client=self.mqs_client,
callback=transform(self._on_messages_ready, callback, q, queue),
)
def getresponse(self, callback=None):
request = self.getrequest()
request.then(transform(self.Response, callback))
return self.http_client.add_request(request)
def getresponse(self, callback=None):
request = self.getrequest()
request.then(transform(self.Response, callback))
return self.http_client.add_request(request)
def write(self, message, delay_seconds=None, callback=None):
return self.connection.send_message(
self, message.get_body_encoded(), delay_seconds,
callback=transform(self._on_message_sent, callback, message),
)
def get_queue(self, queue_name, callback=None):
# TODO Does not support owner_acct_id argument
return self.get_all_queues(
queue_name,
transform(self._on_queue_ready, callback, queue_name),
)
def get_timeout(self, callback=None, _attr='VisibilityTimeout'):
return self.get_attributes(
_attr, transform(
self._coerce_field_value, callback, _attr, int,
),
)
def get_queue(self, queue_name, callback=None):
# TODO Does not support owner_acct_id argument
return self.get_all_queues(
queue_name,
transform(self._on_queue_ready, callback, queue_name),
)
def _get_async(self, queue, count=1, callback=None):
q = self._new_queue(queue)
return self._get_from_sqs(
q, count=count, connection=self.asynsqs,
callback=transform(self._on_messages_ready, callback, q, queue),
)
def set_timeout(self, visibility_timeout, callback=None):
return self.set_attribute('VisibilityTimeout', visibility_timeout,
transform(
self._on_timeout_set,
callback,
))
