3 from __future__ import with_statement
4 from __future__ import division
8 from PyQt4 import QtGui
9 from PyQt4 import QtCore
11 from util import qtpie
12 from util import qui_utils
13 from util import misc as misc_utils
15 import backends.null_backend as null_backend
16 import backends.file_backend as file_backend
19 _moduleLogger = logging.getLogger(__name__)
22 class Dialpad(object):
24 def __init__(self, app, session, errorLog):
26 self._session = session
27 self._errorLog = errorLog
29 self._plus = self._generate_key_button("+", "")
30 self._entry = QtGui.QLineEdit()
32 backAction = QtGui.QAction(None)
33 backAction.setText("Back")
34 backAction.triggered.connect(self._on_backspace)
35 backPieItem = qtpie.QActionPieItem(backAction)
36 clearAction = QtGui.QAction(None)
37 clearAction.setText("Clear")
38 clearAction.triggered.connect(self._on_clear_text)
39 clearPieItem = qtpie.QActionPieItem(clearAction)
41 qtpie.PieFiling.NULL_CENTER,
43 qtpie.PieFiling.NULL_CENTER,
44 qtpie.PieFiling.NULL_CENTER,
46 self._back = qtpie.QPieButton(backPieItem)
47 self._back.set_center(backPieItem)
48 for slice in backSlices:
49 self._back.insertItem(slice)
51 self._entryLayout = QtGui.QHBoxLayout()
52 self._entryLayout.addWidget(self._plus, 0, QtCore.Qt.AlignCenter)
53 self._entryLayout.addWidget(self._entry, 10)
54 self._entryLayout.addWidget(self._back, 0, QtCore.Qt.AlignCenter)
56 self._smsButton = QtGui.QPushButton("SMS")
57 self._smsButton.clicked.connect(self._on_sms_clicked)
58 self._callButton = QtGui.QPushButton("Call")
59 self._callButton.clicked.connect(self._on_call_clicked)
61 self._padLayout = QtGui.QGridLayout()
62 rows = [0, 0, 0, 1, 1, 1, 2, 2, 2]
63 columns = [0, 1, 2] * 3
75 for (num, letters), (row, column) in zip(keys, zip(rows, columns)):
76 self._padLayout.addWidget(
77 self._generate_key_button(num, letters), row, column, QtCore.Qt.AlignCenter
79 self._padLayout.addWidget(self._smsButton, 3, 0)
80 self._padLayout.addWidget(
81 self._generate_key_button("0", ""), 3, 1, QtCore.Qt.AlignCenter
83 self._padLayout.addWidget(self._callButton, 3, 2)
85 self._layout = QtGui.QVBoxLayout()
86 self._layout.addLayout(self._entryLayout)
87 self._layout.addLayout(self._padLayout)
88 self._widget = QtGui.QWidget()
89 self._widget.setLayout(self._layout)
96 self._smsButton.setEnabled(True)
97 self._callButton.setEnabled(True)
100 self._smsButton.setEnabled(False)
101 self._callButton.setEnabled(False)
103 def get_settings(self):
106 def set_settings(self, settings):
112 def refresh(self, force = True):
115 def _generate_key_button(self, center, letters):
116 centerPieItem = self._generate_button_slice(center)
119 for pieItem in self._generate_button_slices(letters)
122 button = qtpie.QPieButton(centerPieItem)
123 button.set_center(centerPieItem)
124 for item in pieItems:
125 button.insertItem(item)
129 def _generate_button_slices(self, letters):
130 if len(letters) == 0:
132 pieItem = qtpie.PieFiling.NULL_CENTER
134 elif len(letters) in [3, 4]:
135 for i in xrange(6 - len(letters)):
136 pieItem = qtpie.PieFiling.NULL_CENTER
139 for letter in letters:
140 pieItem = self._generate_button_slice(letter)
144 pieItem = qtpie.PieFiling.NULL_CENTER
147 raise NotImplementedError("Cannot handle %r" % letters)
149 def _generate_button_slice(self, letter):
150 action = QtGui.QAction(None)
151 action.setText(letter)
152 action.triggered.connect(lambda: self._on_keypress(letter))
153 pieItem = qtpie.QActionPieItem(action)
156 @misc_utils.log_exception(_moduleLogger)
157 def _on_keypress(self, key):
158 self._entry.insert(key)
160 @misc_utils.log_exception(_moduleLogger)
161 def _on_backspace(self, toggled = False):
162 self._entry.backspace()
164 @misc_utils.log_exception(_moduleLogger)
165 def _on_clear_text(self, toggled = False):
169 @QtCore.pyqtSlot(bool)
170 @misc_utils.log_exception(_moduleLogger)
171 def _on_sms_clicked(self, checked = False):
172 number = str(self._entry.text())
178 numbersWithDescriptions = [(number, "")]
179 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
182 @QtCore.pyqtSlot(bool)
183 @misc_utils.log_exception(_moduleLogger)
184 def _on_call_clicked(self, checked = False):
185 number = str(self._entry.text())
191 numbersWithDescriptions = [(number, "")]
192 self._session.draft.clear()
193 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
194 self._session.draft.call()
197 class History(object):
205 HISTORY_ITEM_TYPES = ["Received", "Missed", "Placed", "All"]
206 HISTORY_COLUMNS = ["When", "What", "Number", "From"]
207 assert len(HISTORY_COLUMNS) == MAX_IDX
209 def __init__(self, app, session, errorLog):
210 self._selectedFilter = self.HISTORY_ITEM_TYPES[-1]
212 self._session = session
213 self._session.historyUpdated.connect(self._on_history_updated)
214 self._errorLog = errorLog
216 self._typeSelection = QtGui.QComboBox()
217 self._typeSelection.addItems(self.HISTORY_ITEM_TYPES)
218 self._typeSelection.setCurrentIndex(
219 self.HISTORY_ITEM_TYPES.index(self._selectedFilter)
221 self._typeSelection.currentIndexChanged[str].connect(self._on_filter_changed)
223 self._itemStore = QtGui.QStandardItemModel()
224 self._itemStore.setHorizontalHeaderLabels(self.HISTORY_COLUMNS)
226 self._itemView = QtGui.QTreeView()
227 self._itemView.setModel(self._itemStore)
228 self._itemView.setUniformRowHeights(True)
229 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
230 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
231 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
232 self._itemView.setHeaderHidden(True)
233 self._itemView.activated.connect(self._on_row_activated)
235 self._layout = QtGui.QVBoxLayout()
236 self._layout.addWidget(self._typeSelection)
237 self._layout.addWidget(self._itemView)
238 self._widget = QtGui.QWidget()
239 self._widget.setLayout(self._layout)
241 self._populate_items()
248 self._itemView.setEnabled(True)
251 self._itemView.setEnabled(False)
253 def get_settings(self):
255 "filter": str(self._typeSelection.currentText()),
258 def set_settings(self, settings):
259 selectedFilter = settings.get("filter", self.HISTORY_ITEM_TYPES[-1])
260 if selectedFilter in self.HISTORY_ITEM_TYPES:
261 self._selectedFilter = selectedFilter
262 self._typeSelection.setCurrentIndex(
263 self.HISTORY_ITEM_TYPES.index(selectedFilter)
267 self._itemView.clear()
269 def refresh(self, force=True):
270 self._session.update_history(force)
272 def _populate_items(self):
273 self._itemStore.clear()
274 history = self._session.get_history()
275 history.sort(key=lambda item: item["time"], reverse=True)
276 for event in history:
277 if self._selectedFilter in [self.HISTORY_ITEM_TYPES[-1], event["action"]]:
278 relTime = misc_utils.abbrev_relative_date(event["relTime"])
279 action = event["action"]
280 number = event["number"]
281 prettyNumber = misc_utils.make_pretty(number)
283 if not name or name == number:
284 name = event["location"]
288 timeItem = QtGui.QStandardItem(relTime)
289 actionItem = QtGui.QStandardItem(action)
290 numberItem = QtGui.QStandardItem(prettyNumber)
291 nameItem = QtGui.QStandardItem(name)
292 row = timeItem, actionItem, numberItem, nameItem
294 item.setEditable(False)
295 item.setCheckable(False)
296 if item is not nameItem:
297 itemFont = item.font()
298 itemFont.setPointSize(max(itemFont.pointSize() - 3, 5))
299 item.setFont(itemFont)
300 numberItem.setData(event)
301 self._itemStore.appendRow(row)
303 @QtCore.pyqtSlot(str)
304 @misc_utils.log_exception(_moduleLogger)
305 def _on_filter_changed(self, newItem):
306 self._selectedFilter = str(newItem)
307 self._populate_items()
310 @misc_utils.log_exception(_moduleLogger)
311 def _on_history_updated(self):
312 self._populate_items()
314 @QtCore.pyqtSlot(QtCore.QModelIndex)
315 @misc_utils.log_exception(_moduleLogger)
316 def _on_row_activated(self, index):
317 rowIndex = index.row()
318 item = self._itemStore.item(rowIndex, self.NUMBER_IDX)
319 contactDetails = item.data().toPyObject()
321 title = str(self._itemStore.item(rowIndex, self.FROM_IDX).text())
322 number = str(contactDetails[QtCore.QString("number")])
323 contactId = number # ids don't seem too unique so using numbers
326 # @bug doesn't seem to print multiple entries
327 for i in xrange(self._itemStore.rowCount()):
328 iItem = self._itemStore.item(i, self.NUMBER_IDX)
329 iContactDetails = iItem.data().toPyObject()
330 iNumber = str(iContactDetails[QtCore.QString("number")])
331 if number != iNumber:
333 relTime = misc_utils.abbrev_relative_date(iContactDetails[QtCore.QString("relTime")])
334 action = str(iContactDetails[QtCore.QString("action")])
335 number = str(iContactDetails[QtCore.QString("number")])
336 prettyNumber = misc_utils.make_pretty(number)
337 rowItems = relTime, action, prettyNumber
338 descriptionRows.append("<tr><td>%s</td></tr>" % "</td><td>".join(rowItems))
339 description = "<table>%s</table>" % "".join(descriptionRows)
340 numbersWithDescriptions = [(str(contactDetails[QtCore.QString("number")]), "")]
341 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
344 class Messages(object):
347 VOICEMAIL_MESSAGES = "Voicemail"
348 TEXT_MESSAGES = "SMS"
349 ALL_TYPES = "All Messages"
350 MESSAGE_TYPES = [NO_MESSAGES, VOICEMAIL_MESSAGES, TEXT_MESSAGES, ALL_TYPES]
352 UNREAD_STATUS = "Unread"
353 UNARCHIVED_STATUS = "Inbox"
355 MESSAGE_STATUSES = [UNREAD_STATUS, UNARCHIVED_STATUS, ALL_STATUS]
357 _MIN_MESSAGES_SHOWN = 4
359 def __init__(self, app, session, errorLog):
360 self._selectedTypeFilter = self.ALL_TYPES
361 self._selectedStatusFilter = self.ALL_STATUS
363 self._session = session
364 self._session.messagesUpdated.connect(self._on_messages_updated)
365 self._errorLog = errorLog
367 self._typeSelection = QtGui.QComboBox()
368 self._typeSelection.addItems(self.MESSAGE_TYPES)
369 self._typeSelection.setCurrentIndex(
370 self.MESSAGE_TYPES.index(self._selectedTypeFilter)
372 self._typeSelection.currentIndexChanged[str].connect(self._on_type_filter_changed)
374 self._statusSelection = QtGui.QComboBox()
375 self._statusSelection.addItems(self.MESSAGE_STATUSES)
376 self._statusSelection.setCurrentIndex(
377 self.MESSAGE_STATUSES.index(self._selectedStatusFilter)
379 self._statusSelection.currentIndexChanged[str].connect(self._on_status_filter_changed)
381 self._selectionLayout = QtGui.QHBoxLayout()
382 self._selectionLayout.addWidget(self._typeSelection)
383 self._selectionLayout.addWidget(self._statusSelection)
385 self._itemStore = QtGui.QStandardItemModel()
386 self._itemStore.setHorizontalHeaderLabels(["Messages"])
388 self._htmlDelegate = qui_utils.QHtmlDelegate()
389 self._itemView = QtGui.QTreeView()
390 self._itemView.setModel(self._itemStore)
391 self._itemView.setUniformRowHeights(False)
392 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
393 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
394 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
395 self._itemView.setHeaderHidden(True)
396 self._itemView.setItemDelegate(self._htmlDelegate)
397 self._itemView.activated.connect(self._on_row_activated)
399 self._layout = QtGui.QVBoxLayout()
400 self._layout.addLayout(self._selectionLayout)
401 self._layout.addWidget(self._itemView)
402 self._widget = QtGui.QWidget()
403 self._widget.setLayout(self._layout)
405 self._populate_items()
412 self._itemView.setEnabled(True)
415 self._itemView.setEnabled(False)
417 def get_settings(self):
419 "type": str(self._typeSelection.currentText()),
420 "status": str(self._statusSelection.currentText()),
423 def set_settings(self, settings):
424 selectedType = settings.get("type", self.ALL_TYPES)
425 if selectedType in self.MESSAGE_TYPES:
426 self._selectedTypeFilter = selectedType
427 self._typeSelection.setCurrentIndex(
428 self.MESSAGE_TYPES.index(self._selectedTypeFilter)
431 selectedStatus = settings.get("status", self.ALL_STATUS)
432 if selectedStatus in self.MESSAGE_STATUSES:
433 self._selectedStatusFilter = selectedStatus
434 self._statusSelection.setCurrentIndex(
435 self.MESSAGE_STATUSES.index(self._selectedStatusFilter)
439 self._itemView.clear()
441 def refresh(self, force=True):
442 self._session.update_messages(force)
444 def _populate_items(self):
445 self._itemStore.clear()
446 rawMessages = self._session.get_messages()
447 rawMessages.sort(key=lambda item: item["time"], reverse=True)
448 for item in rawMessages:
449 isUnarchived = not item["isArchived"]
450 isUnread = not item["isRead"]
452 self.UNREAD_STATUS: isUnarchived and isUnread,
453 self.UNARCHIVED_STATUS: isUnarchived,
454 self.ALL_STATUS: True,
455 }[self._selectedStatusFilter]
457 visibleType = self._selectedTypeFilter in [item["type"], self.ALL_TYPES]
458 if visibleType and visibleStatus:
459 relTime = misc_utils.abbrev_relative_date(item["relTime"])
460 number = item["number"]
461 prettyNumber = misc_utils.make_pretty(number)
463 if not name or name == number:
464 name = item["location"]
468 messageParts = list(item["messageParts"])
469 if len(messageParts) == 0:
470 messages = ("No Transcription", )
471 elif len(messageParts) == 1:
472 if messageParts[0][1]:
473 messages = (messageParts[0][1], )
475 messages = ("No Transcription", )
478 "<b>%s</b>: %s" % (messagePart[0], messagePart[1])
479 for messagePart in messageParts
482 firstMessage = "<b>%s - %s</b> <i>(%s)</i>" % (name, prettyNumber, relTime)
484 expandedMessages = [firstMessage]
485 expandedMessages.extend(messages)
486 if (self._MIN_MESSAGES_SHOWN + 1) < len(messages):
487 secondMessage = "<i>%d Messages Hidden...</i>" % (len(messages) - self._MIN_MESSAGES_SHOWN, )
488 collapsedMessages = [firstMessage, secondMessage]
489 collapsedMessages.extend(messages[-(self._MIN_MESSAGES_SHOWN+0):])
491 collapsedMessages = expandedMessages
493 item = dict(item.iteritems())
494 item["collapsedMessages"] = "<br/>\n".join(collapsedMessages)
495 item["expandedMessages"] = "<br/>\n".join(expandedMessages)
497 messageItem = QtGui.QStandardItem(item["collapsedMessages"])
498 # @bug Not showing all of a message
499 messageItem.setData(item)
500 messageItem.setEditable(False)
501 messageItem.setCheckable(False)
502 row = (messageItem, )
503 self._itemStore.appendRow(row)
505 @QtCore.pyqtSlot(str)
506 @misc_utils.log_exception(_moduleLogger)
507 def _on_type_filter_changed(self, newItem):
508 self._selectedTypeFilter = str(newItem)
509 self._populate_items()
511 @QtCore.pyqtSlot(str)
512 @misc_utils.log_exception(_moduleLogger)
513 def _on_status_filter_changed(self, newItem):
514 self._selectedStatusFilter = str(newItem)
515 self._populate_items()
518 @misc_utils.log_exception(_moduleLogger)
519 def _on_messages_updated(self):
520 self._populate_items()
522 @QtCore.pyqtSlot(QtCore.QModelIndex)
523 @misc_utils.log_exception(_moduleLogger)
524 def _on_row_activated(self, index):
525 rowIndex = index.row()
526 item = self._itemStore.item(rowIndex, 0)
527 contactDetails = item.data().toPyObject()
529 name = str(contactDetails[QtCore.QString("name")])
530 number = str(contactDetails[QtCore.QString("number")])
531 if not name or name == number:
532 name = str(contactDetails[QtCore.QString("location")])
536 contactId = str(contactDetails[QtCore.QString("id")])
538 description = str(contactDetails[QtCore.QString("expandedMessages")])
539 numbersWithDescriptions = [(number, "")]
540 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
543 class Contacts(object):
545 # @todo Provide some sort of letter jump
547 def __init__(self, app, session, errorLog):
549 self._session = session
550 self._session.contactsUpdated.connect(self._on_contacts_updated)
551 self._errorLog = errorLog
552 self._addressBookFactories = [
553 null_backend.NullAddressBookFactory(),
554 file_backend.FilesystemAddressBookFactory(app.fsContactsPath),
556 self._addressBooks = []
558 self._listSelection = QtGui.QComboBox()
559 self._listSelection.addItems([])
560 self._listSelection.currentIndexChanged[str].connect(self._on_filter_changed)
562 self._itemStore = QtGui.QStandardItemModel()
563 self._itemStore.setHorizontalHeaderLabels(["Contacts"])
565 self._itemView = QtGui.QTreeView()
566 self._itemView.setModel(self._itemStore)
567 self._itemView.setUniformRowHeights(True)
568 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
569 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
570 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
571 self._itemView.setHeaderHidden(True)
572 self._itemView.activated.connect(self._on_row_activated)
574 self._layout = QtGui.QVBoxLayout()
575 self._layout.addWidget(self._listSelection)
576 self._layout.addWidget(self._itemView)
577 self._widget = QtGui.QWidget()
578 self._widget.setLayout(self._layout)
580 self.update_addressbooks()
581 self._populate_items()
588 self._itemView.setEnabled(True)
591 self._itemView.setEnabled(False)
593 def get_settings(self):
595 "selectedAddressbook": str(self._listSelection.currentText()),
598 def set_settings(self, settings):
599 currentItem = settings.get("selectedAddressbook", "None")
600 bookNames = [book["name"] for book in self._addressBooks]
602 newIndex = bookNames.index(currentItem)
604 # Switch over to None for the user
606 self._listSelection.setCurrentIndex(newIndex)
609 self._itemView.clear()
611 def refresh(self, force=True):
612 self._backend.update_contacts(force)
616 return self._addressBooks[self._listSelection.currentIndex()]["book"]
618 def update_addressbooks(self):
619 self._addressBooks = [
620 {"book": book, "name": book.name}
621 for factory in self._addressBookFactories
622 for book in factory.get_addressbooks()
624 self._addressBooks.append(
626 "book": self._session,
627 "name": "Google Voice",
631 currentItem = str(self._listSelection.currentText())
632 if currentItem == "":
635 while 0 < self._listSelection.count():
636 self._listSelection.removeItem(0)
637 bookNames = [book["name"] for book in self._addressBooks]
639 newIndex = bookNames.index(currentItem)
641 # Switch over to None for the user
643 self._itemStore.clear()
644 _moduleLogger.info("Addressbook %r doesn't exist anymore, switching to None" % currentItem)
645 self._listSelection.addItems(bookNames)
646 self._listSelection.setCurrentIndex(newIndex)
648 def _populate_items(self):
649 self._itemStore.clear()
651 contacts = list(self._backend.get_contacts().itervalues())
652 contacts.sort(key=lambda contact: contact["name"].lower())
653 for item in contacts:
655 numbers = item["numbers"]
656 nameItem = QtGui.QStandardItem(name)
657 nameItem.setEditable(False)
658 nameItem.setCheckable(False)
659 nameItem.setData(item)
661 self._itemStore.appendRow(row)
663 @QtCore.pyqtSlot(str)
664 @misc_utils.log_exception(_moduleLogger)
665 def _on_filter_changed(self, newItem):
666 self._populate_items()
669 @misc_utils.log_exception(_moduleLogger)
670 def _on_contacts_updated(self):
671 self._populate_items()
673 @QtCore.pyqtSlot(QtCore.QModelIndex)
674 @misc_utils.log_exception(_moduleLogger)
675 def _on_row_activated(self, index):
676 rowIndex = index.row()
677 item = self._itemStore.item(rowIndex, 0)
678 contactDetails = item.data().toPyObject()
680 name = str(contactDetails[QtCore.QString("name")])
682 name = str(contactDetails[QtCore.QString("location")])
686 contactId = str(contactDetails[QtCore.QString("contactId")])
687 numbers = contactDetails[QtCore.QString("numbers")]
691 for (k, v) in number.iteritems()
693 for number in numbers
695 numbersWithDescriptions = [
697 number["phoneNumber"],
698 self._choose_phonetype(number),
700 for number in numbers
704 self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
707 def _choose_phonetype(numberDetails):
708 if "phoneTypeName" in numberDetails:
709 return numberDetails["phoneTypeName"]
710 elif "phoneType" in numberDetails:
711 return numberDetails["phoneType"]