summaryrefslogtreecommitdiff
path: root/sleekxmpp/plugins/google/gmail/notifications.py
diff options
context:
space:
mode:
authorLance Stout <lancestout@gmail.com>2013-02-10 11:55:49 -0800
committerLance Stout <lancestout@gmail.com>2013-02-10 11:55:49 -0800
commit88c7c299549cbe1aeeecae9ca7f7ed18eb74a327 (patch)
tree72531f9006448a225cb7e9f4b4ced8c626f62fce /sleekxmpp/plugins/google/gmail/notifications.py
parentd4dde89ea65ca1e03ee45e5b2b50a09747271dde (diff)
downloadslixmpp-88c7c299549cbe1aeeecae9ca7f7ed18eb74a327.tar.gz
slixmpp-88c7c299549cbe1aeeecae9ca7f7ed18eb74a327.tar.bz2
slixmpp-88c7c299549cbe1aeeecae9ca7f7ed18eb74a327.tar.xz
slixmpp-88c7c299549cbe1aeeecae9ca7f7ed18eb74a327.zip
Ensure gmail last result time and tid are always updated.
Diffstat (limited to 'sleekxmpp/plugins/google/gmail/notifications.py')
-rw-r--r--sleekxmpp/plugins/google/gmail/notifications.py33
1 files changed, 20 insertions, 13 deletions
diff --git a/sleekxmpp/plugins/google/gmail/notifications.py b/sleekxmpp/plugins/google/gmail/notifications.py
index e1f9cf38..509a95fd 100644
--- a/sleekxmpp/plugins/google/gmail/notifications.py
+++ b/sleekxmpp/plugins/google/gmail/notifications.py
@@ -60,19 +60,26 @@ class Gmail(BasePlugin):
last_time = self._last_result_time
last_tid = self._last_result_tid
- def check_callback(data):
- self._last_result_time = data['gmail_messages']['result_time']
- threads = data['gmail_messages']['threads']
- if threads:
- self._last_result_tid = threads[0]['tid']
- if callback:
- callback(data)
-
- return self.search(newer_time=last_time,
- newer_tid=last_tid,
- block=block,
- timeout=timeout,
- callback=check_callback)
+ if not block:
+ callback = lambda iq: self._update_last_results(iq, callback)
+
+ resp = self.search(newer_time=last_time,
+ newer_tid=last_tid,
+ block=block,
+ timeout=timeout,
+ callback=callback)
+
+ if block:
+ self._update_last_results(resp)
+ return resp
+
+ def _update_last_results(self, iq, callback=None):
+ self._last_result_time = data['gmail_messages']['result_time']
+ threads = data['gmail_messages']['threads']
+ if threads:
+ self._last_result_tid = threads[0]['tid']
+ if callback:
+ callback(iq)
def search(self, query=None, newer_time=None, newer_tid=None, block=True,
timeout=None, callback=None):