git.maemo.org Git - gc-dialer/blob - src/gv_views.py

   1 #!/usr/bin/env python
   2
   3 from __future__ import with_statement
   4 from __future__ import division
   5
   6 import datetime
   7 import string
   8 import itertools
   9 import logging
  10
  11 from PyQt4 import QtGui
  12 from PyQt4 import QtCore
  13
  14 from util import qtpie
  15 from util import qui_utils
  16 from util import misc as misc_utils
  17
  18 import backends.null_backend as null_backend
  19 import backends.file_backend as file_backend
  20
  21
  22 _moduleLogger = logging.getLogger(__name__)
  23
  24
  25 class Dialpad(object):
  26
  27         def __init__(self, app, session, errorLog):
  28                 self._app = app
  29                 self._session = session
  30                 self._errorLog = errorLog
  31
  32                 self._plus = QtGui.QPushButton("+")
  33                 self._plus.clicked.connect(lambda: self._on_keypress("+"))
  34                 self._entry = QtGui.QLineEdit()
  35
  36                 backAction = QtGui.QAction(None)
  37                 backAction.setText("Back")
  38                 backAction.triggered.connect(self._on_backspace)
  39                 backPieItem = qtpie.QActionPieItem(backAction)
  40                 clearAction = QtGui.QAction(None)
  41                 clearAction.setText("Clear")
  42                 clearAction.triggered.connect(self._on_clear_text)
  43                 clearPieItem = qtpie.QActionPieItem(clearAction)
  44                 backSlices = [
  45                         qtpie.PieFiling.NULL_CENTER,
  46                         clearPieItem,
  47                         qtpie.PieFiling.NULL_CENTER,
  48                         qtpie.PieFiling.NULL_CENTER,
  49                 ]
  50                 self._back = qtpie.QPieButton(backPieItem)
  51                 self._back.set_center(backPieItem)
  52                 for slice in backSlices:
  53                         self._back.insertItem(slice)
  54
  55                 self._entryLayout = QtGui.QHBoxLayout()
  56                 self._entryLayout.addWidget(self._plus, 1, QtCore.Qt.AlignCenter)
  57                 self._entryLayout.addWidget(self._entry, 1000)
  58                 self._entryLayout.addWidget(self._back, 1, QtCore.Qt.AlignCenter)
  59
  60                 smsIcon = self._app.get_icon("messages.png")
  61                 self._smsButton = QtGui.QPushButton(smsIcon, "SMS")
  62                 self._smsButton.clicked.connect(self._on_sms_clicked)
  63                 callIcon = self._app.get_icon("dialpad.png")
  64                 self._callButton = QtGui.QPushButton(callIcon, "Call")
  65                 self._callButton.clicked.connect(self._on_call_clicked)
  66
  67                 self._padLayout = QtGui.QGridLayout()
  68                 rows = [0, 0, 0, 1, 1, 1, 2, 2, 2]
  69                 columns = [0, 1, 2] * 3
  70                 keys = [
  71                         ("1", ""),
  72                         ("2", "ABC"),
  73                         ("3", "DEF"),
  74                         ("4", "GHI"),
  75                         ("5", "JKL"),
  76                         ("6", "MNO"),
  77                         ("7", "PQRS"),
  78                         ("8", "TUV"),
  79                         ("9", "WXYZ"),
  80                 ]
  81                 for (num, letters), (row, column) in zip(keys, zip(rows, columns)):
  82                         self._padLayout.addWidget(self._generate_key_button(num, letters), row, column)
  83                 self._zerothButton = QtGui.QPushButton("0")
  84                 self._zerothButton.clicked.connect(lambda: self._on_keypress("0"))
  85                 self._padLayout.addWidget(self._smsButton, 3, 0)
  86                 self._padLayout.addWidget(self._zerothButton)
  87                 self._padLayout.addWidget(self._callButton, 3, 2)
  88
  89                 self._layout = QtGui.QVBoxLayout()
  90                 self._layout.addLayout(self._entryLayout)
  91                 self._layout.addLayout(self._padLayout)
  92                 self._widget = QtGui.QWidget()
  93                 self._widget.setLayout(self._layout)
  94
  95         @property
  96         def toplevel(self):
  97                 return self._widget
  98
  99         def enable(self):
 100                 self._smsButton.setEnabled(True)
 101                 self._callButton.setEnabled(True)
 102
 103         def disable(self):
 104                 self._smsButton.setEnabled(False)
 105                 self._callButton.setEnabled(False)
 106
 107         def get_settings(self):
 108                 return {}
 109
 110         def set_settings(self, settings):
 111                 pass
 112
 113         def clear(self):
 114                 pass
 115
 116         def refresh(self, force = True):
 117                 pass
 118
 119         def _generate_key_button(self, center, letters):
 120                 button = QtGui.QPushButton("%s\n%s" % (center, letters))
 121                 button.clicked.connect(lambda: self._on_keypress(center))
 122                 return button
 123
 124         @misc_utils.log_exception(_moduleLogger)
 125         def _on_keypress(self, key):
 126                 with qui_utils.notify_error(self._errorLog):
 127                         self._entry.insert(key)
 128
 129         @misc_utils.log_exception(_moduleLogger)
 130         def _on_backspace(self, toggled = False):
 131                 with qui_utils.notify_error(self._errorLog):
 132                         self._entry.backspace()
 133
 134         @misc_utils.log_exception(_moduleLogger)
 135         def _on_clear_text(self, toggled = False):
 136                 with qui_utils.notify_error(self._errorLog):
 137                         self._entry.clear()
 138
 139         @QtCore.pyqtSlot()
 140         @QtCore.pyqtSlot(bool)
 141         @misc_utils.log_exception(_moduleLogger)
 142         def _on_sms_clicked(self, checked = False):
 143                 with qui_utils.notify_error(self._errorLog):
 144                         number = misc_utils.make_ugly(str(self._entry.text()))
 145                         self._entry.clear()
 146
 147                         contactId = number
 148                         title = misc_utils.make_pretty(number)
 149                         description = misc_utils.make_pretty(number)
 150                         numbersWithDescriptions = [(number, "")]
 151                         self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
 152
 153         @QtCore.pyqtSlot()
 154         @QtCore.pyqtSlot(bool)
 155         @misc_utils.log_exception(_moduleLogger)
 156         def _on_call_clicked(self, checked = False):
 157                 with qui_utils.notify_error(self._errorLog):
 158                         number = misc_utils.make_ugly(str(self._entry.text()))
 159                         self._entry.clear()
 160
 161                         contactId = number
 162                         title = misc_utils.make_pretty(number)
 163                         description = misc_utils.make_pretty(number)
 164                         numbersWithDescriptions = [(number, "")]
 165                         self._session.draft.clear()
 166                         self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
 167                         self._session.draft.call()
 168
 169
 170 class TimeCategories(object):
 171
 172         _NOW_SECTION = 0
 173         _TODAY_SECTION = 1
 174         _WEEK_SECTION = 2
 175         _MONTH_SECTION = 3
 176         _REST_SECTION = 4
 177         _MAX_SECTIONS = 5
 178
 179         _NO_ELAPSED = datetime.timedelta(hours=1)
 180         _WEEK_ELAPSED = datetime.timedelta(weeks=1)
 181         _MONTH_ELAPSED = datetime.timedelta(days=30)
 182
 183         def __init__(self, parentItem):
 184                 self._timeItems = [
 185                         QtGui.QStandardItem(description)
 186                         for (i, description) in zip(
 187                                 xrange(self._MAX_SECTIONS),
 188                                 ["Now", "Today", "Week", "Month", "Past"],
 189                         )
 190                 ]
 191                 for item in self._timeItems:
 192                         item.setEditable(False)
 193                         item.setCheckable(False)
 194                         row = (item, )
 195                         parentItem.appendRow(row)
 196
 197                 self._today = datetime.datetime(1900, 1, 1)
 198
 199                 self.prepare_for_update(self._today)
 200
 201         def prepare_for_update(self, newToday):
 202                 self._today = newToday
 203                 for item in self._timeItems:
 204                         item.removeRows(0, item.rowCount())
 205                 try:
 206                         hour = self._today.strftime("%X")
 207                         day = self._today.strftime("%x")
 208                 except ValueError:
 209                         _moduleLogger.exception("Can't format times")
 210                         hour = "Now"
 211                         day = "Today"
 212                 self._timeItems[self._NOW_SECTION].setText(hour)
 213                 self._timeItems[self._TODAY_SECTION].setText(day)
 214
 215         def add_row(self, rowDate, row):
 216                 elapsedTime = self._today - rowDate
 217                 todayTuple = self._today.timetuple()
 218                 rowTuple = rowDate.timetuple()
 219                 if elapsedTime < self._NO_ELAPSED:
 220                         section = self._NOW_SECTION
 221                 elif todayTuple[0:3] == rowTuple[0:3]:
 222                         section = self._TODAY_SECTION
 223                 elif elapsedTime < self._WEEK_ELAPSED:
 224                         section = self._WEEK_SECTION
 225                 elif elapsedTime < self._MONTH_ELAPSED:
 226                         section = self._MONTH_SECTION
 227                 else:
 228                         section = self._REST_SECTION
 229                 self._timeItems[section].appendRow(row)
 230
 231         def get_item(self, timeIndex, rowIndex, column):
 232                 timeItem = self._timeItems[timeIndex]
 233                 item = timeItem.child(rowIndex, column)
 234                 return item
 235
 236
 237 class History(object):
 238
 239         DETAILS_IDX = 0
 240         FROM_IDX = 1
 241         MAX_IDX = 2
 242
 243         HISTORY_ITEM_TYPES = ["Received", "Missed", "Placed", "All"]
 244         HISTORY_COLUMNS = ["Details", "From"]
 245         assert len(HISTORY_COLUMNS) == MAX_IDX
 246
 247         def __init__(self, app, session, errorLog):
 248                 self._selectedFilter = self.HISTORY_ITEM_TYPES[-1]
 249                 self._app = app
 250                 self._session = session
 251                 self._session.historyUpdated.connect(self._on_history_updated)
 252                 self._errorLog = errorLog
 253
 254                 self._typeSelection = QtGui.QComboBox()
 255                 self._typeSelection.addItems(self.HISTORY_ITEM_TYPES)
 256                 self._typeSelection.setCurrentIndex(
 257                         self.HISTORY_ITEM_TYPES.index(self._selectedFilter)
 258                 )
 259                 self._typeSelection.currentIndexChanged[str].connect(self._on_filter_changed)
 260
 261                 self._itemStore = QtGui.QStandardItemModel()
 262                 self._itemStore.setHorizontalHeaderLabels(self.HISTORY_COLUMNS)
 263                 self._categoryManager = TimeCategories(self._itemStore)
 264
 265                 self._itemView = QtGui.QTreeView()
 266                 self._itemView.setModel(self._itemStore)
 267                 self._itemView.setUniformRowHeights(True)
 268                 self._itemView.setRootIsDecorated(False)
 269                 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
 270                 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
 271                 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
 272                 self._itemView.setHeaderHidden(True)
 273                 self._itemView.setItemsExpandable(False)
 274                 self._itemView.header().setResizeMode(QtGui.QHeaderView.ResizeToContents)
 275                 self._itemView.activated.connect(self._on_row_activated)
 276
 277                 self._layout = QtGui.QVBoxLayout()
 278                 self._layout.addWidget(self._typeSelection)
 279                 self._layout.addWidget(self._itemView)
 280                 self._widget = QtGui.QWidget()
 281                 self._widget.setLayout(self._layout)
 282
 283                 self._populate_items()
 284
 285         @property
 286         def toplevel(self):
 287                 return self._widget
 288
 289         def enable(self):
 290                 self._itemView.setEnabled(True)
 291
 292         def disable(self):
 293                 self._itemView.setEnabled(False)
 294
 295         def get_settings(self):
 296                 return {
 297                         "filter": self._selectedFilter,
 298                 }
 299
 300         def set_settings(self, settings):
 301                 selectedFilter = settings.get("filter", self.HISTORY_ITEM_TYPES[-1])
 302                 if selectedFilter in self.HISTORY_ITEM_TYPES:
 303                         self._selectedFilter = selectedFilter
 304                         self._typeSelection.setCurrentIndex(
 305                                 self.HISTORY_ITEM_TYPES.index(selectedFilter)
 306                         )
 307
 308         def clear(self):
 309                 self._itemView.clear()
 310
 311         def refresh(self, force=True):
 312                 self._session.update_history(force)
 313
 314         def _populate_items(self):
 315                 self._categoryManager.prepare_for_update(self._session.get_when_history_updated())
 316
 317                 history = self._session.get_history()
 318                 history.sort(key=lambda item: item["time"], reverse=True)
 319                 for event in history:
 320                         if self._selectedFilter not in [self.HISTORY_ITEM_TYPES[-1], event["action"]]:
 321                                 continue
 322
 323                         relTime = misc_utils.abbrev_relative_date(event["relTime"])
 324                         action = event["action"]
 325                         number = event["number"]
 326                         prettyNumber = misc_utils.make_pretty(number)
 327                         name = event["name"]
 328                         if not name or name == number:
 329                                 name = event["location"]
 330                         if not name:
 331                                 name = "Unknown"
 332
 333                         detailsItem = QtGui.QStandardItem("%s - %s\n%s" % (relTime, action, prettyNumber))
 334                         detailsFont = detailsItem.font()
 335                         detailsFont.setPointSize(max(detailsFont.pointSize() - 7, 5))
 336                         detailsItem.setFont(detailsFont)
 337                         nameItem = QtGui.QStandardItem(name)
 338                         nameFont = nameItem.font()
 339                         nameFont.setPointSize(nameFont.pointSize() + 4)
 340                         nameItem.setFont(nameFont)
 341                         row = detailsItem, nameItem
 342                         for item in row:
 343                                 item.setEditable(False)
 344                                 item.setCheckable(False)
 345                         row[0].setData(event)
 346                         self._categoryManager.add_row(event["time"], row)
 347                 self._itemView.expandAll()
 348
 349         @QtCore.pyqtSlot(str)
 350         @misc_utils.log_exception(_moduleLogger)
 351         def _on_filter_changed(self, newItem):
 352                 with qui_utils.notify_error(self._errorLog):
 353                         self._selectedFilter = str(newItem)
 354                         self._populate_items()
 355
 356         @QtCore.pyqtSlot()
 357         @misc_utils.log_exception(_moduleLogger)
 358         def _on_history_updated(self):
 359                 with qui_utils.notify_error(self._errorLog):
 360                         self._populate_items()
 361
 362         @QtCore.pyqtSlot(QtCore.QModelIndex)
 363         @misc_utils.log_exception(_moduleLogger)
 364         def _on_row_activated(self, index):
 365                 with qui_utils.notify_error(self._errorLog):
 366                         timeIndex = index.parent()
 367                         assert timeIndex.isValid(), "Invalid row"
 368                         timeRow = timeIndex.row()
 369                         row = index.row()
 370                         detailsItem = self._categoryManager.get_item(timeRow, row, self.DETAILS_IDX)
 371                         fromItem = self._categoryManager.get_item(timeRow, row, self.FROM_IDX)
 372                         contactDetails = detailsItem.data().toPyObject()
 373
 374                         title = str(fromItem.text())
 375                         number = str(contactDetails[QtCore.QString("number")])
 376                         contactId = number # ids don't seem too unique so using numbers
 377
 378                         descriptionRows = []
 379                         for t in xrange(self._itemStore.rowCount()):
 380                                 randomTimeItem = self._itemStore.item(t, 0)
 381                                 for i in xrange(randomTimeItem.rowCount()):
 382                                         iItem = randomTimeItem.child(i, 0)
 383                                         iContactDetails = iItem.data().toPyObject()
 384                                         iNumber = str(iContactDetails[QtCore.QString("number")])
 385                                         if number != iNumber:
 386                                                 continue
 387                                         relTime = misc_utils.abbrev_relative_date(iContactDetails[QtCore.QString("relTime")])
 388                                         action = str(iContactDetails[QtCore.QString("action")])
 389                                         number = str(iContactDetails[QtCore.QString("number")])
 390                                         prettyNumber = misc_utils.make_pretty(number)
 391                                         rowItems = relTime, action, prettyNumber
 392                                         descriptionRows.append("<tr><td>%s</td></tr>" % "</td><td>".join(rowItems))
 393                         description = "<table>%s</table>" % "".join(descriptionRows)
 394                         numbersWithDescriptions = [(str(contactDetails[QtCore.QString("number")]), "")]
 395                         self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
 396
 397
 398 class Messages(object):
 399
 400         NO_MESSAGES = "None"
 401         VOICEMAIL_MESSAGES = "Voicemail"
 402         TEXT_MESSAGES = "SMS"
 403         ALL_TYPES = "All Messages"
 404         MESSAGE_TYPES = [NO_MESSAGES, VOICEMAIL_MESSAGES, TEXT_MESSAGES, ALL_TYPES]
 405
 406         UNREAD_STATUS = "Unread"
 407         UNARCHIVED_STATUS = "Inbox"
 408         ALL_STATUS = "Any"
 409         MESSAGE_STATUSES = [UNREAD_STATUS, UNARCHIVED_STATUS, ALL_STATUS]
 410
 411         _MIN_MESSAGES_SHOWN = 4
 412
 413         def __init__(self, app, session, errorLog):
 414                 self._selectedTypeFilter = self.ALL_TYPES
 415                 self._selectedStatusFilter = self.ALL_STATUS
 416                 self._app = app
 417                 self._session = session
 418                 self._session.messagesUpdated.connect(self._on_messages_updated)
 419                 self._errorLog = errorLog
 420
 421                 self._typeSelection = QtGui.QComboBox()
 422                 self._typeSelection.addItems(self.MESSAGE_TYPES)
 423                 self._typeSelection.setCurrentIndex(
 424                         self.MESSAGE_TYPES.index(self._selectedTypeFilter)
 425                 )
 426                 self._typeSelection.currentIndexChanged[str].connect(self._on_type_filter_changed)
 427
 428                 self._statusSelection = QtGui.QComboBox()
 429                 self._statusSelection.addItems(self.MESSAGE_STATUSES)
 430                 self._statusSelection.setCurrentIndex(
 431                         self.MESSAGE_STATUSES.index(self._selectedStatusFilter)
 432                 )
 433                 self._statusSelection.currentIndexChanged[str].connect(self._on_status_filter_changed)
 434
 435                 self._selectionLayout = QtGui.QHBoxLayout()
 436                 self._selectionLayout.addWidget(self._typeSelection)
 437                 self._selectionLayout.addWidget(self._statusSelection)
 438
 439                 self._itemStore = QtGui.QStandardItemModel()
 440                 self._itemStore.setHorizontalHeaderLabels(["Messages"])
 441                 self._categoryManager = TimeCategories(self._itemStore)
 442
 443                 self._htmlDelegate = qui_utils.QHtmlDelegate()
 444                 self._itemView = QtGui.QTreeView()
 445                 self._itemView.setModel(self._itemStore)
 446                 self._itemView.setUniformRowHeights(False)
 447                 self._itemView.setRootIsDecorated(False)
 448                 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
 449                 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
 450                 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
 451                 self._itemView.setHeaderHidden(True)
 452                 self._itemView.setItemsExpandable(False)
 453                 self._itemView.setItemDelegate(self._htmlDelegate)
 454                 self._itemView.activated.connect(self._on_row_activated)
 455
 456                 self._layout = QtGui.QVBoxLayout()
 457                 self._layout.addLayout(self._selectionLayout)
 458                 self._layout.addWidget(self._itemView)
 459                 self._widget = QtGui.QWidget()
 460                 self._widget.setLayout(self._layout)
 461
 462                 self._populate_items()
 463
 464         @property
 465         def toplevel(self):
 466                 return self._widget
 467
 468         def enable(self):
 469                 self._itemView.setEnabled(True)
 470
 471         def disable(self):
 472                 self._itemView.setEnabled(False)
 473
 474         def get_settings(self):
 475                 return {
 476                         "type": self._selectedTypeFilter,
 477                         "status": self._selectedStatusFilter,
 478                 }
 479
 480         def set_settings(self, settings):
 481                 selectedType = settings.get("type", self.ALL_TYPES)
 482                 if selectedType in self.MESSAGE_TYPES:
 483                         self._selectedTypeFilter = selectedType
 484                         self._typeSelection.setCurrentIndex(
 485                                 self.MESSAGE_TYPES.index(self._selectedTypeFilter)
 486                         )
 487
 488                 selectedStatus = settings.get("status", self.ALL_STATUS)
 489                 if selectedStatus in self.MESSAGE_STATUSES:
 490                         self._selectedStatusFilter = selectedStatus
 491                         self._statusSelection.setCurrentIndex(
 492                                 self.MESSAGE_STATUSES.index(self._selectedStatusFilter)
 493                         )
 494
 495         def clear(self):
 496                 self._itemView.clear()
 497
 498         def refresh(self, force=True):
 499                 self._session.update_messages(force)
 500
 501         def _populate_items(self):
 502                 self._categoryManager.prepare_for_update(self._session.get_when_messages_updated())
 503
 504                 rawMessages = self._session.get_messages()
 505                 rawMessages.sort(key=lambda item: item["time"], reverse=True)
 506                 for item in rawMessages:
 507                         isUnarchived = not item["isArchived"]
 508                         isUnread = not item["isRead"]
 509                         visibleStatus = {
 510                                 self.UNREAD_STATUS: isUnarchived and isUnread,
 511                                 self.UNARCHIVED_STATUS: isUnarchived,
 512                                 self.ALL_STATUS: True,
 513                         }[self._selectedStatusFilter]
 514                         visibleType = self._selectedTypeFilter in [item["type"], self.ALL_TYPES]
 515                         if not (visibleType and visibleStatus):
 516                                 continue
 517
 518                         relTime = misc_utils.abbrev_relative_date(item["relTime"])
 519                         number = item["number"]
 520                         prettyNumber = misc_utils.make_pretty(number)
 521                         name = item["name"]
 522                         if not name or name == number:
 523                                 name = item["location"]
 524                         if not name:
 525                                 name = "Unknown"
 526
 527                         messageParts = list(item["messageParts"])
 528                         if len(messageParts) == 0:
 529                                 messages = ("No Transcription", )
 530                         elif len(messageParts) == 1:
 531                                 if messageParts[0][1]:
 532                                         messages = (messageParts[0][1], )
 533                                 else:
 534                                         messages = ("No Transcription", )
 535                         else:
 536                                 messages = [
 537                                         "<b>%s</b>: %s" % (messagePart[0], messagePart[1])
 538                                         for messagePart in messageParts
 539                                 ]
 540
 541                         firstMessage = "<b>%s - %s</b> <i>(%s)</i>" % (name, prettyNumber, relTime)
 542
 543                         expandedMessages = [firstMessage]
 544                         expandedMessages.extend(messages)
 545                         if (self._MIN_MESSAGES_SHOWN + 1) < len(messages):
 546                                 secondMessage = "<i>%d Messages Hidden...</i>" % (len(messages) - self._MIN_MESSAGES_SHOWN, )
 547                                 collapsedMessages = [firstMessage, secondMessage]
 548                                 collapsedMessages.extend(messages[-(self._MIN_MESSAGES_SHOWN+0):])
 549                         else:
 550                                 collapsedMessages = expandedMessages
 551
 552                         item = dict(item.iteritems())
 553                         item["collapsedMessages"] = "<br/>\n".join(collapsedMessages)
 554                         item["expandedMessages"] = "<br/>\n".join(expandedMessages)
 555
 556                         messageItem = QtGui.QStandardItem(item["collapsedMessages"])
 557                         messageItem.setData(item)
 558                         messageItem.setEditable(False)
 559                         messageItem.setCheckable(False)
 560                         row = (messageItem, )
 561                         self._categoryManager.add_row(item["time"], row)
 562                 self._itemView.expandAll()
 563
 564         @QtCore.pyqtSlot(str)
 565         @misc_utils.log_exception(_moduleLogger)
 566         def _on_type_filter_changed(self, newItem):
 567                 with qui_utils.notify_error(self._errorLog):
 568                         self._selectedTypeFilter = str(newItem)
 569                         self._populate_items()
 570
 571         @QtCore.pyqtSlot(str)
 572         @misc_utils.log_exception(_moduleLogger)
 573         def _on_status_filter_changed(self, newItem):
 574                 with qui_utils.notify_error(self._errorLog):
 575                         self._selectedStatusFilter = str(newItem)
 576                         self._populate_items()
 577
 578         @QtCore.pyqtSlot()
 579         @misc_utils.log_exception(_moduleLogger)
 580         def _on_messages_updated(self):
 581                 with qui_utils.notify_error(self._errorLog):
 582                         self._populate_items()
 583
 584         @QtCore.pyqtSlot(QtCore.QModelIndex)
 585         @misc_utils.log_exception(_moduleLogger)
 586         def _on_row_activated(self, index):
 587                 with qui_utils.notify_error(self._errorLog):
 588                         timeIndex = index.parent()
 589                         assert timeIndex.isValid(), "Invalid row"
 590                         timeRow = timeIndex.row()
 591                         row = index.row()
 592                         item = self._categoryManager.get_item(timeRow, row, 0)
 593                         contactDetails = item.data().toPyObject()
 594
 595                         name = str(contactDetails[QtCore.QString("name")])
 596                         number = str(contactDetails[QtCore.QString("number")])
 597                         if not name or name == number:
 598                                 name = str(contactDetails[QtCore.QString("location")])
 599                         if not name:
 600                                 name = "Unknown"
 601
 602                         contactId = str(contactDetails[QtCore.QString("id")])
 603                         title = name
 604                         description = str(contactDetails[QtCore.QString("expandedMessages")])
 605                         numbersWithDescriptions = [(number, "")]
 606                         self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
 607
 608
 609 class Contacts(object):
 610
 611         # @todo Provide some sort of letter jump
 612
 613         def __init__(self, app, session, errorLog):
 614                 self._app = app
 615                 self._session = session
 616                 self._session.contactsUpdated.connect(self._on_contacts_updated)
 617                 self._errorLog = errorLog
 618                 self._addressBookFactories = [
 619                         null_backend.NullAddressBookFactory(),
 620                         file_backend.FilesystemAddressBookFactory(app.fsContactsPath),
 621                 ]
 622                 self._addressBooks = []
 623
 624                 self._listSelection = QtGui.QComboBox()
 625                 self._listSelection.addItems([])
 626                 self._listSelection.currentIndexChanged[str].connect(self._on_filter_changed)
 627                 self._activeList = "None"
 628
 629                 self._itemStore = QtGui.QStandardItemModel()
 630                 self._itemStore.setHorizontalHeaderLabels(["Contacts"])
 631                 self._alphaItem = {}
 632
 633                 self._itemView = QtGui.QTreeView()
 634                 self._itemView.setModel(self._itemStore)
 635                 self._itemView.setUniformRowHeights(True)
 636                 self._itemView.setRootIsDecorated(False)
 637                 self._itemView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
 638                 self._itemView.setSelectionBehavior(QtGui.QAbstractItemView.SelectRows)
 639                 self._itemView.setSelectionMode(QtGui.QAbstractItemView.SingleSelection)
 640                 self._itemView.setHeaderHidden(True)
 641                 self._itemView.setItemsExpandable(False)
 642                 self._itemView.activated.connect(self._on_row_activated)
 643
 644                 self._layout = QtGui.QVBoxLayout()
 645                 self._layout.addWidget(self._listSelection)
 646                 self._layout.addWidget(self._itemView)
 647                 self._widget = QtGui.QWidget()
 648                 self._widget.setLayout(self._layout)
 649
 650                 self.update_addressbooks()
 651                 self._populate_items()
 652
 653         @property
 654         def toplevel(self):
 655                 return self._widget
 656
 657         def enable(self):
 658                 self._itemView.setEnabled(True)
 659
 660         def disable(self):
 661                 self._itemView.setEnabled(False)
 662
 663         def get_settings(self):
 664                 return {
 665                         "selectedAddressbook": self._activeList,
 666                 }
 667
 668         def set_settings(self, settings):
 669                 currentItem = settings.get("selectedAddressbook", "None")
 670                 bookNames = [book["name"] for book in self._addressBooks]
 671                 try:
 672                         newIndex = bookNames.index(currentItem)
 673                 except ValueError:
 674                         # Switch over to None for the user
 675                         newIndex = 0
 676                 self._listSelection.setCurrentIndex(newIndex)
 677                 self._activeList = currentItem
 678
 679         def clear(self):
 680                 self._itemView.clear()
 681
 682         def refresh(self, force=True):
 683                 self._backend.update_contacts(force)
 684
 685         @property
 686         def _backend(self):
 687                 return self._addressBooks[self._listSelection.currentIndex()]["book"]
 688
 689         def update_addressbooks(self):
 690                 self._addressBooks = [
 691                         {"book": book, "name": book.name}
 692                         for factory in self._addressBookFactories
 693                         for book in factory.get_addressbooks()
 694                 ]
 695                 self._addressBooks.append(
 696                         {
 697                                 "book": self._session,
 698                                 "name": "Google Voice",
 699                         }
 700                 )
 701
 702                 currentItem = str(self._listSelection.currentText())
 703                 self._activeList = currentItem
 704                 if currentItem == "":
 705                         # Not loaded yet
 706                         currentItem = "None"
 707                 self._listSelection.clear()
 708                 bookNames = [book["name"] for book in self._addressBooks]
 709                 try:
 710                         newIndex = bookNames.index(currentItem)
 711                 except ValueError:
 712                         # Switch over to None for the user
 713                         newIndex = 0
 714                         self._itemStore.clear()
 715                         _moduleLogger.info("Addressbook %r doesn't exist anymore, switching to None" % currentItem)
 716                 self._listSelection.addItems(bookNames)
 717                 self._listSelection.setCurrentIndex(newIndex)
 718
 719         def _populate_items(self):
 720                 self._itemStore.clear()
 721                 self._alphaItem = dict(
 722                         (letter, QtGui.QStandardItem(letter))
 723                         for letter in self._prefixes()
 724                 )
 725                 for letter in self._prefixes():
 726                         item = self._alphaItem[letter]
 727                         item.setEditable(False)
 728                         item.setCheckable(False)
 729                         row = (item, )
 730                         self._itemStore.appendRow(row)
 731
 732                 for item in self._get_contacts():
 733                         name = item["name"]
 734                         if not name:
 735                                 name = "Unknown"
 736                         numbers = item["numbers"]
 737
 738                         nameItem = QtGui.QStandardItem(name)
 739                         nameItem.setEditable(False)
 740                         nameItem.setCheckable(False)
 741                         nameItem.setData(item)
 742                         nameItemFont = nameItem.font()
 743                         nameItemFont.setPointSize(max(nameItemFont.pointSize() + 4, 5))
 744                         nameItem.setFont(nameItemFont)
 745
 746                         row = (nameItem, )
 747                         rowKey = name[0].upper()
 748                         rowKey = rowKey if rowKey in self._alphaItem else "#"
 749                         self._alphaItem[rowKey].appendRow(row)
 750                 self._itemView.expandAll()
 751
 752         def _prefixes(self):
 753                 return itertools.chain(string.ascii_uppercase, ("#", ))
 754
 755         def _jump_to_prefix(self, letter):
 756                 i = list(self._prefixes()).index(letter)
 757                 rootIndex = self._itemView.rootIndex()
 758                 currentIndex = self._itemView.model().index(i, 0, rootIndex)
 759                 self._itemView.scrollTo(currentIndex)
 760                 self._itemView.setItemSelected(self._itemView.topLevelItem(i), True)
 761
 762         def _get_contacts(self):
 763                 contacts = list(self._backend.get_contacts().itervalues())
 764                 contacts.sort(key=lambda contact: contact["name"].lower())
 765                 return contacts
 766
 767         @QtCore.pyqtSlot(str)
 768         @misc_utils.log_exception(_moduleLogger)
 769         def _on_filter_changed(self, newItem):
 770                 with qui_utils.notify_error(self._errorLog):
 771                         self._activeList = str(newItem)
 772                         self.refresh(force=False)
 773                         self._populate_items()
 774
 775         @QtCore.pyqtSlot()
 776         @misc_utils.log_exception(_moduleLogger)
 777         def _on_contacts_updated(self):
 778                 with qui_utils.notify_error(self._errorLog):
 779                         self._populate_items()
 780
 781         @QtCore.pyqtSlot(QtCore.QModelIndex)
 782         @misc_utils.log_exception(_moduleLogger)
 783         def _on_row_activated(self, index):
 784                 with qui_utils.notify_error(self._errorLog):
 785                         letterIndex = index.parent()
 786                         assert letterIndex.isValid(), "Invalid row"
 787                         letterRow = letterIndex.row()
 788                         letter = list(self._prefixes())[letterRow]
 789                         letterItem = self._alphaItem[letter]
 790                         rowIndex = index.row()
 791                         item = letterItem.child(rowIndex, 0)
 792                         contactDetails = item.data().toPyObject()
 793
 794                         name = str(contactDetails[QtCore.QString("name")])
 795                         if not name:
 796                                 name = str(contactDetails[QtCore.QString("location")])
 797                         if not name:
 798                                 name = "Unknown"
 799
 800                         contactId = str(contactDetails[QtCore.QString("contactId")])
 801                         numbers = contactDetails[QtCore.QString("numbers")]
 802                         numbers = [
 803                                 dict(
 804                                         (str(k), str(v))
 805                                         for (k, v) in number.iteritems()
 806                                 )
 807                                 for number in numbers
 808                         ]
 809                         numbersWithDescriptions = [
 810                                 (
 811                                         number["phoneNumber"],
 812                                         self._choose_phonetype(number),
 813                                 )
 814                                 for number in numbers
 815                         ]
 816                         title = name
 817                         description = name
 818                         self._session.draft.add_contact(contactId, title, description, numbersWithDescriptions)
 819
 820         @staticmethod
 821         def _choose_phonetype(numberDetails):
 822                 if "phoneTypeName" in numberDetails:
 823                         return numberDetails["phoneTypeName"]
 824                 elif "phoneType" in numberDetails:
 825                         return numberDetails["phoneType"]
 826                 else:
 827                         return ""