Trying to preserve unicode data
[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 import constants
14 from util import qui_utils
15 from util import misc as misc_utils
16
17
18 _moduleLogger = logging.getLogger(__name__)
19
20
21 class CredentialsDialog(object):
22
23         def __init__(self, app):
24                 self._usernameField = QtGui.QLineEdit()
25                 self._passwordField = QtGui.QLineEdit()
26                 self._passwordField.setEchoMode(QtGui.QLineEdit.PasswordEchoOnEdit)
27
28                 self._credLayout = QtGui.QGridLayout()
29                 self._credLayout.addWidget(QtGui.QLabel("Username"), 0, 0)
30                 self._credLayout.addWidget(self._usernameField, 0, 1)
31                 self._credLayout.addWidget(QtGui.QLabel("Password"), 1, 0)
32                 self._credLayout.addWidget(self._passwordField, 1, 1)
33
34                 self._loginButton = QtGui.QPushButton("&Login")
35                 self._buttonLayout = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Cancel)
36                 self._buttonLayout.addButton(self._loginButton, QtGui.QDialogButtonBox.AcceptRole)
37
38                 self._layout = QtGui.QVBoxLayout()
39                 self._layout.addLayout(self._credLayout)
40                 self._layout.addWidget(self._buttonLayout)
41
42                 self._dialog = QtGui.QDialog()
43                 self._dialog.setWindowTitle("Login")
44                 self._dialog.setLayout(self._layout)
45                 self._dialog.setAttribute(QtCore.Qt.WA_DeleteOnClose, False)
46                 qui_utils.set_autorient(self._dialog, True)
47                 self._buttonLayout.accepted.connect(self._dialog.accept)
48                 self._buttonLayout.rejected.connect(self._dialog.reject)
49
50                 self._closeWindowAction = QtGui.QAction(None)
51                 self._closeWindowAction.setText("Close")
52                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
53                 self._closeWindowAction.triggered.connect(self._on_close_window)
54
55                 self._dialog.addAction(self._closeWindowAction)
56                 self._dialog.addAction(app.quitAction)
57                 self._dialog.addAction(app.fullscreenAction)
58
59         def run(self, defaultUsername, defaultPassword, parent=None):
60                 self._dialog.setParent(parent, QtCore.Qt.Dialog)
61                 try:
62                         self._usernameField.setText(defaultUsername)
63                         self._passwordField.setText(defaultPassword)
64
65                         response = self._dialog.exec_()
66                         if response == QtGui.QDialog.Accepted:
67                                 return str(self._usernameField.text()), str(self._passwordField.text())
68                         elif response == QtGui.QDialog.Rejected:
69                                 raise RuntimeError("Login Cancelled")
70                         else:
71                                 raise RuntimeError("Unknown Response")
72                 finally:
73                         self._dialog.setParent(None, QtCore.Qt.Dialog)
74
75         def close(self):
76                 self._dialog.reject()
77
78         @QtCore.pyqtSlot()
79         @QtCore.pyqtSlot(bool)
80         @misc_utils.log_exception(_moduleLogger)
81         def _on_close_window(self, checked = True):
82                 self._dialog.reject()
83
84
85 class AboutDialog(object):
86
87         def __init__(self, app):
88                 self._title = QtGui.QLabel(
89                         "<h1>%s</h1><h3>Version: %s</h3>" % (
90                                 constants.__pretty_app_name__, constants.__version__
91                         )
92                 )
93                 self._title.setTextFormat(QtCore.Qt.RichText)
94                 self._title.setAlignment(QtCore.Qt.AlignCenter)
95                 self._copyright = QtGui.QLabel("<h6>Developed by Ed Page<h6><h6>Icons: See website</h6>")
96                 self._copyright.setTextFormat(QtCore.Qt.RichText)
97                 self._copyright.setAlignment(QtCore.Qt.AlignCenter)
98                 self._link = QtGui.QLabel('<a href="http://gc-dialer.garage.maemo.org">DialCentral Website</a>')
99                 self._link.setTextFormat(QtCore.Qt.RichText)
100                 self._link.setAlignment(QtCore.Qt.AlignCenter)
101                 self._link.setOpenExternalLinks(True)
102
103                 self._layout = QtGui.QVBoxLayout()
104                 self._layout.addWidget(self._title)
105                 self._layout.addWidget(self._copyright)
106                 self._layout.addWidget(self._link)
107
108                 self._dialog = QtGui.QDialog()
109                 self._dialog.setWindowTitle("About")
110                 self._dialog.setLayout(self._layout)
111                 qui_utils.set_autorient(self._dialog, True)
112
113                 self._closeWindowAction = QtGui.QAction(None)
114                 self._closeWindowAction.setText("Close")
115                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
116                 self._closeWindowAction.triggered.connect(self._on_close_window)
117
118                 self._dialog.addAction(self._closeWindowAction)
119                 self._dialog.addAction(app.quitAction)
120                 self._dialog.addAction(app.fullscreenAction)
121
122         def run(self, parent=None):
123                 self._dialog.setParent(parent)
124
125                 response = self._dialog.exec_()
126                 return response
127
128         def close(self):
129                 self._dialog.reject()
130
131         @QtCore.pyqtSlot()
132         @QtCore.pyqtSlot(bool)
133         @misc_utils.log_exception(_moduleLogger)
134         def _on_close_window(self, checked = True):
135                 self._dialog.reject()
136
137
138 class AccountDialog(object):
139
140         # @bug Can't enter custom callback numbers
141
142         def __init__(self, app):
143                 self._doClear = False
144
145                 self._accountNumberLabel = QtGui.QLabel("NUMBER NOT SET")
146                 self._clearButton = QtGui.QPushButton("Clear Account")
147                 self._clearButton.clicked.connect(self._on_clear)
148
149                 self._callbackSelector = QtGui.QComboBox()
150                 #self._callbackSelector.setEditable(True)
151                 self._callbackSelector.setInsertPolicy(QtGui.QComboBox.InsertAtTop)
152
153                 self._credLayout = QtGui.QGridLayout()
154                 self._credLayout.addWidget(QtGui.QLabel("Account"), 0, 0)
155                 self._credLayout.addWidget(self._accountNumberLabel, 0, 1)
156                 self._credLayout.addWidget(QtGui.QLabel("Callback"), 1, 0)
157                 self._credLayout.addWidget(self._callbackSelector, 1, 1)
158                 self._credLayout.addWidget(QtGui.QLabel(""), 2, 0)
159                 self._credLayout.addWidget(self._clearButton, 2, 1)
160                 self._credLayout.addWidget(QtGui.QLabel(""), 3, 0)
161
162                 self._loginButton = QtGui.QPushButton("&Apply")
163                 self._buttonLayout = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Cancel)
164                 self._buttonLayout.addButton(self._loginButton, QtGui.QDialogButtonBox.AcceptRole)
165
166                 self._layout = QtGui.QVBoxLayout()
167                 self._layout.addLayout(self._credLayout)
168                 self._layout.addWidget(self._buttonLayout)
169
170                 self._dialog = QtGui.QDialog()
171                 self._dialog.setWindowTitle("Account")
172                 self._dialog.setLayout(self._layout)
173                 qui_utils.set_autorient(self._dialog, True)
174                 self._buttonLayout.accepted.connect(self._dialog.accept)
175                 self._buttonLayout.rejected.connect(self._dialog.reject)
176
177                 self._closeWindowAction = QtGui.QAction(None)
178                 self._closeWindowAction.setText("Close")
179                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
180                 self._closeWindowAction.triggered.connect(self._on_close_window)
181
182                 self._dialog.addAction(self._closeWindowAction)
183                 self._dialog.addAction(app.quitAction)
184                 self._dialog.addAction(app.fullscreenAction)
185
186         @property
187         def doClear(self):
188                 return self._doClear
189
190         accountNumber = property(
191                 lambda self: str(self._accountNumberLabel.text()),
192                 lambda self, num: self._accountNumberLabel.setText(num),
193         )
194
195         @property
196         def selectedCallback(self):
197                 index = self._callbackSelector.currentIndex()
198                 data = str(self._callbackSelector.itemData(index).toPyObject())
199                 return data
200
201         def set_callbacks(self, choices, default):
202                 self._callbackSelector.clear()
203
204                 self._callbackSelector.addItem("Not Set", "")
205
206                 uglyDefault = misc_utils.make_ugly(default)
207                 for i, (number, description) in enumerate(choices.iteritems()):
208                         prettyNumber = misc_utils.make_pretty(number)
209                         uglyNumber = misc_utils.make_ugly(number)
210                         if not uglyNumber:
211                                 continue
212
213                         self._callbackSelector.addItem("%s - %s" % (prettyNumber, description), uglyNumber)
214                         if uglyNumber == uglyDefault:
215                                 self._callbackSelector.setCurrentIndex(i)
216
217
218         def run(self, parent=None):
219                 self._doClear = False
220                 self._dialog.setParent(parent)
221
222                 response = self._dialog.exec_()
223                 return response
224
225         def close(self):
226                 self._dialog.reject()
227
228         @QtCore.pyqtSlot()
229         @QtCore.pyqtSlot(bool)
230         def _on_clear(self, checked = False):
231                 self._doClear = True
232                 self._dialog.accept()
233
234         @QtCore.pyqtSlot()
235         @QtCore.pyqtSlot(bool)
236         @misc_utils.log_exception(_moduleLogger)
237         def _on_close_window(self, checked = True):
238                 self._dialog.reject()
239
240
241 class SMSEntryWindow(object):
242
243         MAX_CHAR = 160
244
245         def __init__(self, parent, app, session, errorLog):
246                 self._session = session
247                 self._session.draft.recipientsChanged.connect(self._on_recipients_changed)
248                 self._session.draft.sendingMessage.connect(self._on_op_started)
249                 self._session.draft.calling.connect(self._on_op_started)
250                 self._session.draft.cancelling.connect(self._on_op_started)
251                 self._session.draft.calling.connect(self._on_calling_started)
252                 self._session.draft.called.connect(self._on_op_finished)
253                 self._session.draft.sentMessage.connect(self._on_op_finished)
254                 self._session.draft.cancelled.connect(self._on_op_finished)
255                 self._session.draft.error.connect(self._on_op_error)
256                 self._errorLog = errorLog
257
258                 self._errorDisplay = qui_utils.ErrorDisplay(self._errorLog)
259
260                 self._targetLayout = QtGui.QVBoxLayout()
261                 self._targetList = QtGui.QWidget()
262                 self._targetList.setLayout(self._targetLayout)
263                 self._history = QtGui.QLabel()
264                 self._history.setTextFormat(QtCore.Qt.RichText)
265                 self._history.setWordWrap(True)
266                 self._smsEntry = QtGui.QTextEdit()
267                 self._smsEntry.textChanged.connect(self._on_letter_count_changed)
268
269                 self._entryLayout = QtGui.QVBoxLayout()
270                 self._entryLayout.addWidget(self._targetList)
271                 self._entryLayout.addWidget(self._history)
272                 self._entryLayout.addWidget(self._smsEntry)
273                 self._entryLayout.setContentsMargins(0, 0, 0, 0)
274                 self._entryWidget = QtGui.QWidget()
275                 self._entryWidget.setLayout(self._entryLayout)
276                 self._entryWidget.setContentsMargins(0, 0, 0, 0)
277                 self._scrollEntry = QtGui.QScrollArea()
278                 self._scrollEntry.setWidget(self._entryWidget)
279                 self._scrollEntry.setWidgetResizable(True)
280                 self._scrollEntry.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignBottom)
281                 self._scrollEntry.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
282                 self._scrollEntry.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
283
284                 self._characterCountLabel = QtGui.QLabel("0 (0)")
285                 self._singleNumberSelector = QtGui.QComboBox()
286                 self._smsButton = QtGui.QPushButton("SMS")
287                 self._smsButton.clicked.connect(self._on_sms_clicked)
288                 self._smsButton.setEnabled(False)
289                 self._dialButton = QtGui.QPushButton("Dial")
290                 self._dialButton.clicked.connect(self._on_call_clicked)
291                 self._cancelButton = QtGui.QPushButton("Cancel Call")
292                 self._cancelButton.clicked.connect(self._on_cancel_clicked)
293                 self._cancelButton.setVisible(False)
294
295                 self._buttonLayout = QtGui.QHBoxLayout()
296                 self._buttonLayout.addWidget(self._characterCountLabel)
297                 self._buttonLayout.addWidget(self._singleNumberSelector)
298                 self._buttonLayout.addWidget(self._smsButton)
299                 self._buttonLayout.addWidget(self._dialButton)
300                 self._buttonLayout.addWidget(self._cancelButton)
301
302                 self._layout = QtGui.QVBoxLayout()
303                 self._layout.addWidget(self._errorDisplay.toplevel)
304                 self._layout.addWidget(self._scrollEntry)
305                 self._layout.addLayout(self._buttonLayout)
306
307                 centralWidget = QtGui.QWidget()
308                 centralWidget.setLayout(self._layout)
309
310                 self._window = QtGui.QMainWindow(parent)
311                 qui_utils.set_autorient(self._window, True)
312                 qui_utils.set_stackable(self._window, True)
313                 self._window.setWindowTitle("Contact")
314                 self._window.setCentralWidget(centralWidget)
315
316                 self._closeWindowAction = QtGui.QAction(None)
317                 self._closeWindowAction.setText("Close")
318                 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
319                 self._closeWindowAction.triggered.connect(self._on_close_window)
320
321                 fileMenu = self._window.menuBar().addMenu("&File")
322                 fileMenu.addAction(self._closeWindowAction)
323                 fileMenu.addAction(app.quitAction)
324                 viewMenu = self._window.menuBar().addMenu("&View")
325                 viewMenu.addAction(app.fullscreenAction)
326
327                 self._scrollTimer = QtCore.QTimer()
328                 self._scrollTimer.setInterval(0)
329                 self._scrollTimer.setSingleShot(True)
330                 self._scrollTimer.timeout.connect(self._on_delayed_scroll_to_bottom)
331
332                 self._window.show()
333                 self._update_recipients()
334
335         def close(self):
336                 self._window.destroy()
337                 self._window = None
338
339         def _update_letter_count(self):
340                 count = self._smsEntry.toPlainText().size()
341                 numTexts, numCharInText = divmod(count, self.MAX_CHAR)
342                 numTexts += 1
343                 numCharsLeftInText = self.MAX_CHAR - numCharInText
344                 self._characterCountLabel.setText("%d (%d)" % (numCharsLeftInText, numTexts))
345
346         def _update_button_state(self):
347                 if self._session.draft.get_num_contacts() == 0:
348                         self._dialButton.setEnabled(False)
349                         self._smsButton.setEnabled(False)
350                 elif self._session.draft.get_num_contacts() == 1:
351                         count = self._smsEntry.toPlainText().size()
352                         if count == 0:
353                                 self._dialButton.setEnabled(True)
354                                 self._smsButton.setEnabled(False)
355                         else:
356                                 self._dialButton.setEnabled(False)
357                                 self._smsButton.setEnabled(True)
358                 else:
359                         self._dialButton.setEnabled(False)
360                         self._smsButton.setEnabled(True)
361
362         def _update_recipients(self):
363                 draftContactsCount = self._session.draft.get_num_contacts()
364                 if draftContactsCount == 0:
365                         self._window.hide()
366                 elif draftContactsCount == 1:
367                         (cid, ) = self._session.draft.get_contacts()
368                         title = self._session.draft.get_title(cid)
369                         description = self._session.draft.get_description(cid)
370                         numbers = self._session.draft.get_numbers(cid)
371
372                         self._targetList.setVisible(False)
373                         if description:
374                                 self._history.setText(description)
375                                 self._history.setVisible(True)
376                         else:
377                                 self._history.setText("")
378                                 self._history.setVisible(False)
379                         self._populate_number_selector(self._singleNumberSelector, cid, numbers)
380
381                         self._scroll_to_bottom()
382                         self._window.setWindowTitle(title)
383                         self._window.show()
384                         self._smsEntry.setFocus(QtCore.Qt.OtherFocusReason)
385                 else:
386                         self._targetList.setVisible(True)
387                         while self._targetLayout.count():
388                                 removedLayoutItem = self._targetLayout.takeAt(self._targetLayout.count()-1)
389                                 removedWidget = removedLayoutItem.widget()
390                                 removedWidget.close()
391                         for cid in self._session.draft.get_contacts():
392                                 title = self._session.draft.get_title(cid)
393                                 description = self._session.draft.get_description(cid)
394                                 numbers = self._session.draft.get_numbers(cid)
395
396                                 titleLabel = QtGui.QLabel(title)
397                                 numberSelector = QtGui.QComboBox()
398                                 self._populate_number_selector(numberSelector, cid, numbers)
399                                 deleteButton = QtGui.QPushButton("Delete")
400                                 callback = functools.partial(
401                                         self._on_remove_contact,
402                                         cid
403                                 )
404                                 callback.__name__ = "b"
405                                 deleteButton.clicked.connect(callback)
406
407                                 rowLayout = QtGui.QHBoxLayout()
408                                 rowLayout.addWidget(titleLabel)
409                                 rowLayout.addWidget(numberSelector)
410                                 rowLayout.addWidget(deleteButton)
411                                 rowWidget = QtGui.QWidget()
412                                 rowWidget.setLayout(rowLayout)
413                                 self._targetLayout.addWidget(rowWidget)
414                         self._history.setText("")
415                         self._history.setVisible(False)
416                         self._singleNumberSelector.setVisible(False)
417
418                         self._scroll_to_bottom()
419                         self._window.setWindowTitle("Contacts")
420                         self._window.show()
421                         self._smsEntry.setFocus(QtCore.Qt.OtherFocusReason)
422
423         def _populate_number_selector(self, selector, cid, numbers):
424                 selector.clear()
425
426                 if len(numbers) == 1:
427                         numbers, defaultIndex = _get_contact_numbers(self._session, cid, numbers[0])
428                 else:
429                         defaultIndex = 0
430
431                 for number, description in numbers:
432                         if description:
433                                 label = "%s - %s" % (number, description)
434                         else:
435                                 label = number
436                         selector.addItem(label)
437                 selector.setVisible(True)
438                 if 1 < len(numbers):
439                         selector.setEnabled(True)
440                         selector.setCurrentIndex(defaultIndex)
441                 else:
442                         selector.setEnabled(False)
443                 callback = functools.partial(
444                         self._on_change_number,
445                         cid
446                 )
447                 callback.__name__ = "thanks partials for not having names and pyqt for requiring them"
448                 selector.currentIndexChanged.connect(
449                         QtCore.pyqtSlot(int)(callback)
450                 )
451
452         def _scroll_to_bottom(self):
453                 self._scrollTimer.start()
454
455         @misc_utils.log_exception(_moduleLogger)
456         def _on_delayed_scroll_to_bottom(self):
457                 self._scrollEntry.ensureWidgetVisible(self._smsEntry)
458
459         @misc_utils.log_exception(_moduleLogger)
460         def _on_sms_clicked(self, arg):
461                 message = unicode(self._smsEntry.toPlainText())
462                 self._session.draft.send(message)
463                 self._smsEntry.setPlainText("")
464
465         @misc_utils.log_exception(_moduleLogger)
466         def _on_call_clicked(self, arg):
467                 self._session.draft.call()
468                 self._smsEntry.setPlainText("")
469
470         @QtCore.pyqtSlot()
471         @misc_utils.log_exception(_moduleLogger)
472         def _on_cancel_clicked(self, message):
473                 self._session.draft.cancel()
474
475         @misc_utils.log_exception(_moduleLogger)
476         def _on_remove_contact(self, cid, toggled):
477                 self._session.draft.remove_contact(cid)
478
479         @misc_utils.log_exception(_moduleLogger)
480         def _on_change_number(self, cid, index):
481                 # Exception thrown when the first item is removed
482                 numbers = self._session.draft.get_numbers(cid)
483                 number = numbers[index][0]
484                 self._session.draft.set_selected_number(cid, number)
485
486         @QtCore.pyqtSlot()
487         @misc_utils.log_exception(_moduleLogger)
488         def _on_recipients_changed(self):
489                 self._update_recipients()
490
491         @QtCore.pyqtSlot()
492         @misc_utils.log_exception(_moduleLogger)
493         def _on_op_started(self):
494                 self._smsEntry.setReadOnly(True)
495                 self._smsButton.setVisible(False)
496                 self._dialButton.setVisible(False)
497                 self._window.show()
498
499         @QtCore.pyqtSlot()
500         @misc_utils.log_exception(_moduleLogger)
501         def _on_calling_started(self):
502                 self._cancelButton.setVisible(True)
503
504         @QtCore.pyqtSlot()
505         @misc_utils.log_exception(_moduleLogger)
506         def _on_op_finished(self):
507                 self._window.hide()
508
509                 self._smsEntry.setReadOnly(False)
510                 self._cancelButton.setVisible(False)
511                 self._smsButton.setVisible(True)
512                 self._dialButton.setVisible(True)
513
514         @QtCore.pyqtSlot()
515         @misc_utils.log_exception(_moduleLogger)
516         def _on_op_error(self, message):
517                 self._smsEntry.setReadOnly(False)
518                 self._cancelButton.setVisible(False)
519                 self._smsButton.setVisible(True)
520                 self._dialButton.setVisible(True)
521
522                 self._errorLog.push_error(message)
523
524         @QtCore.pyqtSlot()
525         @misc_utils.log_exception(_moduleLogger)
526         def _on_letter_count_changed(self):
527                 self._update_letter_count()
528                 self._update_button_state()
529
530         @QtCore.pyqtSlot()
531         @QtCore.pyqtSlot(bool)
532         @misc_utils.log_exception(_moduleLogger)
533         def _on_close_window(self, checked = True):
534                 self._window.hide()
535
536
537 def _get_contact_numbers(session, contactId, numberDescription):
538         contactPhoneNumbers = []
539         if contactId and contactId != "0":
540                 try:
541                         contactDetails = copy.deepcopy(session.get_contacts()[contactId])
542                         contactPhoneNumbers = contactDetails["numbers"]
543                 except KeyError:
544                         contactPhoneNumbers = []
545                 contactPhoneNumbers = [
546                         (contactPhoneNumber["phoneNumber"], contactPhoneNumber["phoneType"])
547                         for contactPhoneNumber in contactPhoneNumbers
548                 ]
549                 if contactPhoneNumbers:
550                         uglyContactNumbers = (
551                                 misc_utils.make_ugly(contactNumber)
552                                 for (contactNumber, _) in contactPhoneNumbers
553                         )
554                         defaultMatches = [
555                                 misc_utils.similar_ugly_numbers(numberDescription[0], contactNumber)
556                                 for contactNumber in uglyContactNumbers
557                         ]
558                         try:
559                                 defaultIndex = defaultMatches.index(True)
560                         except ValueError:
561                                 contactPhoneNumbers.append(numberDescription)
562                                 defaultIndex = len(contactPhoneNumbers)-1
563                                 _moduleLogger.warn(
564                                         "Could not find contact %r's number %s among %r" % (
565                                                 contactId, numberDescription, contactPhoneNumbers
566                                         )
567                                 )
568
569         if not contactPhoneNumbers:
570                 contactPhoneNumbers = [numberDescription]
571                 defaultIndex = -1
572
573         return contactPhoneNumbers, defaultIndex