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