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 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 _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 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 set_timeout(self, visibility_timeout, callback=None): return self.set_attribute('VisibilityTimeout', visibility_timeout, transform( self._on_timeout_set, callback, ))