Scrolling work
[gc-dialer] / src / dialogs.py
1 #!/usr/bin/env python
2
3 from __future__ import with_statement
4 from __future__ import division
5
6 import functools
7 import copy
8 import logging
9
10 from PyQt4 import QtGui
11 from PyQt4 import QtCore
12
13 from util import qui_utils
14 from util import misc as misc_utils
15
16
17 _moduleLogger = logging.getLogger(__name__)
18
19
20 class CredentialsDialog(object):
21
22         def __init__(self, app):
23                 self._usernameField = QtGui.QLineEdit()
24                 self._passwordField = QtGui.QLineEdit()
25                 self._passwordField.setEchoMode(QtGui.QLineEdit.PasswordEchoOnEdit)
26
27                 self._credLayout = QtGui.QGridLayout()
28                 self._credLayout.addWidget(QtGui.QLabel("Username"), 0, 0)
29                 self._credLayout.addWidget(self._usernameField, 0, 1)
30                 self._credLayout.addWidget(QtGui.QLabel("Password"), 1, 0)
31                 self._credLayout.addWidget(self._passwordField, 1, 1)
32
33                 self._loginButton = QtGui.QPushButton("&Login")
34                 self._buttonLayout = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Cancel)
35                 self._buttonLayout.addButton(self._loginButton, QtGui.QDialogButtonBox.AcceptRole)
36
37                 self._layout = QtGui.QVBoxLayout()
38                 self._layout.addLayout(self._credLayout)
39                 self._layout.addWidget(self._buttonLayout)
40
41                 self._dialog = QtGui.QDialog()
42                 self._dialog.setWindowTitle("Login")
43                 self._dialog.setLayout(self._layout)
44                 self._dialog.setAttribute(QtCore.Qt.WA_DeleteOnClose, False)
45                 qui_utils.set_autorient(self._dialog, True)
46                 self._buttonLayout.accepted.connect(self._dialog.accept)
47                 self._buttonLayout.rejected.connect(self._dialog.reject)
48
49                 self._closeWindowAction = QtGui.QAction(None)
50                 self._closeWindowAction.setText("Close")
51                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
52                 self._closeWindowAction.triggered.connect(self._on_close_window)
53
54                 self._dialog.addAction(self._closeWindowAction)
55                 self._dialog.addAction(app.quitAction)
56                 self._dialog.addAction(app.fullscreenAction)
57
58         def run(self, defaultUsername, defaultPassword, parent=None):
59                 self._dialog.setParent(parent, QtCore.Qt.Dialog)
60                 try:
61                         self._usernameField.setText(defaultUsername)
62                         self._passwordField.setText(defaultPassword)
63
64                         response = self._dialog.exec_()
65                         if response == QtGui.QDialog.Accepted:
66                                 return str(self._usernameField.text()), str(self._passwordField.text())
67                         elif response == QtGui.QDialog.Rejected:
68                                 raise RuntimeError("Login Cancelled")
69                         else:
70                                 raise RuntimeError("Unknown Response")
71                 finally:
72                         self._dialog.setParent(None, QtCore.Qt.Dialog)
73
74         @QtCore.pyqtSlot()
75         @QtCore.pyqtSlot(bool)
76         @misc_utils.log_exception(_moduleLogger)
77         def _on_close_window(self, checked = True):
78                 self._dialog.reject()
79
80
81 class AccountDialog(object):
82
83         # @bug Can't configure callback number
84
85         def __init__(self, app):
86                 self._doClear = False
87
88                 self._accountNumberLabel = QtGui.QLabel("NUMBER NOT SET")
89                 self._clearButton = QtGui.QPushButton("Clear Account")
90                 self._clearButton.clicked.connect(self._on_clear)
91
92                 self._credLayout = QtGui.QGridLayout()
93                 self._credLayout.addWidget(QtGui.QLabel("Account"), 0, 0)
94                 self._credLayout.addWidget(self._accountNumberLabel, 0, 1)
95                 self._credLayout.addWidget(QtGui.QLabel("Callback"), 1, 0)
96                 self._credLayout.addWidget(QtGui.QLabel(""), 2, 0)
97                 self._credLayout.addWidget(self._clearButton, 2, 1)
98                 self._credLayout.addWidget(QtGui.QLabel(""), 3, 0)
99
100                 self._loginButton = QtGui.QPushButton("&Apply")
101                 self._buttonLayout = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Cancel)
102                 self._buttonLayout.addButton(self._loginButton, QtGui.QDialogButtonBox.AcceptRole)
103
104                 self._layout = QtGui.QVBoxLayout()
105                 self._layout.addLayout(self._credLayout)
106                 self._layout.addWidget(self._buttonLayout)
107
108                 self._dialog = QtGui.QDialog()
109                 self._dialog.setWindowTitle("Login")
110                 self._dialog.setLayout(self._layout)
111                 qui_utils.set_autorient(self._dialog, True)
112                 self._buttonLayout.accepted.connect(self._dialog.accept)
113                 self._buttonLayout.rejected.connect(self._dialog.reject)
114
115                 self._closeWindowAction = QtGui.QAction(None)
116                 self._closeWindowAction.setText("Close")
117                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
118                 self._closeWindowAction.triggered.connect(self._on_close_window)
119
120                 self._dialog.addAction(self._closeWindowAction)
121                 self._dialog.addAction(app.quitAction)
122                 self._dialog.addAction(app.fullscreenAction)
123
124         @property
125         def doClear(self):
126                 return self._doClear
127
128         accountNumber = property(
129                 lambda self: str(self._accountNumberLabel.text()),
130                 lambda self, num: self._accountNumberLabel.setText(num),
131         )
132
133         def run(self, parent=None):
134                 self._doClear = False
135                 self._dialog.setParent(parent)
136
137                 response = self._dialog.exec_()
138                 return response
139
140         @QtCore.pyqtSlot()
141         @QtCore.pyqtSlot(bool)
142         def _on_clear(self, checked = False):
143                 self._doClear = True
144                 self._dialog.accept()
145
146         @QtCore.pyqtSlot()
147         @QtCore.pyqtSlot(bool)
148         @misc_utils.log_exception(_moduleLogger)
149         def _on_close_window(self, checked = True):
150                 self._dialog.reject()
151
152
153 class SMSEntryWindow(object):
154
155         MAX_CHAR = 160
156
157         # @bug on n900 no scrolling for history qtextedit
158
159         def __init__(self, parent, app, session, errorLog):
160                 self._session = session
161                 self._session.draft.recipientsChanged.connect(self._on_recipients_changed)
162                 self._session.draft.called.connect(self._on_op_finished)
163                 self._session.draft.sentMessage.connect(self._on_op_finished)
164                 self._session.draft.cancelled.connect(self._on_op_finished)
165                 self._errorLog = errorLog
166
167                 self._targetLayout = QtGui.QVBoxLayout()
168                 self._targetList = QtGui.QWidget()
169                 self._targetList.setLayout(self._targetLayout)
170                 self._history = QtGui.QTextEdit()
171                 self._history.setReadOnly(True)
172                 self._history.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignBottom)
173                 self._history.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
174                 self._history.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
175                 self._smsEntry = QtGui.QTextEdit()
176                 self._smsEntry.textChanged.connect(self._on_letter_count_changed)
177
178                 self._entryLayout = QtGui.QVBoxLayout()
179                 self._entryLayout.addWidget(self._targetList)
180                 self._entryLayout.addWidget(self._history)
181                 self._entryLayout.addWidget(self._smsEntry)
182                 self._entryLayout.setContentsMargins(0, 0, 0, 0)
183                 self._entryWidget = QtGui.QWidget()
184                 self._entryWidget.setLayout(self._entryLayout)
185                 self._entryWidget.setContentsMargins(0, 0, 0, 0)
186                 self._scrollEntry = QtGui.QScrollArea()
187                 self._scrollEntry.setWidget(self._entryWidget)
188                 self._scrollEntry.setWidgetResizable(True)
189                 self._scrollEntry.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignBottom)
190                 self._scrollEntry.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
191                 self._scrollEntry.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
192
193                 self._characterCountLabel = QtGui.QLabel("0 (0)")
194                 self._singleNumberSelector = QtGui.QComboBox()
195                 self._smsButton = QtGui.QPushButton("SMS")
196                 self._smsButton.clicked.connect(self._on_sms_clicked)
197                 self._dialButton = QtGui.QPushButton("Dial")
198                 self._dialButton.clicked.connect(self._on_call_clicked)
199
200                 self._buttonLayout = QtGui.QHBoxLayout()
201                 self._buttonLayout.addWidget(self._characterCountLabel)
202                 self._buttonLayout.addWidget(self._singleNumberSelector)
203                 self._buttonLayout.addWidget(self._smsButton)
204                 self._buttonLayout.addWidget(self._dialButton)
205
206                 self._layout = QtGui.QVBoxLayout()
207                 self._layout.addWidget(self._scrollEntry)
208                 self._layout.addLayout(self._buttonLayout)
209
210                 centralWidget = QtGui.QWidget()
211                 centralWidget.setLayout(self._layout)
212
213                 self._window = QtGui.QMainWindow(parent)
214                 qui_utils.set_autorient(self._window, True)
215                 qui_utils.set_stackable(self._window, True)
216                 self._window.setWindowTitle("Contact")
217                 self._window.setCentralWidget(centralWidget)
218
219                 self._closeWindowAction = QtGui.QAction(None)
220                 self._closeWindowAction.setText("Close")
221                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
222                 self._closeWindowAction.triggered.connect(self._on_close_window)
223
224                 fileMenu = self._window.menuBar().addMenu("&File")
225                 fileMenu.addAction(self._closeWindowAction)
226                 fileMenu.addAction(app.quitAction)
227                 viewMenu = self._window.menuBar().addMenu("&View")
228                 viewMenu.addAction(app.fullscreenAction)
229
230                 self._window.show()
231                 self._update_recipients()
232
233         def _update_letter_count(self):
234                 count = self._smsEntry.toPlainText().size()
235                 numTexts, numCharInText = divmod(count, self.MAX_CHAR)
236                 numTexts += 1
237                 numCharsLeftInText = self.MAX_CHAR - numCharInText
238                 self._characterCountLabel.setText("%d (%d)" % (numCharsLeftInText, numTexts))
239
240         def _update_button_state(self):
241                 if self._session.draft.get_num_contacts() == 0:
242                         self._dialButton.setEnabled(False)
243                         self._smsButton.setEnabled(False)
244                 elif self._session.draft.get_num_contacts() == 1:
245                         count = self._smsEntry.toPlainText().size()
246                         if count == 0:
247                                 self._dialButton.setEnabled(True)
248                                 self._smsButton.setEnabled(False)
249                         else:
250                                 self._dialButton.setEnabled(False)
251                                 self._smsButton.setEnabled(True)
252                 else:
253                         self._dialButton.setEnabled(False)
254                         self._smsButton.setEnabled(True)
255
256         def _update_recipients(self):
257                 draftContactsCount = self._session.draft.get_num_contacts()
258                 if draftContactsCount == 0:
259                         self._window.hide()
260                 elif draftContactsCount == 1:
261                         (cid, ) = self._session.draft.get_contacts()
262                         title = self._session.draft.get_title(cid)
263                         description = self._session.draft.get_description(cid)
264                         numbers = self._session.draft.get_numbers(cid)
265
266                         self._targetList.setVisible(False)
267                         if description:
268                                 self._history.setHtml(description)
269                                 self._history.setVisible(True)
270                         else:
271                                 self._history.setHtml("")
272                                 self._history.setVisible(False)
273                         self._populate_number_selector(self._singleNumberSelector, cid, numbers)
274
275                         self._scroll_to_bottom()
276                         self._window.setWindowTitle(title)
277                         self._window.show()
278                         self._smsEntry.setFocus(QtCore.Qt.OtherFocusReason)
279                 else:
280                         self._targetList.setVisible(True)
281                         while self._targetLayout.count():
282                                 removedLayoutItem = self._targetLayout.takeAt(self._targetLayout.count()-1)
283                                 removedWidget = removedLayoutItem.widget()
284                                 removedWidget.close()
285                         for cid in self._session.draft.get_contacts():
286                                 title = self._session.draft.get_title(cid)
287                                 description = self._session.draft.get_description(cid)
288                                 numbers = self._session.draft.get_numbers(cid)
289
290                                 titleLabel = QtGui.QLabel(title)
291                                 numberSelector = QtGui.QComboBox()
292                                 self._populate_number_selector(numberSelector, cid, numbers)
293                                 deleteButton = QtGui.QPushButton("Delete")
294                                 callback = functools.partial(
295                                         self._on_remove_contact,
296                                         cid
297                                 )
298                                 callback.__name__ = "b"
299                                 deleteButton.clicked.connect(callback)
300
301                                 rowLayout = QtGui.QHBoxLayout()
302                                 rowLayout.addWidget(titleLabel)
303                                 rowLayout.addWidget(numberSelector)
304                                 rowLayout.addWidget(deleteButton)
305                                 rowWidget = QtGui.QWidget()
306                                 rowWidget.setLayout(rowLayout)
307                                 self._targetLayout.addWidget(rowWidget)
308                         self._history.setHtml("")
309                         self._history.setVisible(False)
310                         self._singleNumberSelector.setVisible(False)
311
312                         self._scroll_to_bottom()
313                         self._window.setWindowTitle("Contacts")
314                         self._window.show()
315                         self._smsEntry.setFocus(QtCore.Qt.OtherFocusReason)
316
317         def _populate_number_selector(self, selector, cid, numbers):
318                 while 0 < selector.count():
319                         selector.removeItem(0)
320
321                 if len(numbers) == 1:
322                         numbers, defaultIndex = _get_contact_numbers(self._session, cid, numbers[0])
323                 else:
324                         defaultIndex = 0
325
326                 for number, description in numbers:
327                         if description:
328                                 label = "%s - %s" % (number, description)
329                         else:
330                                 label = number
331                         selector.addItem(label)
332                 selector.setVisible(True)
333                 if 1 < len(numbers):
334                         selector.setEnabled(True)
335                         selector.setCurrentIndex(defaultIndex)
336                 else:
337                         selector.setEnabled(False)
338                 callback = functools.partial(
339                         self._on_change_number,
340                         cid
341                 )
342                 callback.__name__ = "thanks partials for not having names and pyqt for requiring them"
343                 selector.currentIndexChanged.connect(
344                         QtCore.pyqtSlot(int)(callback)
345                 )
346
347         def _scroll_to_bottom(self):
348                 self._scrollEntry.ensureWidgetVisible(self._smsEntry)
349
350         @misc_utils.log_exception(_moduleLogger)
351         def _on_sms_clicked(self, arg):
352                 message = str(self._smsEntry.toPlainText())
353                 self._session.draft.send(message)
354                 self._smsEntry.setPlainText("")
355
356         @misc_utils.log_exception(_moduleLogger)
357         def _on_call_clicked(self, arg):
358                 self._session.draft.call()
359                 self._smsEntry.setPlainText("")
360
361         @misc_utils.log_exception(_moduleLogger)
362         def _on_remove_contact(self, cid, toggled):
363                 self._session.draft.remove_contact(cid)
364
365         @misc_utils.log_exception(_moduleLogger)
366         def _on_change_number(self, cid, index):
367                 # Exception thrown when the first item is removed
368                 numbers = self._session.draft.get_numbers(cid)
369                 number = numbers[index][0]
370                 self._session.draft.set_selected_number(cid, number)
371
372         @QtCore.pyqtSlot()
373         @misc_utils.log_exception(_moduleLogger)
374         def _on_recipients_changed(self):
375                 self._update_recipients()
376
377         @QtCore.pyqtSlot()
378         @misc_utils.log_exception(_moduleLogger)
379         def _on_op_finished(self):
380                 self._window.hide()
381
382         @QtCore.pyqtSlot()
383         @misc_utils.log_exception(_moduleLogger)
384         def _on_letter_count_changed(self):
385                 self._update_letter_count()
386                 self._update_button_state()
387
388         @QtCore.pyqtSlot()
389         @QtCore.pyqtSlot(bool)
390         @misc_utils.log_exception(_moduleLogger)
391         def _on_close_window(self, checked = True):
392                 self._window.hide()
393
394
395 def _get_contact_numbers(session, contactId, numberDescription):
396         contactPhoneNumbers = []
397         if contactId and contactId != "0":
398                 try:
399                         contactDetails = copy.deepcopy(session.get_contacts()[contactId])
400                         contactPhoneNumbers = contactDetails["numbers"]
401                 except KeyError:
402                         contactPhoneNumbers = []
403                 contactPhoneNumbers = [
404                         (contactPhoneNumber["phoneNumber"], contactPhoneNumber["phoneType"])
405                         for contactPhoneNumber in contactPhoneNumbers
406                 ]
407                 if contactPhoneNumbers:
408                         uglyContactNumbers = (
409                                 misc_utils.make_ugly(contactNumber)
410                                 for (contactNumber, _) in contactPhoneNumbers
411                         )
412                         defaultMatches = [
413                                 misc_utils.similar_ugly_numbers(numberDescription[0], contactNumber)
414                                 for contactNumber in uglyContactNumbers
415                         ]
416                         try:
417                                 defaultIndex = defaultMatches.index(True)
418                         except ValueError:
419                                 contactPhoneNumbers.append(numberDescription)
420                                 defaultIndex = len(contactPhoneNumbers)-1
421                                 _moduleLogger.warn(
422                                         "Could not find contact %r's number %s among %r" % (
423                                                 contactId, numberDescription, contactPhoneNumbers
424                                         )
425                                 )
426
427         if not contactPhoneNumbers:
428                 contactPhoneNumbers = [numberDescription]
429                 defaultIndex = -1
430
431         return contactPhoneNumbers, defaultIndex