nojhan/weboob-devel
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

enqueue new message when sending one

Browse source
This commit is contained in:
Romain Bignon 2010-05-03 19:15:50 +02:00
commit 5fb975a3e1
1 changed files with 33 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

40
weboob/backends/aum/backend.py
View file

@ -20,12 +20,15 @@ Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
from __future__ import with_statement from __future__ import with_statement
from datetime import datetime
from weboob.backend import BaseBackend from weboob.backend import BaseBackend
from weboob.capabilities.messages import ICapMessages, ICapMessagesReply from weboob.capabilities.messages import ICapMessages, ICapMessagesReply, Message
from weboob.capabilities.dating import ICapDating from weboob.capabilities.dating import ICapDating
from weboob.tools.browser import BrowserUnavailable from weboob.tools.browser import BrowserUnavailable
from .browser import AdopteUnMec from .browser import AdopteUnMec
from .exceptions import AdopteCantPostMail
from .optim.profiles_walker import ProfilesWalker from .optim.profiles_walker import ProfilesWalker
class AuMBackend(BaseBackend, ICapMessages, ICapMessagesReply, ICapDating): class AuMBackend(BaseBackend, ICapMessages, ICapMessagesReply, ICapDating):
@ -43,6 +46,7 @@ class AuMBackend(BaseBackend, ICapMessages, ICapMessagesReply, ICapDating):
# Private # Private
_browser = None _browser = None
_profiles_walker = None _profiles_walker = None
_queue_messages = None
def __getattr__(self, name): def __getattr__(self, name):
if name == 'browser': if name == 'browser':
@ -50,7 +54,7 @@ class AuMBackend(BaseBackend, ICapMessages, ICapMessagesReply, ICapDating):
self._browser = AdopteUnMec(self.config['username'], self.config['password']) self._browser = AdopteUnMec(self.config['username'], self.config['password'])
return self._browser return self._browser
if name == 'queue_messages': if name == 'queue_messages':
if not hasattr(self, '_queue_messages'): if self._queue_messages is None:
self._queue_messages = [] self._queue_messages = []
return self._queue_messages return self._queue_messages
raise AttributeError, name raise AttributeError, name
@ -95,10 +99,31 @@ class AuMBackend(BaseBackend, ICapMessages, ICapMessagesReply, ICapDating):
pass pass
def post_reply(self, thread_id, reply_id, title, message): def post_reply(self, thread_id, reply_id, title, message):
for message in self._iter_messages(thread_id, True): # Enqueue current awaiting messages
self.queue_messages.append(message) for mail in self._iter_messages(thread_id, True):
self.queue_messages.append(mail)
with self.browser: with self.browser:
return self.browser.post(thread_id, message) try:
self.browser.post(thread_id, message)
except AdopteCantPostMail, e:
mail = Message(thread_id,
reply_id,
'Unable to send mail to %s' % thread_id,
'AuM',
datetime.now(),
content=u'Unable to send message to %s:\n\t%s\n\n---\n%s' (thread_id, unicode(e), message),
is_html=False)
self.queue_messages.append(mail)
else:
# Enqueue my messages and every next messages.
mails = self.browser.get_thread_mails(thread_id)
my_name = self.browser.get_my_name()
for mail in mails:
self.queue_messages.append(mail)
if mail.get_from() == my_name:
break
def get_profile(self, _id): def get_profile(self, _id):
try: try:
@ -111,5 +136,6 @@ class AuMBackend(BaseBackend, ICapMessages, ICapMessagesReply, ICapDating):
self._profile_walker = ProfilesWalker(self.weboob.scheduler, self.storage, self.browser) self._profile_walker = ProfilesWalker(self.weboob.scheduler, self.storage, self.browser)
def stop_profiles_walker(self): def stop_profiles_walker(self):
self._profiles_walker.stop() if self._profiles_walker:
self._profiles_walker = None self._profiles_walker.stop()
self._profiles_walker = None