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