1 from __future__ import with_statement
15 from PyQt4 import QtCore
17 from util import qore_utils
18 from util import concurrent
19 from util import misc as misc_utils
24 _moduleLogger = logging.getLogger(__name__)
27 @contextlib.contextmanager
28 def notify_busy(log, message):
29 log.push_busy(message)
36 class _DraftContact(object):
38 def __init__(self, title, description, numbersWithDescriptions):
40 self.description = description
41 self.numbers = numbersWithDescriptions
42 self.selectedNumber = numbersWithDescriptions[0][0]
45 class Draft(QtCore.QObject):
47 sendingMessage = QtCore.pyqtSignal()
48 sentMessage = QtCore.pyqtSignal()
49 calling = QtCore.pyqtSignal()
50 called = QtCore.pyqtSignal()
51 cancelling = QtCore.pyqtSignal()
52 cancelled = QtCore.pyqtSignal()
53 error = QtCore.pyqtSignal(str)
55 recipientsChanged = QtCore.pyqtSignal()
57 def __init__(self, pool, backend, errorLog):
58 QtCore.QObject.__init__(self)
59 self._errorLog = errorLog
62 self._backend = backend
65 assert 0 < len(self._contacts), "No contacts selected"
66 numbers = [misc_utils.make_ugly(contact.selectedNumber) for contact in self._contacts.itervalues()]
67 le = concurrent.AsyncLinearExecution(self._pool, self._send)
68 le.start(numbers, text)
71 assert len(self._contacts) == 1, "Must select 1 and only 1 contact"
72 (contact, ) = self._contacts.itervalues()
73 number = misc_utils.make_ugly(contact.selectedNumber)
74 le = concurrent.AsyncLinearExecution(self._pool, self._call)
78 le = concurrent.AsyncLinearExecution(self._pool, self._cancel)
81 def add_contact(self, contactId, title, description, numbersWithDescriptions):
82 if contactId in self._contacts:
83 _moduleLogger.info("Adding duplicate contact %r" % contactId)
84 # @todo Remove this evil hack to re-popup the dialog
85 self.recipientsChanged.emit()
87 contactDetails = _DraftContact(title, description, numbersWithDescriptions)
88 self._contacts[contactId] = contactDetails
89 self.recipientsChanged.emit()
91 def remove_contact(self, contactId):
92 assert contactId in self._contacts, "Contact missing"
93 del self._contacts[contactId]
94 self.recipientsChanged.emit()
96 def get_contacts(self):
97 return self._contacts.iterkeys()
99 def get_num_contacts(self):
100 return len(self._contacts)
102 def get_title(self, cid):
103 return self._contacts[cid].title
105 def get_description(self, cid):
106 return self._contacts[cid].description
108 def get_numbers(self, cid):
109 return self._contacts[cid].numbers
111 def get_selected_number(self, cid):
112 return self._contacts[cid].selectedNumber
114 def set_selected_number(self, cid, number):
115 # @note I'm lazy, this isn't firing any kind of signal since only one
116 # controller right now and that is the viewer
117 assert number in (nWD[0] for nWD in self._contacts[cid].numbers), "Number not selectable"
118 self._contacts[cid].selectedNumber = number
121 oldContacts = self._contacts
124 self.recipientsChanged.emit()
126 def _send(self, numbers, text):
127 self.sendingMessage.emit()
129 with notify_busy(self._errorLog, "Sending Text"):
131 self._backend[0].send_sms,
135 self.sentMessage.emit()
138 self.error.emit(str(e))
140 def _call(self, number):
143 with notify_busy(self._errorLog, "Calling"):
145 self._backend[0].call,
152 self.error.emit(str(e))
155 self.cancelling.emit()
157 with notify_busy(self._errorLog, "Cancelling"):
159 self._backend[0].cancel,
163 self.cancelled.emit()
165 self.error.emit(str(e))
168 class Session(QtCore.QObject):
170 # @todo Somehow add support for csv contacts
172 stateChange = QtCore.pyqtSignal(str)
173 loggedOut = QtCore.pyqtSignal()
174 loggedIn = QtCore.pyqtSignal()
175 callbackNumberChanged = QtCore.pyqtSignal(str)
177 contactsUpdated = QtCore.pyqtSignal()
178 messagesUpdated = QtCore.pyqtSignal()
179 historyUpdated = QtCore.pyqtSignal()
180 dndStateChange = QtCore.pyqtSignal(bool)
182 error = QtCore.pyqtSignal(str)
184 LOGGEDOUT_STATE = "logged out"
185 LOGGINGIN_STATE = "logging in"
186 LOGGEDIN_STATE = "logged in"
188 _OLDEST_COMPATIBLE_FORMAT_VERSION = misc_utils.parse_version("1.1.90")
193 def __init__(self, errorLog, cachePath = None):
194 QtCore.QObject.__init__(self)
195 self._errorLog = errorLog
196 self._pool = qore_utils.AsyncPool()
198 self._loggedInTime = self._LOGGEDOUT_TIME
200 self._cachePath = cachePath
201 self._username = None
202 self._draft = Draft(self._pool, self._backend, self._errorLog)
205 self._contactUpdateTime = datetime.datetime(1971, 1, 1)
207 self._messageUpdateTime = datetime.datetime(1971, 1, 1)
209 self._historyUpdateTime = datetime.datetime(1971, 1, 1)
216 self._LOGGEDOUT_TIME: self.LOGGEDOUT_STATE,
217 self._LOGGINGIN_TIME: self.LOGGINGIN_STATE,
218 }.get(self._loggedInTime, self.LOGGEDIN_STATE)
224 def login(self, username, password):
225 assert self.state == self.LOGGEDOUT_STATE, "Can only log-in when logged out (currently %s" % self.state
226 assert username != "", "No username specified"
227 if self._cachePath is not None:
228 cookiePath = os.path.join(self._cachePath, "%s.cookies" % username)
232 if self._username != username or not self._backend:
233 from backends import gv_backend
235 self._backend[0:0] = [gv_backend.GVDialer(cookiePath)]
238 le = concurrent.AsyncLinearExecution(self._pool, self._login)
239 le.start(username, password)
242 assert self.state != self.LOGGEDOUT_STATE, "Can only logout if logged in (currently %s" % self.state
244 self._loggedInTime = self._LOGGEDOUT_TIME
245 self._backend[0].persist()
246 self._save_to_cache()
249 assert self.state == self.LOGGEDOUT_STATE, "Can only clear when logged out (currently %s" % self.state
250 self._backend[0].logout()
255 def logout_and_clear(self):
256 assert self.state != self.LOGGEDOUT_STATE, "Can only logout if logged in (currently %s" % self.state
258 self._loggedInTime = self._LOGGEDOUT_TIME
261 def update_contacts(self, force = True):
262 if not force and self._contacts:
264 le = concurrent.AsyncLinearExecution(self._pool, self._update_contacts)
265 self._perform_op_while_loggedin(le)
267 def get_contacts(self):
268 return self._contacts
270 def get_when_contacts_updated(self):
271 return self._contactUpdateTime
273 def update_messages(self, force = True):
274 if not force and self._messages:
276 le = concurrent.AsyncLinearExecution(self._pool, self._update_messages)
277 self._perform_op_while_loggedin(le)
279 def get_messages(self):
280 return self._messages
282 def get_when_messages_updated(self):
283 return self._messageUpdateTime
285 def update_history(self, force = True):
286 if not force and self._history:
288 le = concurrent.AsyncLinearExecution(self._pool, self._update_history)
289 self._perform_op_while_loggedin(le)
291 def get_history(self):
294 def get_when_history_updated(self):
295 return self._historyUpdateTime
297 def update_dnd(self):
298 le = concurrent.AsyncLinearExecution(self._pool, self._update_dnd)
299 self._perform_op_while_loggedin(le)
301 def set_dnd(self, dnd):
302 le = concurrent.AsyncLinearExecution(self._pool, self._set_dnd)
305 def _set_dnd(self, dnd):
306 # I'm paranoid about our state geting out of sync so we set no matter
307 # what but act as if we have the cannonical state
308 assert self.state == self.LOGGEDIN_STATE, "DND requires being logged in (currently %s" % self.state
311 with notify_busy(self._errorLog, "Setting DND Status"):
313 self._backend[0].set_dnd,
318 self.error.emit(str(e))
321 if oldDnd != self._dnd:
322 self.dndStateChange.emit(self._dnd)
327 def get_account_number(self):
328 return self._backend[0].get_account_number()
330 def get_callback_numbers(self):
331 # @todo Remove evilness (might call is_authed which can block)
332 return self._backend[0].get_callback_numbers()
334 def get_callback_number(self):
335 return self._callback
337 def set_callback_number(self, callback):
338 le = concurrent.AsyncLinearExecution(self._pool, self._set_callback_number)
341 def _set_callback_number(self, callback):
342 # I'm paranoid about our state geting out of sync so we set no matter
343 # what but act as if we have the cannonical state
344 assert self.state == self.LOGGEDIN_STATE, "Callbacks configurable only when logged in (currently %s" % self.state
345 oldCallback = self._callback
347 with notify_busy(self._errorLog, "Setting Callback"):
349 self._backend[0].set_callback_number,
354 self.error.emit(str(e))
356 self._callback = callback
357 if oldCallback != self._callback:
358 self.callbackNumberChanged.emit(self._callback)
360 def _login(self, username, password):
361 with notify_busy(self._errorLog, "Logging In"):
362 self._loggedInTime = self._LOGGINGIN_TIME
363 self.stateChange.emit(self.LOGGINGIN_STATE)
364 finalState = self.LOGGEDOUT_STATE
367 if not isLoggedIn and self._backend[0].is_quick_login_possible():
369 self._backend[0].is_authed,
374 _moduleLogger.info("Logged in through cookies")
376 # Force a clearing of the cookies
378 self._backend[0].logout,
385 self._backend[0].login,
386 (username, password),
390 _moduleLogger.info("Logged in through credentials")
393 self._loggedInTime = int(time.time())
394 oldUsername = self._username
395 self._username = username
396 finalState = self.LOGGEDIN_STATE
397 if oldUsername != self._username:
398 needOps = not self._load()
405 loginOps = self._loginOps[:]
408 del self._loginOps[:]
409 for asyncOp in loginOps:
412 self._loggedInTime = self._LOGGEDOUT_TIME
413 self.error.emit(str(e))
415 self.stateChange.emit(finalState)
416 if isLoggedIn and self._callback:
417 self.set_callback_number(self._callback)
420 updateContacts = len(self._contacts) != 0
421 updateMessages = len(self._messages) != 0
422 updateHistory = len(self._history) != 0
424 oldCallback = self._callback
432 loadedFromCache = self._load_from_cache()
434 updateContacts = True
435 updateMessages = True
439 self.contactsUpdated.emit()
441 self.messagesUpdated.emit()
443 self.historyUpdated.emit()
444 if oldDnd != self._dnd:
445 self.dndStateChange.emit(self._dnd)
446 if oldCallback != self._callback:
447 self.callbackNumberChanged.emit(self._callback)
449 return loadedFromCache
451 def _load_from_cache(self):
452 if self._cachePath is None:
454 cachePath = os.path.join(self._cachePath, "%s.cache" % self._username)
457 with open(cachePath, "rb") as f:
458 dumpedData = pickle.load(f)
459 except (pickle.PickleError, IOError, EOFError, ValueError):
460 _moduleLogger.exception("Pickle fun loading")
463 _moduleLogger.exception("Weirdness loading")
468 contacts, contactUpdateTime,
469 messages, messageUpdateTime,
470 history, historyUpdateTime,
474 if misc_utils.compare_versions(
475 self._OLDEST_COMPATIBLE_FORMAT_VERSION,
476 misc_utils.parse_version(version),
478 _moduleLogger.info("Loaded cache")
479 self._contacts = contacts
480 self._contactUpdateTime = contactUpdateTime
481 self._messages = messages
482 self._messageUpdateTime = messageUpdateTime
483 self._history = history
484 self._historyUpdateTime = historyUpdateTime
486 self._callback = callback
490 "Skipping cache due to version mismatch (%s-%s)" % (
496 def _save_to_cache(self):
497 _moduleLogger.info("Saving cache")
498 if self._cachePath is None:
500 cachePath = os.path.join(self._cachePath, "%s.cache" % self._username)
504 constants.__version__, constants.__build__,
505 self._contacts, self._contactUpdateTime,
506 self._messages, self._messageUpdateTime,
507 self._history, self._historyUpdateTime,
508 self._dnd, self._callback
510 with open(cachePath, "wb") as f:
511 pickle.dump(dataToDump, f, pickle.HIGHEST_PROTOCOL)
512 _moduleLogger.info("Cache saved")
513 except (pickle.PickleError, IOError):
514 _moduleLogger.exception("While saving")
516 def _clear_cache(self):
517 updateContacts = len(self._contacts) != 0
518 updateMessages = len(self._messages) != 0
519 updateHistory = len(self._history) != 0
521 oldCallback = self._callback
524 self._contactUpdateTime = datetime.datetime(1, 1, 1)
526 self._messageUpdateTime = datetime.datetime(1, 1, 1)
528 self._historyUpdateTime = datetime.datetime(1, 1, 1)
533 self.contactsUpdated.emit()
535 self.messagesUpdated.emit()
537 self.historyUpdated.emit()
538 if oldDnd != self._dnd:
539 self.dndStateChange.emit(self._dnd)
540 if oldCallback != self._callback:
541 self.callbackNumberChanged.emit(self._callback)
543 self._save_to_cache()
545 def _update_contacts(self):
547 with notify_busy(self._errorLog, "Updating Contacts"):
548 self._contacts = yield (
549 self._backend[0].get_contacts,
554 self.error.emit(str(e))
556 self._contactUpdateTime = datetime.datetime.now()
557 self.contactsUpdated.emit()
559 def _update_messages(self):
561 with notify_busy(self._errorLog, "Updating Messages"):
562 self._messages = yield (
563 self._backend[0].get_messages,
568 self.error.emit(str(e))
570 self._messageUpdateTime = datetime.datetime.now()
571 self.messagesUpdated.emit()
573 def _update_history(self):
575 with notify_busy(self._errorLog, "Updating History"):
576 self._history = yield (
577 self._backend[0].get_recent,
582 self.error.emit(str(e))
584 self._historyUpdateTime = datetime.datetime.now()
585 self.historyUpdated.emit()
587 def _update_dnd(self):
591 self._backend[0].is_dnd,
596 self.error.emit(str(e))
598 if oldDnd != self._dnd:
599 self.dndStateChange(self._dnd)
601 def _perform_op_while_loggedin(self, op):
602 if self.state == self.LOGGEDIN_STATE:
605 self._push_login_op(op)
607 def _push_login_op(self, asyncOp):
608 assert self.state != self.LOGGEDIN_STATE, "Can only queue work when logged out"
609 if asyncOp in self._loginOps:
610 _moduleLogger.info("Skipping queueing duplicate op: %r" % asyncOp)
612 self._loginOps.append(asyncOp)