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