3 from __future__ import with_statement
4 from __future__ import division
9 from PyQt4 import QtGui
10 from PyQt4 import QtCore
12 from util import qtpie
13 from util import qui_utils
14 from util import misc as misc_utils
16 import backends.null_backend as null_backend
17 import backends.file_backend as file_backend
20 _moduleLogger = logging.getLogger(__name__)
23 class Dialpad(object):
25 def __init__(self, app, session, errorLog):
27 self._session = session
28 self._errorLog = errorLog
30 self._plus = self._generate_key_button("+", "")
31 self._entry = QtGui.QLineEdit()
33 backAction = QtGui.QAction(None)
34 backAction.setText("Back")
35 backAction.triggered.connect(self._on_backspace)
36 backPieItem = qtpie.QActionPieItem(backAction)
37 clearAction = QtGui.QAction(None)
38 clearAction.setText("Clear")
39 clearAction.triggered.connect(self._on_clear_text)
40 clearPieItem = qtpie.QActionPieItem(clearAction)
41 self._back = qtpie.QPieButton(backPieItem)
42 self._back.set_center(backPieItem)
43 self._back.insertItem(qtpie.PieFiling.NULL_CENTER)
44 self._back.insertItem(clearPieItem)
45 self._back.insertItem(qtpie.PieFiling.NULL_CENTER)
46 self._back.insertItem(qtpie.PieFiling.NULL_CENTER)
48 self._entryLayout = QtGui.QHBoxLayout()
49 self._entryLayout.addWidget(self._plus, 0, QtCore.Qt.AlignCenter)
50 self._entryLayout.addWidget(self._entry, 10)
51 self._entryLayout.addWidget(self._back, 0, QtCore.Qt.AlignCenter)
53 self._smsButton = QtGui.QPushButton("SMS")
54 self._smsButton.clicked.connect(self._on_sms_clicked)
55 self._callButton = QtGui.QPushButton("Call")
56 self._callButton.clicked.connect(self._on_call_clicked)
58 self._padLayout = QtGui.QGridLayout()
59 rows = [0, 0, 0, 1, 1, 1, 2, 2, 2]
60 columns = [0, 1, 2] * 3
72 for (num, letters), (row, column) in zip(keys, zip(rows, columns)):
73 self._padLayout.addWidget(
74 self._generate_key_button(num, letters), row, column, QtCore.Qt.AlignCenter
76 self._padLayout.addWidget(self._smsButton, 3, 0)
77 self._padLayout.addWidget(
78 self._generate_key_button("0", ""), 3, 1, QtCore.Qt.AlignCenter
80 self._padLayout.addWidget(self._callButton, 3, 2)
82 self._layout = QtGui.QVBoxLayout()
83 self._layout.addLayout(self._entryLayout)
84 self._layout.addLayout(self._padLayout)
85 self._widget = QtGui.QWidget()
86 self._widget.setLayout(self._layout)
93 self._smsButton.setEnabled(True)
94 self._callButton.setEnabled(True)
97 self._smsButton.setEnabled(False)
98 self._callButton.setEnabled(False)
103 def refresh(self, force = True):
106 def _generate_key_button(self, center, letters):
107 centerPieItem = self._generate_button_slice(center)
108 button = qtpie.QPieButton(centerPieItem)
109 button.set_center(centerPieItem)
111 if len(letters) == 0:
113 pieItem = qtpie.PieFiling.NULL_CENTER
114 button.insertItem(pieItem)
115 elif len(letters) in [3, 4]:
116 for i in xrange(6 - len(letters)):
117 pieItem = qtpie.PieFiling.NULL_CENTER
118 button.insertItem(pieItem)
120 for letter in letters:
121 pieItem = self._generate_button_slice(letter)
122 button.insertItem(pieItem)
125 pieItem = qtpie.PieFiling.NULL_CENTER
126 button.insertItem(pieItem)
128 raise NotImplementedError("Cannot handle %r" % letters)
131 def _generate_button_slice(self, letter):
132 action = QtGui.QAction(None)
133 action.setText(letter)
134 action.triggered.connect(lambda: self._on_keypress(letter))
135 pieItem = qtpie.QActionPieItem(action)
138 @misc_utils.log_exception(_moduleLogger)
139 def _on_keypress(self, key):
140 self._entry.insert(key)
142 @misc_utils.log_exception(_moduleLogger)
143 def _on_backspace(self, toggled = False):
144 self._entry.backspace()
146 @misc_utils.log_exception(_moduleLogger)
147 def _on_clear_text(self, toggled = False):
151 @QtCore.pyqtSlot(bool)
152 @misc_utils.log_exception(_moduleLogger)
153 def _on_sms_clicked(self, checked = False):
154 number = str(self._entry.text())
160 numbersWithDescriptions = [(number, "")]
161 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
164 @QtCore.pyqtSlot(bool)
165 @misc_utils.log_exception(_moduleLogger)
166 def _on_call_clicked(self, checked = False):
167 number = str(self._entry.text())
173 numbersWithDescriptions = [(number, "")]
174 self._session.draft.clear()
175 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
176 self._session.draft.call()
179 class History(object):
187 HISTORY_ITEM_TYPES = ["Received", "Missed", "Placed", "All"]
188 HISTORY_COLUMNS = ["When", "What", "Number", "From"]
189 assert len(HISTORY_COLUMNS) == MAX_IDX
191 def __init__(self, app, session, errorLog):
192 self._selectedFilter = self.HISTORY_ITEM_TYPES[-1]
194 self._session = session
195 self._session.historyUpdated.connect(self._on_history_updated)
196 self._errorLog = errorLog
198 self._typeSelection = QtGui.QComboBox()
199 self._typeSelection.addItems(self.HISTORY_ITEM_TYPES)
200 self._typeSelection.setCurrentIndex(
201 self.HISTORY_ITEM_TYPES.index(self._selectedFilter)
203 self._typeSelection.currentIndexChanged[str].connect(self._on_filter_changed)
205 self._itemStore = QtGui.QStandardItemModel()
206 self._itemStore.setHorizontalHeaderLabels(self.HISTORY_COLUMNS)
208 self._itemView = QtGui.QTreeView()
209 self._itemView.setModel(self._itemStore)
210 self._itemView.setUniformRowHeights(True)
211 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
212 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
213 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
214 self._itemView.setHeaderHidden(True)
215 self._itemView.activated.connect(self._on_row_activated)
217 self._layout = QtGui.QVBoxLayout()
218 self._layout.addWidget(self._typeSelection)
219 self._layout.addWidget(self._itemView)
220 self._widget = QtGui.QWidget()
221 self._widget.setLayout(self._layout)
223 self._populate_items()
230 self._itemView.setEnabled(True)
233 self._itemView.setEnabled(False)
236 self._itemView.clear()
238 def refresh(self, force=True):
239 self._session.update_history(force)
241 def _populate_items(self):
242 self._itemStore.clear()
243 history = self._session.get_history()
244 history.sort(key=lambda item: item["time"], reverse=True)
245 for event in history:
246 if self._selectedFilter in [self.HISTORY_ITEM_TYPES[-1], event["action"]]:
247 relTime = abbrev_relative_date(event["relTime"])
248 action = event["action"]
249 number = event["number"]
250 prettyNumber = make_pretty(number)
252 if not name or name == number:
253 name = event["location"]
257 timeItem = QtGui.QStandardItem(relTime)
258 actionItem = QtGui.QStandardItem(action)
259 numberItem = QtGui.QStandardItem(prettyNumber)
260 nameItem = QtGui.QStandardItem(name)
261 row = timeItem, actionItem, numberItem, nameItem
263 item.setEditable(False)
264 item.setCheckable(False)
265 if item is not nameItem:
266 itemFont = item.font()
267 itemFont.setPointSize(max(itemFont.pointSize() - 3, 5))
268 item.setFont(itemFont)
269 numberItem.setData(event)
270 self._itemStore.appendRow(row)
272 @QtCore.pyqtSlot(str)
273 @misc_utils.log_exception(_moduleLogger)
274 def _on_filter_changed(self, newItem):
275 self._selectedFilter = str(newItem)
276 self._populate_items()
279 @misc_utils.log_exception(_moduleLogger)
280 def _on_history_updated(self):
281 self._populate_items()
283 @QtCore.pyqtSlot(QtCore.QModelIndex)
284 @misc_utils.log_exception(_moduleLogger)
285 def _on_row_activated(self, index):
286 rowIndex = index.row()
287 item = self._itemStore.item(rowIndex, self.NUMBER_IDX)
288 contactDetails = item.data().toPyObject()
290 title = str(self._itemStore.item(rowIndex, self.FROM_IDX).text())
291 number = str(contactDetails[QtCore.QString("number")])
292 contactId = number # ids don't seem too unique so using numbers
295 # @bug doesn't seem to print multiple entries
296 for i in xrange(self._itemStore.rowCount()):
297 iItem = self._itemStore.item(i, self.NUMBER_IDX)
298 iContactDetails = iItem.data().toPyObject()
299 iNumber = str(iContactDetails[QtCore.QString("number")])
300 if number != iNumber:
302 relTime = abbrev_relative_date(iContactDetails[QtCore.QString("relTime")])
303 action = str(iContactDetails[QtCore.QString("action")])
304 number = str(iContactDetails[QtCore.QString("number")])
305 prettyNumber = make_pretty(number)
306 rowItems = relTime, action, prettyNumber
307 descriptionRows.append("<tr><td>%s</td></tr>" % "</td><td>".join(rowItems))
308 description = "<table>%s</table>" % "".join(descriptionRows)
309 numbersWithDescriptions = [(str(contactDetails[QtCore.QString("number")]), "")]
310 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
313 class Messages(object):
316 VOICEMAIL_MESSAGES = "Voicemail"
317 TEXT_MESSAGES = "SMS"
318 ALL_TYPES = "All Messages"
319 MESSAGE_TYPES = [NO_MESSAGES, VOICEMAIL_MESSAGES, TEXT_MESSAGES, ALL_TYPES]
321 UNREAD_STATUS = "Unread"
322 UNARCHIVED_STATUS = "Inbox"
324 MESSAGE_STATUSES = [UNREAD_STATUS, UNARCHIVED_STATUS, ALL_STATUS]
326 _MIN_MESSAGES_SHOWN = 4
328 def __init__(self, app, session, errorLog):
329 self._selectedTypeFilter = self.ALL_TYPES
330 self._selectedStatusFilter = self.ALL_STATUS
332 self._session = session
333 self._session.messagesUpdated.connect(self._on_messages_updated)
334 self._errorLog = errorLog
336 self._typeSelection = QtGui.QComboBox()
337 self._typeSelection.addItems(self.MESSAGE_TYPES)
338 self._typeSelection.setCurrentIndex(
339 self.MESSAGE_TYPES.index(self._selectedTypeFilter)
341 self._typeSelection.currentIndexChanged[str].connect(self._on_type_filter_changed)
343 self._statusSelection = QtGui.QComboBox()
344 self._statusSelection.addItems(self.MESSAGE_STATUSES)
345 self._statusSelection.setCurrentIndex(
346 self.MESSAGE_STATUSES.index(self._selectedStatusFilter)
348 self._statusSelection.currentIndexChanged[str].connect(self._on_status_filter_changed)
350 self._selectionLayout = QtGui.QHBoxLayout()
351 self._selectionLayout.addWidget(self._typeSelection)
352 self._selectionLayout.addWidget(self._statusSelection)
354 self._itemStore = QtGui.QStandardItemModel()
355 self._itemStore.setHorizontalHeaderLabels(["Messages"])
357 self._htmlDelegate = qui_utils.QHtmlDelegate()
358 self._itemView = QtGui.QTreeView()
359 self._itemView.setModel(self._itemStore)
360 self._itemView.setUniformRowHeights(False)
361 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
362 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
363 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
364 self._itemView.setHeaderHidden(True)
365 self._itemView.setItemDelegate(self._htmlDelegate)
366 self._itemView.activated.connect(self._on_row_activated)
368 self._layout = QtGui.QVBoxLayout()
369 self._layout.addLayout(self._selectionLayout)
370 self._layout.addWidget(self._itemView)
371 self._widget = QtGui.QWidget()
372 self._widget.setLayout(self._layout)
374 self._populate_items()
381 self._itemView.setEnabled(True)
384 self._itemView.setEnabled(False)
387 self._itemView.clear()
389 def refresh(self, force=True):
390 self._session.update_messages(force)
392 def _populate_items(self):
393 self._itemStore.clear()
394 rawMessages = self._session.get_messages()
395 rawMessages.sort(key=lambda item: item["time"], reverse=True)
396 for item in rawMessages:
397 isUnarchived = not item["isArchived"]
398 isUnread = not item["isRead"]
400 self.UNREAD_STATUS: isUnarchived and isUnread,
401 self.UNARCHIVED_STATUS: isUnarchived,
402 self.ALL_STATUS: True,
403 }[self._selectedStatusFilter]
405 visibleType = self._selectedTypeFilter in [item["type"], self.ALL_TYPES]
406 if visibleType and visibleStatus:
407 relTime = abbrev_relative_date(item["relTime"])
408 number = item["number"]
409 prettyNumber = make_pretty(number)
411 if not name or name == number:
412 name = item["location"]
416 messageParts = list(item["messageParts"])
417 if len(messageParts) == 0:
418 messages = ("No Transcription", )
419 elif len(messageParts) == 1:
420 if messageParts[0][1]:
421 messages = (messageParts[0][1], )
423 messages = ("No Transcription", )
426 "<b>%s</b>: %s" % (messagePart[0], messagePart[1])
427 for messagePart in messageParts
430 firstMessage = "<b>%s - %s</b> <i>(%s)</i>" % (name, prettyNumber, relTime)
432 expandedMessages = [firstMessage]
433 expandedMessages.extend(messages)
434 if (self._MIN_MESSAGES_SHOWN + 1) < len(messages):
435 secondMessage = "<i>%d Messages Hidden...</i>" % (len(messages) - self._MIN_MESSAGES_SHOWN, )
436 collapsedMessages = [firstMessage, secondMessage]
437 collapsedMessages.extend(messages[-(self._MIN_MESSAGES_SHOWN+0):])
439 collapsedMessages = expandedMessages
441 item = dict(item.iteritems())
442 item["collapsedMessages"] = "<br/>\n".join(collapsedMessages)
443 item["expandedMessages"] = "<br/>\n".join(expandedMessages)
445 messageItem = QtGui.QStandardItem(item["collapsedMessages"])
446 # @bug Not showing all of a message
447 messageItem.setData(item)
448 messageItem.setEditable(False)
449 messageItem.setCheckable(False)
450 row = (messageItem, )
451 self._itemStore.appendRow(row)
453 @QtCore.pyqtSlot(str)
454 @misc_utils.log_exception(_moduleLogger)
455 def _on_type_filter_changed(self, newItem):
456 self._selectedTypeFilter = str(newItem)
457 self._populate_items()
459 @QtCore.pyqtSlot(str)
460 @misc_utils.log_exception(_moduleLogger)
461 def _on_status_filter_changed(self, newItem):
462 self._selectedStatusFilter = str(newItem)
463 self._populate_items()
466 @misc_utils.log_exception(_moduleLogger)
467 def _on_messages_updated(self):
468 self._populate_items()
470 @QtCore.pyqtSlot(QtCore.QModelIndex)
471 @misc_utils.log_exception(_moduleLogger)
472 def _on_row_activated(self, index):
473 rowIndex = index.row()
474 item = self._itemStore.item(rowIndex, 0)
475 contactDetails = item.data().toPyObject()
477 name = str(contactDetails[QtCore.QString("name")])
478 number = str(contactDetails[QtCore.QString("number")])
479 if not name or name == number:
480 name = str(contactDetails[QtCore.QString("location")])
484 contactId = str(contactDetails[QtCore.QString("id")])
486 description = str(contactDetails[QtCore.QString("expandedMessages")])
487 numbersWithDescriptions = [(number, "")]
488 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
491 class Contacts(object):
493 def __init__(self, app, session, errorLog):
495 self._session = session
496 self._session.contactsUpdated.connect(self._on_contacts_updated)
497 self._errorLog = errorLog
498 self._addressBookFactories = [
499 null_backend.NullAddressBookFactory(),
500 file_backend.FilesystemAddressBookFactory(app.fsContactsPath),
502 self._addressBooks = []
504 self._listSelection = QtGui.QComboBox()
505 self._listSelection.addItems([])
506 self._listSelection.currentIndexChanged[str].connect(self._on_filter_changed)
508 self._itemStore = QtGui.QStandardItemModel()
509 self._itemStore.setHorizontalHeaderLabels(["Contacts"])
511 self._itemView = QtGui.QTreeView()
512 self._itemView.setModel(self._itemStore)
513 self._itemView.setUniformRowHeights(True)
514 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
515 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
516 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
517 self._itemView.setHeaderHidden(True)
518 self._itemView.activated.connect(self._on_row_activated)
520 self._layout = QtGui.QVBoxLayout()
521 self._layout.addWidget(self._listSelection)
522 self._layout.addWidget(self._itemView)
523 self._widget = QtGui.QWidget()
524 self._widget.setLayout(self._layout)
526 self.update_addressbooks()
527 self._populate_items()
534 self._itemView.setEnabled(True)
537 self._itemView.setEnabled(False)
540 self._itemView.clear()
542 def refresh(self, force=True):
543 self._backend.update_contacts(force)
547 return self._addressBooks[self._listSelection.currentIndex()]["book"]
549 def update_addressbooks(self):
550 self._addressBooks = [
551 {"book": book, "name": book.name}
552 for factory in self._addressBookFactories
553 for book in factory.get_addressbooks()
555 self._addressBooks.append(
557 "book": self._session,
558 "name": "Google Voice",
562 currentItem = str(self._listSelection.currentText())
563 if currentItem == "":
566 while 0 < self._listSelection.count():
567 self._listSelection.removeItem(0)
568 bookNames = [book["name"] for book in self._addressBooks]
570 newIndex = bookNames.index(currentItem)
572 # Switch over to None for the user
574 self._itemStore.clear()
575 _moduleLogger.info("Addressbook %r doesn't exist anymore, switching to None" % currentItem)
576 self._listSelection.addItems(bookNames)
577 self._listSelection.setCurrentIndex(newIndex)
579 def _populate_items(self):
580 self._itemStore.clear()
582 contacts = list(self._backend.get_contacts().itervalues())
583 contacts.sort(key=lambda contact: contact["name"].lower())
584 for item in contacts:
586 numbers = item["numbers"]
587 nameItem = QtGui.QStandardItem(name)
588 nameItem.setEditable(False)
589 nameItem.setCheckable(False)
590 nameItem.setData(item)
592 self._itemStore.appendRow(row)
594 @QtCore.pyqtSlot(str)
595 @misc_utils.log_exception(_moduleLogger)
596 def _on_filter_changed(self, newItem):
597 self._populate_items()
600 @misc_utils.log_exception(_moduleLogger)
601 def _on_contacts_updated(self):
602 self._populate_items()
604 @QtCore.pyqtSlot(QtCore.QModelIndex)
605 @misc_utils.log_exception(_moduleLogger)
606 def _on_row_activated(self, index):
607 rowIndex = index.row()
608 item = self._itemStore.item(rowIndex, 0)
609 contactDetails = item.data().toPyObject()
611 name = str(contactDetails[QtCore.QString("name")])
613 name = str(contactDetails[QtCore.QString("location")])
617 contactId = str(contactDetails[QtCore.QString("contactId")])
618 numbers = contactDetails[QtCore.QString("numbers")]
622 for (k, v) in number.iteritems()
624 for number in numbers
626 numbersWithDescriptions = [
628 number["phoneNumber"],
629 self._choose_phonetype(number),
631 for number in numbers
635 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
638 def _choose_phonetype(numberDetails):
639 if "phoneTypeName" in numberDetails:
640 return numberDetails["phoneTypeName"]
641 elif "phoneType" in numberDetails:
642 return numberDetails["phoneType"]
647 def make_ugly(prettynumber):
649 function to take a phone number and strip out all non-numeric
652 >>> make_ugly("+012-(345)-678-90")
655 return normalize_number(prettynumber)
658 def normalize_number(prettynumber):
660 function to take a phone number and strip out all non-numeric
663 >>> normalize_number("+012-(345)-678-90")
665 >>> normalize_number("1-(345)-678-9000")
667 >>> normalize_number("+1-(345)-678-9000")
670 uglynumber = re.sub('[^0-9+]', '', prettynumber)
672 if uglynumber.startswith("+"):
674 elif uglynumber.startswith("1"):
675 uglynumber = "+"+uglynumber
676 elif 10 <= len(uglynumber):
677 assert uglynumber[0] not in ("+", "1")
678 uglynumber = "+1"+uglynumber
685 def _make_pretty_with_areacode(phonenumber):
686 prettynumber = "(%s)" % (phonenumber[0:3], )
687 if 3 < len(phonenumber):
688 prettynumber += " %s" % (phonenumber[3:6], )
689 if 6 < len(phonenumber):
690 prettynumber += "-%s" % (phonenumber[6:], )
694 def _make_pretty_local(phonenumber):
695 prettynumber = "%s" % (phonenumber[0:3], )
696 if 3 < len(phonenumber):
697 prettynumber += "-%s" % (phonenumber[3:], )
701 def _make_pretty_international(phonenumber):
702 prettynumber = phonenumber
703 if phonenumber.startswith("1"):
705 prettynumber += _make_pretty_with_areacode(phonenumber[1:])
709 def make_pretty(phonenumber):
711 Function to take a phone number and return the pretty version
713 if phonenumber begins with 0:
715 if phonenumber begins with 1: ( for gizmo callback numbers )
717 if phonenumber is 13 digits:
719 if phonenumber is 10 digits:
721 >>> make_pretty("12")
723 >>> make_pretty("1234567")
725 >>> make_pretty("2345678901")
727 >>> make_pretty("12345678901")
729 >>> make_pretty("01234567890")
731 >>> make_pretty("+01234567890")
733 >>> make_pretty("+12")
735 >>> make_pretty("+123")
737 >>> make_pretty("+1234")
740 if phonenumber is None or phonenumber is "":
743 phonenumber = normalize_number(phonenumber)
745 if phonenumber[0] == "+":
746 prettynumber = _make_pretty_international(phonenumber[1:])
747 if not prettynumber.startswith("+"):
748 prettynumber = "+"+prettynumber
749 elif 8 < len(phonenumber) and phonenumber[0] in ("1", ):
750 prettynumber = _make_pretty_international(phonenumber)
751 elif 7 < len(phonenumber):
752 prettynumber = _make_pretty_with_areacode(phonenumber)
753 elif 3 < len(phonenumber):
754 prettynumber = _make_pretty_local(phonenumber)
756 prettynumber = phonenumber
757 return prettynumber.strip()
760 def abbrev_relative_date(date):
762 >>> abbrev_relative_date("42 hours ago")
764 >>> abbrev_relative_date("2 days ago")
766 >>> abbrev_relative_date("4 weeks ago")
769 parts = date.split(" ")
770 return "%s %s" % (parts[0], parts[1][0])