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