3 from __future__ import with_statement
4 from __future__ import division
10 from PyQt4 import QtGui
11 from PyQt4 import QtCore
13 from util import qtpie
14 from util import qui_utils
15 from util import misc as misc_utils
17 import backends.null_backend as null_backend
18 import backends.file_backend as file_backend
21 _moduleLogger = logging.getLogger(__name__)
24 class Dialpad(object):
26 def __init__(self, app, session, errorLog):
28 self._session = session
29 self._errorLog = errorLog
31 self._plus = QtGui.QPushButton("+")
32 self._plus.clicked.connect(lambda: self._on_keypress("+"))
33 self._entry = QtGui.QLineEdit()
35 backAction = QtGui.QAction(None)
36 backAction.setText("Back")
37 backAction.triggered.connect(self._on_backspace)
38 backPieItem = qtpie.QActionPieItem(backAction)
39 clearAction = QtGui.QAction(None)
40 clearAction.setText("Clear")
41 clearAction.triggered.connect(self._on_clear_text)
42 clearPieItem = qtpie.QActionPieItem(clearAction)
44 qtpie.PieFiling.NULL_CENTER,
46 qtpie.PieFiling.NULL_CENTER,
47 qtpie.PieFiling.NULL_CENTER,
49 self._back = qtpie.QPieButton(backPieItem)
50 self._back.set_center(backPieItem)
51 for slice in backSlices:
52 self._back.insertItem(slice)
54 self._entryLayout = QtGui.QHBoxLayout()
55 self._entryLayout.addWidget(self._plus, 1, QtCore.Qt.AlignCenter)
56 self._entryLayout.addWidget(self._entry, 1000)
57 self._entryLayout.addWidget(self._back, 1, QtCore.Qt.AlignCenter)
59 self._smsButton = QtGui.QPushButton("SMS")
60 self._smsButton.clicked.connect(self._on_sms_clicked)
61 self._callButton = QtGui.QPushButton("Call")
62 self._callButton.clicked.connect(self._on_call_clicked)
64 self._padLayout = QtGui.QGridLayout()
65 rows = [0, 0, 0, 1, 1, 1, 2, 2, 2]
66 columns = [0, 1, 2] * 3
78 for (num, letters), (row, column) in zip(keys, zip(rows, columns)):
79 self._padLayout.addWidget(self._generate_key_button(num, letters), row, column)
80 self._zerothButton = QtGui.QPushButton("0")
81 self._zerothButton.clicked.connect(lambda: self._on_keypress("0"))
82 self._padLayout.addWidget(self._smsButton, 3, 0)
83 self._padLayout.addWidget(self._zerothButton)
84 self._padLayout.addWidget(self._callButton, 3, 2)
86 self._layout = QtGui.QVBoxLayout()
87 self._layout.addLayout(self._entryLayout)
88 self._layout.addLayout(self._padLayout)
89 self._widget = QtGui.QWidget()
90 self._widget.setLayout(self._layout)
97 self._smsButton.setEnabled(True)
98 self._callButton.setEnabled(True)
101 self._smsButton.setEnabled(False)
102 self._callButton.setEnabled(False)
104 def get_settings(self):
107 def set_settings(self, settings):
113 def refresh(self, force = True):
116 def _generate_key_button(self, center, letters):
117 button = QtGui.QPushButton("%s\n%s" % (center, letters))
118 button.clicked.connect(lambda: self._on_keypress(center))
121 @misc_utils.log_exception(_moduleLogger)
122 def _on_keypress(self, key):
123 self._entry.insert(key)
125 @misc_utils.log_exception(_moduleLogger)
126 def _on_backspace(self, toggled = False):
127 self._entry.backspace()
129 @misc_utils.log_exception(_moduleLogger)
130 def _on_clear_text(self, toggled = False):
134 @QtCore.pyqtSlot(bool)
135 @misc_utils.log_exception(_moduleLogger)
136 def _on_sms_clicked(self, checked = False):
137 number = misc_utils.make_ugly(str(self._entry.text()))
141 title = misc_utils.make_pretty(number)
142 description = misc_utils.make_pretty(number)
143 numbersWithDescriptions = [(number, "")]
144 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
147 @QtCore.pyqtSlot(bool)
148 @misc_utils.log_exception(_moduleLogger)
149 def _on_call_clicked(self, checked = False):
150 number = misc_utils.make_ugly(str(self._entry.text()))
154 title = misc_utils.make_pretty(number)
155 description = misc_utils.make_pretty(number)
156 numbersWithDescriptions = [(number, "")]
157 self._session.draft.clear()
158 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
159 self._session.draft.call()
162 class History(object):
168 HISTORY_ITEM_TYPES = ["Received", "Missed", "Placed", "All"]
169 HISTORY_COLUMNS = ["Details", "From"]
170 assert len(HISTORY_COLUMNS) == MAX_IDX
172 def __init__(self, app, session, errorLog):
173 self._selectedFilter = self.HISTORY_ITEM_TYPES[-1]
175 self._session = session
176 self._session.historyUpdated.connect(self._on_history_updated)
177 self._errorLog = errorLog
179 self._typeSelection = QtGui.QComboBox()
180 self._typeSelection.addItems(self.HISTORY_ITEM_TYPES)
181 self._typeSelection.setCurrentIndex(
182 self.HISTORY_ITEM_TYPES.index(self._selectedFilter)
184 self._typeSelection.currentIndexChanged[str].connect(self._on_filter_changed)
186 self._itemStore = QtGui.QStandardItemModel()
187 self._itemStore.setHorizontalHeaderLabels(self.HISTORY_COLUMNS)
189 self._itemView = QtGui.QTreeView()
190 self._itemView.setModel(self._itemStore)
191 self._itemView.setUniformRowHeights(True)
192 self._itemView.setRootIsDecorated(False)
193 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
194 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
195 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
196 self._itemView.setHeaderHidden(True)
197 self._itemView.header().setResizeMode(QtGui.QHeaderView.ResizeToContents)
198 self._itemView.activated.connect(self._on_row_activated)
200 self._layout = QtGui.QVBoxLayout()
201 self._layout.addWidget(self._typeSelection)
202 self._layout.addWidget(self._itemView)
203 self._widget = QtGui.QWidget()
204 self._widget.setLayout(self._layout)
206 self._populate_items()
213 self._itemView.setEnabled(True)
216 self._itemView.setEnabled(False)
218 def get_settings(self):
220 "filter": str(self._typeSelection.currentText()),
223 def set_settings(self, settings):
224 selectedFilter = settings.get("filter", self.HISTORY_ITEM_TYPES[-1])
225 if selectedFilter in self.HISTORY_ITEM_TYPES:
226 self._selectedFilter = selectedFilter
227 self._typeSelection.setCurrentIndex(
228 self.HISTORY_ITEM_TYPES.index(selectedFilter)
232 self._itemView.clear()
234 def refresh(self, force=True):
235 self._session.update_history(force)
237 def _populate_items(self):
238 self._itemStore.clear()
240 history = self._session.get_history()
241 history.sort(key=lambda item: item["time"], reverse=True)
242 for event in history:
243 if self._selectedFilter in [self.HISTORY_ITEM_TYPES[-1], event["action"]]:
244 relTime = misc_utils.abbrev_relative_date(event["relTime"])
245 action = event["action"]
246 number = event["number"]
247 prettyNumber = misc_utils.make_pretty(number)
249 if not name or name == number:
250 name = event["location"]
254 detailsItem = QtGui.QStandardItem("%s - %s\n%s" % (relTime, action, prettyNumber))
255 detailsFont = detailsItem.font()
256 detailsFont.setPointSize(detailsFont.pointSize() - 4)
257 detailsItem.setFont(detailsFont)
258 nameItem = QtGui.QStandardItem(name)
259 nameFont = nameItem.font()
260 nameFont.setPointSize(nameFont.pointSize() + 4)
261 nameItem.setFont(nameFont)
262 row = detailsItem, nameItem
264 item.setEditable(False)
265 item.setCheckable(False)
266 if item is not nameItem:
267 itemFont = item.font()
268 itemFont.setPointSize(max(itemFont.pointSize() - 3, 5))
269 item.setFont(itemFont)
270 row[0].setData(event)
271 self._itemStore.appendRow(row)
273 @QtCore.pyqtSlot(str)
274 @misc_utils.log_exception(_moduleLogger)
275 def _on_filter_changed(self, newItem):
276 self._selectedFilter = str(newItem)
277 self._populate_items()
280 @misc_utils.log_exception(_moduleLogger)
281 def _on_history_updated(self):
282 self._populate_items()
284 @QtCore.pyqtSlot(QtCore.QModelIndex)
285 @misc_utils.log_exception(_moduleLogger)
286 def _on_row_activated(self, index):
287 rowIndex = index.row()
288 item = self._itemStore.item(rowIndex, 0)
289 contactDetails = item.data().toPyObject()
291 title = str(self._itemStore.item(rowIndex, self.FROM_IDX).text())
292 number = str(contactDetails[QtCore.QString("number")])
293 contactId = number # ids don't seem too unique so using numbers
296 for i in xrange(self._itemStore.rowCount()):
297 iItem = self._itemStore.item(i, 0)
298 iContactDetails = iItem.data().toPyObject()
299 iNumber = str(iContactDetails[QtCore.QString("number")])
300 if number != iNumber:
302 relTime = misc_utils.abbrev_relative_date(iContactDetails[QtCore.QString("relTime")])
303 action = str(iContactDetails[QtCore.QString("action")])
304 number = str(iContactDetails[QtCore.QString("number")])
305 prettyNumber = misc_utils.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.setRootIsDecorated(False)
362 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
363 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
364 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
365 self._itemView.setHeaderHidden(True)
366 self._itemView.setItemDelegate(self._htmlDelegate)
367 self._itemView.activated.connect(self._on_row_activated)
369 self._layout = QtGui.QVBoxLayout()
370 self._layout.addLayout(self._selectionLayout)
371 self._layout.addWidget(self._itemView)
372 self._widget = QtGui.QWidget()
373 self._widget.setLayout(self._layout)
375 self._populate_items()
382 self._itemView.setEnabled(True)
385 self._itemView.setEnabled(False)
387 def get_settings(self):
389 "type": str(self._typeSelection.currentText()),
390 "status": str(self._statusSelection.currentText()),
393 def set_settings(self, settings):
394 selectedType = settings.get("type", self.ALL_TYPES)
395 if selectedType in self.MESSAGE_TYPES:
396 self._selectedTypeFilter = selectedType
397 self._typeSelection.setCurrentIndex(
398 self.MESSAGE_TYPES.index(self._selectedTypeFilter)
401 selectedStatus = settings.get("status", self.ALL_STATUS)
402 if selectedStatus in self.MESSAGE_STATUSES:
403 self._selectedStatusFilter = selectedStatus
404 self._statusSelection.setCurrentIndex(
405 self.MESSAGE_STATUSES.index(self._selectedStatusFilter)
409 self._itemView.clear()
411 def refresh(self, force=True):
412 self._session.update_messages(force)
414 def _populate_items(self):
415 self._itemStore.clear()
416 rawMessages = self._session.get_messages()
417 rawMessages.sort(key=lambda item: item["time"], reverse=True)
418 for item in rawMessages:
419 isUnarchived = not item["isArchived"]
420 isUnread = not item["isRead"]
422 self.UNREAD_STATUS: isUnarchived and isUnread,
423 self.UNARCHIVED_STATUS: isUnarchived,
424 self.ALL_STATUS: True,
425 }[self._selectedStatusFilter]
427 visibleType = self._selectedTypeFilter in [item["type"], self.ALL_TYPES]
428 if visibleType and visibleStatus:
429 relTime = misc_utils.abbrev_relative_date(item["relTime"])
430 number = item["number"]
431 prettyNumber = misc_utils.make_pretty(number)
433 if not name or name == number:
434 name = item["location"]
438 messageParts = list(item["messageParts"])
439 if len(messageParts) == 0:
440 messages = ("No Transcription", )
441 elif len(messageParts) == 1:
442 if messageParts[0][1]:
443 messages = (messageParts[0][1], )
445 messages = ("No Transcription", )
448 "<b>%s</b>: %s" % (messagePart[0], messagePart[1])
449 for messagePart in messageParts
452 firstMessage = "<b>%s - %s</b> <i>(%s)</i>" % (name, prettyNumber, relTime)
454 expandedMessages = [firstMessage]
455 expandedMessages.extend(messages)
456 if (self._MIN_MESSAGES_SHOWN + 1) < len(messages):
457 secondMessage = "<i>%d Messages Hidden...</i>" % (len(messages) - self._MIN_MESSAGES_SHOWN, )
458 collapsedMessages = [firstMessage, secondMessage]
459 collapsedMessages.extend(messages[-(self._MIN_MESSAGES_SHOWN+0):])
461 collapsedMessages = expandedMessages
463 item = dict(item.iteritems())
464 item["collapsedMessages"] = "<br/>\n".join(collapsedMessages)
465 item["expandedMessages"] = "<br/>\n".join(expandedMessages)
467 messageItem = QtGui.QStandardItem(item["collapsedMessages"])
468 messageItem.setData(item)
469 messageItem.setEditable(False)
470 messageItem.setCheckable(False)
471 row = (messageItem, )
472 self._itemStore.appendRow(row)
474 @QtCore.pyqtSlot(str)
475 @misc_utils.log_exception(_moduleLogger)
476 def _on_type_filter_changed(self, newItem):
477 self._selectedTypeFilter = str(newItem)
478 self._populate_items()
480 @QtCore.pyqtSlot(str)
481 @misc_utils.log_exception(_moduleLogger)
482 def _on_status_filter_changed(self, newItem):
483 self._selectedStatusFilter = str(newItem)
484 self._populate_items()
487 @misc_utils.log_exception(_moduleLogger)
488 def _on_messages_updated(self):
489 self._populate_items()
491 @QtCore.pyqtSlot(QtCore.QModelIndex)
492 @misc_utils.log_exception(_moduleLogger)
493 def _on_row_activated(self, index):
494 rowIndex = index.row()
495 item = self._itemStore.item(rowIndex, 0)
496 contactDetails = item.data().toPyObject()
498 name = str(contactDetails[QtCore.QString("name")])
499 number = str(contactDetails[QtCore.QString("number")])
500 if not name or name == number:
501 name = str(contactDetails[QtCore.QString("location")])
505 contactId = str(contactDetails[QtCore.QString("id")])
507 description = str(contactDetails[QtCore.QString("expandedMessages")])
508 numbersWithDescriptions = [(number, "")]
509 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
512 class Contacts(object):
514 # @todo Provide some sort of letter jump
516 def __init__(self, app, session, errorLog):
518 self._session = session
519 self._session.contactsUpdated.connect(self._on_contacts_updated)
520 self._errorLog = errorLog
521 self._addressBookFactories = [
522 null_backend.NullAddressBookFactory(),
523 file_backend.FilesystemAddressBookFactory(app.fsContactsPath),
525 self._addressBooks = []
527 self._listSelection = QtGui.QComboBox()
528 self._listSelection.addItems([])
529 self._listSelection.currentIndexChanged[str].connect(self._on_filter_changed)
531 self._itemStore = QtGui.QStandardItemModel()
532 self._itemStore.setHorizontalHeaderLabels(["Contacts"])
535 self._itemView = QtGui.QTreeView()
536 self._itemView.setModel(self._itemStore)
537 self._itemView.setUniformRowHeights(True)
538 self._itemView.setRootIsDecorated(False)
539 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
540 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
541 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
542 self._itemView.setHeaderHidden(True)
543 self._itemView.setItemsExpandable(False)
544 self._itemView.activated.connect(self._on_row_activated)
546 self._layout = QtGui.QVBoxLayout()
547 self._layout.addWidget(self._listSelection)
548 self._layout.addWidget(self._itemView)
549 self._widget = QtGui.QWidget()
550 self._widget.setLayout(self._layout)
552 self.update_addressbooks()
553 self._populate_items()
560 self._itemView.setEnabled(True)
563 self._itemView.setEnabled(False)
565 def get_settings(self):
567 "selectedAddressbook": str(self._listSelection.currentText()),
570 def set_settings(self, settings):
571 currentItem = settings.get("selectedAddressbook", "None")
572 bookNames = [book["name"] for book in self._addressBooks]
574 newIndex = bookNames.index(currentItem)
576 # Switch over to None for the user
578 self._listSelection.setCurrentIndex(newIndex)
581 self._itemView.clear()
583 def refresh(self, force=True):
584 self._backend.update_contacts(force)
588 return self._addressBooks[self._listSelection.currentIndex()]["book"]
590 def update_addressbooks(self):
591 self._addressBooks = [
592 {"book": book, "name": book.name}
593 for factory in self._addressBookFactories
594 for book in factory.get_addressbooks()
596 self._addressBooks.append(
598 "book": self._session,
599 "name": "Google Voice",
603 currentItem = str(self._listSelection.currentText())
604 if currentItem == "":
607 self._listSelection.clear()
608 bookNames = [book["name"] for book in self._addressBooks]
610 newIndex = bookNames.index(currentItem)
612 # Switch over to None for the user
614 self._itemStore.clear()
615 _moduleLogger.info("Addressbook %r doesn't exist anymore, switching to None" % currentItem)
616 self._listSelection.addItems(bookNames)
617 self._listSelection.setCurrentIndex(newIndex)
619 def _populate_items(self):
620 self._itemStore.clear()
621 self._alphaItem = dict(
622 (letter, QtGui.QStandardItem(letter))
623 for letter in self._prefixes()
625 for letter in self._prefixes():
626 item = self._alphaItem[letter]
627 item.setEditable(False)
628 item.setCheckable(False)
630 self._itemStore.appendRow(row)
632 for item in self._get_contacts():
634 numbers = item["numbers"]
635 nameItem = QtGui.QStandardItem(name)
636 nameItem.setEditable(False)
637 nameItem.setCheckable(False)
638 nameItem.setData(item)
640 rowKey = name[0].upper()
641 rowKey = rowKey if rowKey in self._alphaItem else "#"
642 self._alphaItem[rowKey].appendRow(row)
643 self._itemView.expandAll()
646 return itertools.chain(string.ascii_uppercase, ("#", ))
648 def _jump_to_prefix(self, letter):
649 i = list(self._prefixes()).index(letter)
650 rootIndex = self._itemView.rootIndex()
651 currentIndex = self._itemView.model().index(i, 0, rootIndex)
652 self._itemView.scrollTo(currentIndex)
653 self._itemView.setItemSelected(self._itemView.topLevelItem(i), True)
655 def _get_contacts(self):
656 contacts = list(self._backend.get_contacts().itervalues())
657 contacts.sort(key=lambda contact: contact["name"].lower())
660 @QtCore.pyqtSlot(str)
661 @misc_utils.log_exception(_moduleLogger)
662 def _on_filter_changed(self, newItem):
663 self.refresh(force=False)
664 self._populate_items()
667 @misc_utils.log_exception(_moduleLogger)
668 def _on_contacts_updated(self):
669 self._populate_items()
671 @QtCore.pyqtSlot(QtCore.QModelIndex)
672 @misc_utils.log_exception(_moduleLogger)
673 def _on_row_activated(self, index):
674 letterIndex = index.parent()
675 assert letterIndex.isValid()
676 letterRow = letterIndex.row()
677 letter = list(self._prefixes())[letterRow]
678 letterItem = self._alphaItem[letter]
679 rowIndex = index.row()
680 item = letterItem.child(rowIndex, 0)
681 contactDetails = item.data().toPyObject()
683 name = str(contactDetails[QtCore.QString("name")])
685 name = str(contactDetails[QtCore.QString("location")])
689 contactId = str(contactDetails[QtCore.QString("contactId")])
690 numbers = contactDetails[QtCore.QString("numbers")]
694 for (k, v) in number.iteritems()
696 for number in numbers
698 numbersWithDescriptions = [
700 number["phoneNumber"],
701 self._choose_phonetype(number),
703 for number in numbers
707 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
710 def _choose_phonetype(numberDetails):
711 if "phoneTypeName" in numberDetails:
712 return numberDetails["phoneTypeName"]
713 elif "phoneType" in numberDetails:
714 return numberDetails["phoneType"]