data.queue: simplify put method

This commit is contained in:
Jimmy Zelinskie 2016-07-14 14:52:59 -04:00
parent 3d558f6090
commit 609f4fccd8

View file

@ -135,23 +135,17 @@ class WorkQueue(object):
Put an item, if it shouldn't be processed for some number of seconds, Put an item, if it shouldn't be processed for some number of seconds,
specify that amount as available_after. Returns the ID of the queue item added. specify that amount as available_after. Returns the ID of the queue item added.
""" """
item = QueueItem.create(
params = { queue_name=self._canonical_name([self._queue_name] + canonical_name_list),
'queue_name': self._canonical_name([self._queue_name] + canonical_name_list), body=message,
'body': message, retries_remaining=retries_remaining,
'retries_remaining': retries_remaining, available_after=datetime.utcnow() + timedelta(seconds=available_after or 0),
} )
available_date = datetime.utcnow() + timedelta(seconds=available_after or 0)
params['available_after'] = available_date
with self._transaction_factory(db):
r = str(QueueItem.create(**params).id)
if self._metric_queue: if self._metric_queue:
self._metric_queue.put_deprecated('Added', 1, dimensions={'queue': self._queue_name}) self._metric_queue.put_deprecated('Added', 1, dimensions={'queue': self._queue_name})
return r return str(item.id)
def get(self, processing_time=300, ordering_required=False): def get(self, processing_time=300, ordering_required=False):
""" """