07a9ae942b91a4e897a8427ea68d8cfc096d8876
[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 enter custom callback numbers
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._callbackSelector = QtGui.QComboBox()
93                 #self._callbackSelector.setEditable(True)
94                 self._callbackSelector.setInsertPolicy(QtGui.QComboBox.InsertAtTop)
95
96                 self._credLayout = QtGui.QGridLayout()
97                 self._credLayout.addWidget(QtGui.QLabel("Account"), 0, 0)
98                 self._credLayout.addWidget(self._accountNumberLabel, 0, 1)
99                 self._credLayout.addWidget(QtGui.QLabel("Callback"), 1, 0)
100                 self._credLayout.addWidget(self._callbackSelector, 1, 1)
101                 self._credLayout.addWidget(QtGui.QLabel(""), 2, 0)
102                 self._credLayout.addWidget(self._clearButton, 2, 1)
103                 self._credLayout.addWidget(QtGui.QLabel(""), 3, 0)
104
105                 self._loginButton = QtGui.QPushButton("&Apply")
106                 self._buttonLayout = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Cancel)
107                 self._buttonLayout.addButton(self._loginButton, QtGui.QDialogButtonBox.AcceptRole)
108
109                 self._layout = QtGui.QVBoxLayout()
110                 self._layout.addLayout(self._credLayout)
111                 self._layout.addWidget(self._buttonLayout)
112
113                 self._dialog = QtGui.QDialog()
114                 self._dialog.setWindowTitle("Login")
115                 self._dialog.setLayout(self._layout)
116                 qui_utils.set_autorient(self._dialog, True)
117                 self._buttonLayout.accepted.connect(self._dialog.accept)
118                 self._buttonLayout.rejected.connect(self._dialog.reject)
119
120                 self._closeWindowAction = QtGui.QAction(None)
121                 self._closeWindowAction.setText("Close")
122                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
123                 self._closeWindowAction.triggered.connect(self._on_close_window)
124
125                 self._dialog.addAction(self._closeWindowAction)
126                 self._dialog.addAction(app.quitAction)
127                 self._dialog.addAction(app.fullscreenAction)
128
129         @property
130         def doClear(self):
131                 return self._doClear
132
133         accountNumber = property(
134                 lambda self: str(self._accountNumberLabel.text()),
135                 lambda self, num: self._accountNumberLabel.setText(num),
136         )
137
138         @property
139         def selectedCallback(self):
140                 index = self._callbackSelector.currentIndex()
141                 data = str(self._callbackSelector.itemData(index).toPyObject())
142                 return data
143
144         def set_callbacks(self, choices, default):
145                 self._callbackSelector.clear()
146
147                 uglyDefault = misc_utils.make_ugly(default)
148                 for i, (number, description) in enumerate(choices.iteritems()):
149                         prettyNumber = misc_utils.make_pretty(number)
150                         uglyNumber = misc_utils.make_ugly(number)
151                         if not uglyNumber:
152                                 continue
153
154                         self._callbackSelector.addItem("%s - %s" % (prettyNumber, description), uglyNumber)
155                         if uglyNumber == uglyDefault:
156                                 self._callbackSelector.setCurrentIndex(i)
157
158
159         def run(self, parent=None):
160                 self._doClear = False
161                 self._dialog.setParent(parent)
162
163                 response = self._dialog.exec_()
164                 return response
165
166         @QtCore.pyqtSlot()
167         @QtCore.pyqtSlot(bool)
168         def _on_clear(self, checked = False):
169                 self._doClear = True
170                 self._dialog.accept()
171
172         @QtCore.pyqtSlot()
173         @QtCore.pyqtSlot(bool)
174         @misc_utils.log_exception(_moduleLogger)
175         def _on_close_window(self, checked = True):
176                 self._dialog.reject()
177
178
179 class SMSEntryWindow(object):
180
181         MAX_CHAR = 160
182
183         def __init__(self, parent, app, session, errorLog):
184                 self._session = session
185                 self._session.draft.recipientsChanged.connect(self._on_recipients_changed)
186                 self._session.draft.sendingMessage.connect(self._on_op_started)
187                 self._session.draft.calling.connect(self._on_op_started)
188                 self._session.draft.cancelling.connect(self._on_op_started)
189                 self._session.draft.calling.connect(self._on_calling_started)
190                 self._session.draft.called.connect(self._on_op_finished)
191                 self._session.draft.sentMessage.connect(self._on_op_finished)
192                 self._session.draft.cancelled.connect(self._on_op_finished)
193                 self._session.draft.error.connect(self._on_op_error)
194                 self._errorLog = errorLog
195
196                 self._errorDisplay = qui_utils.ErrorDisplay(self._errorLog)
197
198                 self._targetLayout = QtGui.QVBoxLayout()
199                 self._targetList = QtGui.QWidget()
200                 self._targetList.setLayout(self._targetLayout)
201                 self._history = QtGui.QLabel()
202                 self._history.setTextFormat(QtCore.Qt.RichText)
203                 self._history.setWordWrap(True)
204                 self._smsEntry = QtGui.QTextEdit()
205                 self._smsEntry.textChanged.connect(self._on_letter_count_changed)
206
207                 self._entryLayout = QtGui.QVBoxLayout()
208                 self._entryLayout.addWidget(self._targetList)
209                 self._entryLayout.addWidget(self._history)
210                 self._entryLayout.addWidget(self._smsEntry)
211                 self._entryLayout.setContentsMargins(0, 0, 0, 0)
212                 self._entryWidget = QtGui.QWidget()
213                 self._entryWidget.setLayout(self._entryLayout)
214                 self._entryWidget.setContentsMargins(0, 0, 0, 0)
215                 self._scrollEntry = QtGui.QScrollArea()
216                 self._scrollEntry.setWidget(self._entryWidget)
217                 self._scrollEntry.setWidgetResizable(True)
218                 self._scrollEntry.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignBottom)
219                 self._scrollEntry.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
220                 self._scrollEntry.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
221
222                 self._characterCountLabel = QtGui.QLabel("0 (0)")
223                 self._singleNumberSelector = QtGui.QComboBox()
224                 self._smsButton = QtGui.QPushButton("SMS")
225                 self._smsButton.clicked.connect(self._on_sms_clicked)
226                 self._smsButton.setEnabled(False)
227                 self._dialButton = QtGui.QPushButton("Dial")
228                 self._dialButton.clicked.connect(self._on_call_clicked)
229                 self._cancelButton = QtGui.QPushButton("Cancel Call")
230                 self._cancelButton.clicked.connect(self._on_cancel_clicked)
231                 self._cancelButton.setVisible(False)
232
233                 self._buttonLayout = QtGui.QHBoxLayout()
234                 self._buttonLayout.addWidget(self._characterCountLabel)
235                 self._buttonLayout.addWidget(self._singleNumberSelector)
236                 self._buttonLayout.addWidget(self._smsButton)
237                 self._buttonLayout.addWidget(self._dialButton)
238                 self._buttonLayout.addWidget(self._cancelButton)
239
240                 self._layout = QtGui.QVBoxLayout()
241                 self._layout.addWidget(self._errorDisplay.toplevel)
242                 self._layout.addWidget(self._scrollEntry)
243                 self._layout.addLayout(self._buttonLayout)
244
245                 centralWidget = QtGui.QWidget()
246                 centralWidget.setLayout(self._layout)
247
248                 self._window = QtGui.QMainWindow(parent)
249                 qui_utils.set_autorient(self._window, True)
250                 qui_utils.set_stackable(self._window, True)
251                 self._window.setWindowTitle("Contact")
252                 self._window.setCentralWidget(centralWidget)
253
254                 self._closeWindowAction = QtGui.QAction(None)
255                 self._closeWindowAction.setText("Close")
256                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
257                 self._closeWindowAction.triggered.connect(self._on_close_window)
258
259                 fileMenu = self._window.menuBar().addMenu("&File")
260                 fileMenu.addAction(self._closeWindowAction)
261                 fileMenu.addAction(app.quitAction)
262                 viewMenu = self._window.menuBar().addMenu("&View")
263                 viewMenu.addAction(app.fullscreenAction)
264
265                 self._window.show()
266                 self._update_recipients()
267
268         def _update_letter_count(self):
269                 count = self._smsEntry.toPlainText().size()
270                 numTexts, numCharInText = divmod(count, self.MAX_CHAR)
271                 numTexts += 1
272                 numCharsLeftInText = self.MAX_CHAR - numCharInText
273                 self._characterCountLabel.setText("%d (%d)" % (numCharsLeftInText, numTexts))
274
275         def _update_button_state(self):
276                 if self._session.draft.get_num_contacts() == 0:
277                         self._dialButton.setEnabled(False)
278                         self._smsButton.setEnabled(False)
279                 elif self._session.draft.get_num_contacts() == 1:
280                         count = self._smsEntry.toPlainText().size()
281                         if count == 0:
282                                 self._dialButton.setEnabled(True)
283                                 self._smsButton.setEnabled(False)
284                         else:
285                                 self._dialButton.setEnabled(False)
286                                 self._smsButton.setEnabled(True)
287                 else:
288                         self._dialButton.setEnabled(False)
289                         self._smsButton.setEnabled(True)
290
291         def _update_recipients(self):
292                 draftContactsCount = self._session.draft.get_num_contacts()
293                 if draftContactsCount == 0:
294                         self._window.hide()
295                 elif draftContactsCount == 1:
296                         (cid, ) = self._session.draft.get_contacts()
297                         title = self._session.draft.get_title(cid)
298                         description = self._session.draft.get_description(cid)
299                         numbers = self._session.draft.get_numbers(cid)
300
301                         self._targetList.setVisible(False)
302                         if description:
303                                 self._history.setText(description)
304                                 self._history.setVisible(True)
305                         else:
306                                 self._history.setText("")
307                                 self._history.setVisible(False)
308                         self._populate_number_selector(self._singleNumberSelector, cid, numbers)
309
310                         self._scroll_to_bottom()
311                         self._window.setWindowTitle(title)
312                         self._window.show()
313                         self._smsEntry.setFocus(QtCore.Qt.OtherFocusReason)
314                 else:
315                         self._targetList.setVisible(True)
316                         while self._targetLayout.count():
317                                 removedLayoutItem = self._targetLayout.takeAt(self._targetLayout.count()-1)
318                                 removedWidget = removedLayoutItem.widget()
319                                 removedWidget.close()
320                         for cid in self._session.draft.get_contacts():
321                                 title = self._session.draft.get_title(cid)
322                                 description = self._session.draft.get_description(cid)
323                                 numbers = self._session.draft.get_numbers(cid)
324
325                                 titleLabel = QtGui.QLabel(title)
326                                 numberSelector = QtGui.QComboBox()
327                                 self._populate_number_selector(numberSelector, cid, numbers)
328                                 deleteButton = QtGui.QPushButton("Delete")
329                                 callback = functools.partial(
330                                         self._on_remove_contact,
331                                         cid
332                                 )
333                                 callback.__name__ = "b"
334                                 deleteButton.clicked.connect(callback)
335
336                                 rowLayout = QtGui.QHBoxLayout()
337                                 rowLayout.addWidget(titleLabel)
338                                 rowLayout.addWidget(numberSelector)
339                                 rowLayout.addWidget(deleteButton)
340                                 rowWidget = QtGui.QWidget()
341                                 rowWidget.setLayout(rowLayout)
342                                 self._targetLayout.addWidget(rowWidget)
343                         self._history.setText("")
344                         self._history.setVisible(False)
345                         self._singleNumberSelector.setVisible(False)
346
347                         self._scroll_to_bottom()
348                         self._window.setWindowTitle("Contacts")
349                         self._window.show()
350                         self._smsEntry.setFocus(QtCore.Qt.OtherFocusReason)
351
352         def _populate_number_selector(self, selector, cid, numbers):
353                 selector.clear()
354
355                 if len(numbers) == 1:
356                         numbers, defaultIndex = _get_contact_numbers(self._session, cid, numbers[0])
357                 else:
358                         defaultIndex = 0
359
360                 for number, description in numbers:
361                         if description:
362                                 label = "%s - %s" % (number, description)
363                         else:
364                                 label = number
365                         selector.addItem(label)
366                 selector.setVisible(True)
367                 if 1 < len(numbers):
368                         selector.setEnabled(True)
369                         selector.setCurrentIndex(defaultIndex)
370                 else:
371                         selector.setEnabled(False)
372                 callback = functools.partial(
373                         self._on_change_number,
374                         cid
375                 )
376                 callback.__name__ = "thanks partials for not having names and pyqt for requiring them"
377                 selector.currentIndexChanged.connect(
378                         QtCore.pyqtSlot(int)(callback)
379                 )
380
381         def _scroll_to_bottom(self):
382                 self._scrollEntry.ensureWidgetVisible(self._smsEntry)
383
384         @misc_utils.log_exception(_moduleLogger)
385         def _on_sms_clicked(self, arg):
386                 message = str(self._smsEntry.toPlainText())
387                 self._session.draft.send(message)
388                 self._smsEntry.setPlainText("")
389
390         @misc_utils.log_exception(_moduleLogger)
391         def _on_call_clicked(self, arg):
392                 self._session.draft.call()
393                 self._smsEntry.setPlainText("")
394
395         @QtCore.pyqtSlot()
396         @misc_utils.log_exception(_moduleLogger)
397         def _on_cancel_clicked(self, message):
398                 self._session.draft.cancel()
399
400         @misc_utils.log_exception(_moduleLogger)
401         def _on_remove_contact(self, cid, toggled):
402                 self._session.draft.remove_contact(cid)
403
404         @misc_utils.log_exception(_moduleLogger)
405         def _on_change_number(self, cid, index):
406                 # Exception thrown when the first item is removed
407                 numbers = self._session.draft.get_numbers(cid)
408                 number = numbers[index][0]
409                 self._session.draft.set_selected_number(cid, number)
410
411         @QtCore.pyqtSlot()
412         @misc_utils.log_exception(_moduleLogger)
413         def _on_recipients_changed(self):
414                 self._update_recipients()
415
416         @QtCore.pyqtSlot()
417         @misc_utils.log_exception(_moduleLogger)
418         def _on_op_started(self):
419                 self._smsEntry.setReadOnly(True)
420                 self._smsButton.setVisible(False)
421                 self._dialButton.setVisible(False)
422                 self._window.show()
423
424         @QtCore.pyqtSlot()
425         @misc_utils.log_exception(_moduleLogger)
426         def _on_calling_started(self):
427                 self._cancelButton.setVisible(True)
428
429         @QtCore.pyqtSlot()
430         @misc_utils.log_exception(_moduleLogger)
431         def _on_op_finished(self):
432                 self._window.hide()
433
434                 self._smsEntry.setReadOnly(False)
435                 self._cancelButton.setVisible(False)
436                 self._smsButton.setVisible(True)
437                 self._dialButton.setVisible(True)
438
439         @QtCore.pyqtSlot()
440         @misc_utils.log_exception(_moduleLogger)
441         def _on_op_error(self, message):
442                 self._smsEntry.setReadOnly(False)
443                 self._cancelButton.setVisible(False)
444                 self._smsButton.setVisible(True)
445                 self._dialButton.setVisible(True)
446
447                 self._errorLog.push_message(message)
448
449         @QtCore.pyqtSlot()
450         @misc_utils.log_exception(_moduleLogger)
451         def _on_letter_count_changed(self):
452                 self._update_letter_count()
453                 self._update_button_state()
454
455         @QtCore.pyqtSlot()
456         @QtCore.pyqtSlot(bool)
457         @misc_utils.log_exception(_moduleLogger)
458         def _on_close_window(self, checked = True):
459                 self._window.hide()
460
461
462 def _get_contact_numbers(session, contactId, numberDescription):
463         contactPhoneNumbers = []
464         if contactId and contactId != "0":
465                 try:
466                         contactDetails = copy.deepcopy(session.get_contacts()[contactId])
467                         contactPhoneNumbers = contactDetails["numbers"]
468                 except KeyError:
469                         contactPhoneNumbers = []
470                 contactPhoneNumbers = [
471                         (contactPhoneNumber["phoneNumber"], contactPhoneNumber["phoneType"])
472                         for contactPhoneNumber in contactPhoneNumbers
473                 ]
474                 if contactPhoneNumbers:
475                         uglyContactNumbers = (
476                                 misc_utils.make_ugly(contactNumber)
477                                 for (contactNumber, _) in contactPhoneNumbers
478                         )
479                         defaultMatches = [
480                                 misc_utils.similar_ugly_numbers(numberDescription[0], contactNumber)
481                                 for contactNumber in uglyContactNumbers
482                         ]
483                         try:
484                                 defaultIndex = defaultMatches.index(True)
485                         except ValueError:
486                                 contactPhoneNumbers.append(numberDescription)
487                                 defaultIndex = len(contactPhoneNumbers)-1
488                                 _moduleLogger.warn(
489                                         "Could not find contact %r's number %s among %r" % (
490                                                 contactId, numberDescription, contactPhoneNumbers
491                                         )
492                                 )
493
494         if not contactPhoneNumbers:
495                 contactPhoneNumbers = [numberDescription]
496                 defaultIndex = -1
497
498         return contactPhoneNumbers, defaultIndex