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

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