3 from __future__ import with_statement
4 from __future__ import division
10 from PyQt4 import QtGui
11 from PyQt4 import QtCore
13 from util import qui_utils
14 from util import misc as misc_utils
17 _moduleLogger = logging.getLogger(__name__)
20 class CredentialsDialog(object):
22 def __init__(self, app):
23 self._usernameField = QtGui.QLineEdit()
24 self._passwordField = QtGui.QLineEdit()
25 self._passwordField.setEchoMode(QtGui.QLineEdit.PasswordEchoOnEdit)
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)
33 self._loginButton = QtGui.QPushButton("&Login")
34 self._buttonLayout = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Cancel)
35 self._buttonLayout.addButton(self._loginButton, QtGui.QDialogButtonBox.AcceptRole)
37 self._layout = QtGui.QVBoxLayout()
38 self._layout.addLayout(self._credLayout)
39 self._layout.addWidget(self._buttonLayout)
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)
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)
54 self._dialog.addAction(self._closeWindowAction)
55 self._dialog.addAction(app.quitAction)
56 self._dialog.addAction(app.fullscreenAction)
58 def run(self, defaultUsername, defaultPassword, parent=None):
59 self._dialog.setParent(parent, QtCore.Qt.Dialog)
61 self._usernameField.setText(defaultUsername)
62 self._passwordField.setText(defaultPassword)
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")
70 raise RuntimeError("Unknown Response")
72 self._dialog.setParent(None, QtCore.Qt.Dialog)
75 @QtCore.pyqtSlot(bool)
76 @misc_utils.log_exception(_moduleLogger)
77 def _on_close_window(self, checked = True):
81 class AccountDialog(object):
83 # @bug Can't enter custom callback numbers
85 def __init__(self, app):
88 self._accountNumberLabel = QtGui.QLabel("NUMBER NOT SET")
89 self._clearButton = QtGui.QPushButton("Clear Account")
90 self._clearButton.clicked.connect(self._on_clear)
92 self._callbackSelector = QtGui.QComboBox()
93 #self._callbackSelector.setEditable(True)
94 self._callbackSelector.setInsertPolicy(QtGui.QComboBox.InsertAtTop)
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)
105 self._loginButton = QtGui.QPushButton("&Apply")
106 self._buttonLayout = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Cancel)
107 self._buttonLayout.addButton(self._loginButton, QtGui.QDialogButtonBox.AcceptRole)
109 self._layout = QtGui.QVBoxLayout()
110 self._layout.addLayout(self._credLayout)
111 self._layout.addWidget(self._buttonLayout)
113 self._dialog = QtGui.QDialog()
114 self._dialog.setWindowTitle("Account")
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)
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)
125 self._dialog.addAction(self._closeWindowAction)
126 self._dialog.addAction(app.quitAction)
127 self._dialog.addAction(app.fullscreenAction)
133 accountNumber = property(
134 lambda self: str(self._accountNumberLabel.text()),
135 lambda self, num: self._accountNumberLabel.setText(num),
139 def selectedCallback(self):
140 index = self._callbackSelector.currentIndex()
141 data = str(self._callbackSelector.itemData(index).toPyObject())
144 def set_callbacks(self, choices, default):
145 self._callbackSelector.clear()
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)
154 self._callbackSelector.addItem("%s - %s" % (prettyNumber, description), uglyNumber)
155 if uglyNumber == uglyDefault:
156 self._callbackSelector.setCurrentIndex(i)
159 def run(self, parent=None):
160 self._doClear = False
161 self._dialog.setParent(parent)
163 response = self._dialog.exec_()
167 @QtCore.pyqtSlot(bool)
168 def _on_clear(self, checked = False):
170 self._dialog.accept()
173 @QtCore.pyqtSlot(bool)
174 @misc_utils.log_exception(_moduleLogger)
175 def _on_close_window(self, checked = True):
176 self._dialog.reject()
179 class SMSEntryWindow(object):
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
196 self._errorDisplay = qui_utils.ErrorDisplay(self._errorLog)
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)
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)
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)
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)
240 self._layout = QtGui.QVBoxLayout()
241 self._layout.addWidget(self._errorDisplay.toplevel)
242 self._layout.addWidget(self._scrollEntry)
243 self._layout.addLayout(self._buttonLayout)
245 centralWidget = QtGui.QWidget()
246 centralWidget.setLayout(self._layout)
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)
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)
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)
265 self._scrollTimer = QtCore.QTimer()
266 self._scrollTimer.setInterval(0)
267 self._scrollTimer.setSingleShot(True)
268 self._scrollTimer.timeout.connect(self._on_delayed_scroll_to_bottom)
271 self._update_recipients()
273 def _update_letter_count(self):
274 count = self._smsEntry.toPlainText().size()
275 numTexts, numCharInText = divmod(count, self.MAX_CHAR)
277 numCharsLeftInText = self.MAX_CHAR - numCharInText
278 self._characterCountLabel.setText("%d (%d)" % (numCharsLeftInText, numTexts))
280 def _update_button_state(self):
281 if self._session.draft.get_num_contacts() == 0:
282 self._dialButton.setEnabled(False)
283 self._smsButton.setEnabled(False)
284 elif self._session.draft.get_num_contacts() == 1:
285 count = self._smsEntry.toPlainText().size()
287 self._dialButton.setEnabled(True)
288 self._smsButton.setEnabled(False)
290 self._dialButton.setEnabled(False)
291 self._smsButton.setEnabled(True)
293 self._dialButton.setEnabled(False)
294 self._smsButton.setEnabled(True)
296 def _update_recipients(self):
297 draftContactsCount = self._session.draft.get_num_contacts()
298 if draftContactsCount == 0:
300 elif draftContactsCount == 1:
301 (cid, ) = self._session.draft.get_contacts()
302 title = self._session.draft.get_title(cid)
303 description = self._session.draft.get_description(cid)
304 numbers = self._session.draft.get_numbers(cid)
306 self._targetList.setVisible(False)
308 self._history.setText(description)
309 self._history.setVisible(True)
311 self._history.setText("")
312 self._history.setVisible(False)
313 self._populate_number_selector(self._singleNumberSelector, cid, numbers)
315 self._scroll_to_bottom()
316 self._window.setWindowTitle(title)
318 self._smsEntry.setFocus(QtCore.Qt.OtherFocusReason)
320 self._targetList.setVisible(True)
321 while self._targetLayout.count():
322 removedLayoutItem = self._targetLayout.takeAt(self._targetLayout.count()-1)
323 removedWidget = removedLayoutItem.widget()
324 removedWidget.close()
325 for cid in self._session.draft.get_contacts():
326 title = self._session.draft.get_title(cid)
327 description = self._session.draft.get_description(cid)
328 numbers = self._session.draft.get_numbers(cid)
330 titleLabel = QtGui.QLabel(title)
331 numberSelector = QtGui.QComboBox()
332 self._populate_number_selector(numberSelector, cid, numbers)
333 deleteButton = QtGui.QPushButton("Delete")
334 callback = functools.partial(
335 self._on_remove_contact,
338 callback.__name__ = "b"
339 deleteButton.clicked.connect(callback)
341 rowLayout = QtGui.QHBoxLayout()
342 rowLayout.addWidget(titleLabel)
343 rowLayout.addWidget(numberSelector)
344 rowLayout.addWidget(deleteButton)
345 rowWidget = QtGui.QWidget()
346 rowWidget.setLayout(rowLayout)
347 self._targetLayout.addWidget(rowWidget)
348 self._history.setText("")
349 self._history.setVisible(False)
350 self._singleNumberSelector.setVisible(False)
352 self._scroll_to_bottom()
353 self._window.setWindowTitle("Contacts")
355 self._smsEntry.setFocus(QtCore.Qt.OtherFocusReason)
357 def _populate_number_selector(self, selector, cid, numbers):
360 if len(numbers) == 1:
361 numbers, defaultIndex = _get_contact_numbers(self._session, cid, numbers[0])
365 for number, description in numbers:
367 label = "%s - %s" % (number, description)
370 selector.addItem(label)
371 selector.setVisible(True)
373 selector.setEnabled(True)
374 selector.setCurrentIndex(defaultIndex)
376 selector.setEnabled(False)
377 callback = functools.partial(
378 self._on_change_number,
381 callback.__name__ = "thanks partials for not having names and pyqt for requiring them"
382 selector.currentIndexChanged.connect(
383 QtCore.pyqtSlot(int)(callback)
386 def _scroll_to_bottom(self):
387 self._scrollTimer.start()
389 @misc_utils.log_exception(_moduleLogger)
390 def _on_delayed_scroll_to_bottom(self):
391 self._scrollEntry.ensureWidgetVisible(self._smsEntry)
393 @misc_utils.log_exception(_moduleLogger)
394 def _on_sms_clicked(self, arg):
395 message = str(self._smsEntry.toPlainText())
396 self._session.draft.send(message)
397 self._smsEntry.setPlainText("")
399 @misc_utils.log_exception(_moduleLogger)
400 def _on_call_clicked(self, arg):
401 self._session.draft.call()
402 self._smsEntry.setPlainText("")
405 @misc_utils.log_exception(_moduleLogger)
406 def _on_cancel_clicked(self, message):
407 self._session.draft.cancel()
409 @misc_utils.log_exception(_moduleLogger)
410 def _on_remove_contact(self, cid, toggled):
411 self._session.draft.remove_contact(cid)
413 @misc_utils.log_exception(_moduleLogger)
414 def _on_change_number(self, cid, index):
415 # Exception thrown when the first item is removed
416 numbers = self._session.draft.get_numbers(cid)
417 number = numbers[index][0]
418 self._session.draft.set_selected_number(cid, number)
421 @misc_utils.log_exception(_moduleLogger)
422 def _on_recipients_changed(self):
423 self._update_recipients()
426 @misc_utils.log_exception(_moduleLogger)
427 def _on_op_started(self):
428 self._smsEntry.setReadOnly(True)
429 self._smsButton.setVisible(False)
430 self._dialButton.setVisible(False)
434 @misc_utils.log_exception(_moduleLogger)
435 def _on_calling_started(self):
436 self._cancelButton.setVisible(True)
439 @misc_utils.log_exception(_moduleLogger)
440 def _on_op_finished(self):
443 self._smsEntry.setReadOnly(False)
444 self._cancelButton.setVisible(False)
445 self._smsButton.setVisible(True)
446 self._dialButton.setVisible(True)
449 @misc_utils.log_exception(_moduleLogger)
450 def _on_op_error(self, message):
451 self._smsEntry.setReadOnly(False)
452 self._cancelButton.setVisible(False)
453 self._smsButton.setVisible(True)
454 self._dialButton.setVisible(True)
456 self._errorLog.push_message(message)
459 @misc_utils.log_exception(_moduleLogger)
460 def _on_letter_count_changed(self):
461 self._update_letter_count()
462 self._update_button_state()
465 @QtCore.pyqtSlot(bool)
466 @misc_utils.log_exception(_moduleLogger)
467 def _on_close_window(self, checked = True):
471 def _get_contact_numbers(session, contactId, numberDescription):
472 contactPhoneNumbers = []
473 if contactId and contactId != "0":
475 contactDetails = copy.deepcopy(session.get_contacts()[contactId])
476 contactPhoneNumbers = contactDetails["numbers"]
478 contactPhoneNumbers = []
479 contactPhoneNumbers = [
480 (contactPhoneNumber["phoneNumber"], contactPhoneNumber["phoneType"])
481 for contactPhoneNumber in contactPhoneNumbers
483 if contactPhoneNumbers:
484 uglyContactNumbers = (
485 misc_utils.make_ugly(contactNumber)
486 for (contactNumber, _) in contactPhoneNumbers
489 misc_utils.similar_ugly_numbers(numberDescription[0], contactNumber)
490 for contactNumber in uglyContactNumbers
493 defaultIndex = defaultMatches.index(True)
495 contactPhoneNumbers.append(numberDescription)
496 defaultIndex = len(contactPhoneNumbers)-1
498 "Could not find contact %r's number %s among %r" % (
499 contactId, numberDescription, contactPhoneNumbers
503 if not contactPhoneNumbers:
504 contactPhoneNumbers = [numberDescription]
507 return contactPhoneNumbers, defaultIndex