Some AUM fixups

This commit is contained in:
Roger Philibert 2011-04-13 12:43:26 +02:00 committed by Romain Bignon
commit 7a0aad5e8f
3 changed files with 14 additions and 4 deletions

View file

@ -22,6 +22,7 @@ from __future__ import with_statement
from weboob.tools.browser import BrowserUnavailable
from weboob.capabilities.dating import Optimization
from weboob.capabilities.contact import QueryError
from weboob.tools.log import getLogger
@ -56,6 +57,9 @@ class QueriesQueue(Optimization):
return self.check_cron is not None
def enqueue_query(self, id, priority=999):
id_queue = [_id[1] for _id in self.queue]
if int(id) in id_queue:
raise QueryError('This id is already queued')
self.queue.append((int(priority), int(id)))
self.save()
# Try to flush queue to send it now.