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