no need for max arg on get_update_queue anymore

This commit is contained in:
James Turk 2011-08-04 15:29:59 -04:00
parent 6cbbf47c4a
commit d9d63e1660

View File

@ -138,24 +138,17 @@ class Client(object):
return self.fs.get_version(url, n)
def get_update_queue(self, max=0):
def get_update_queue(self):
# results are always sorted by random to avoid piling on single server
# first we try to update anything that we've never retrieved
new = self.db.tracked.find({'next_update':
{'$exists': False}}).sort('_random')
if max:
new = new.limit(max)
queue = list(new)
# pull the rest from those for which next_update is in the past
next = self.db.tracked.find({'next_update':
{'$lt': datetime.datetime.utcnow()}}).sort('_random')
if max:
max -= len(queue)
next = next.limit(max)
queue.extend(next)
return queue