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 def __init__(self, app):
86 self._accountNumberLabel = QtGui.QLabel("NUMBER NOT SET")
87 self._clearButton = QtGui.QPushButton("Clear Account")
88 self._clearButton.clicked.connect(self._on_clear)
90 self._credLayout = QtGui.QGridLayout()
91 self._credLayout.addWidget(QtGui.QLabel("Account"), 0, 0)
92 self._credLayout.addWidget(self._accountNumberLabel, 0, 1)
93 self._credLayout.addWidget(QtGui.QLabel("Callback"), 1, 0)
94 self._credLayout.addWidget(QtGui.QLabel(""), 2, 0)
95 self._credLayout.addWidget(self._clearButton, 2, 1)
96 self._credLayout.addWidget(QtGui.QLabel(""), 3, 0)
98 self._loginButton = QtGui.QPushButton("&Apply")
99 self._buttonLayout = QtGui.QDialogButtonBox(QtGui.QDialogButtonBox.Cancel)
100 self._buttonLayout.addButton(self._loginButton, QtGui.QDialogButtonBox.AcceptRole)
102 self._layout = QtGui.QVBoxLayout()
103 self._layout.addLayout(self._credLayout)
104 self._layout.addWidget(self._buttonLayout)
106 self._dialog = QtGui.QDialog()
107 self._dialog.setWindowTitle("Login")
108 self._dialog.setLayout(self._layout)
109 qui_utils.set_autorient(self._dialog, True)
110 self._buttonLayout.accepted.connect(self._dialog.accept)
111 self._buttonLayout.rejected.connect(self._dialog.reject)
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)
118 self._dialog.addAction(self._closeWindowAction)
119 self._dialog.addAction(app.quitAction)
120 self._dialog.addAction(app.fullscreenAction)
126 accountNumber = property(
127 lambda self: str(self._accountNumberLabel.text()),
128 lambda self, num: self._accountNumberLabel.setText(num),
131 def run(self, parent=None):
132 self._doClear = False
133 self._dialog.setParent(parent)
135 response = self._dialog.exec_()
139 @QtCore.pyqtSlot(bool)
140 def _on_clear(self, checked = False):
142 self._dialog.accept()
145 @QtCore.pyqtSlot(bool)
146 @misc_utils.log_exception(_moduleLogger)
147 def _on_close_window(self, checked = True):
148 self._dialog.reject()
151 class SMSEntryWindow(object):
153 def __init__(self, parent, app, session, errorLog):
155 self._session = session
156 self._session.draft.recipientsChanged.connect(self._on_recipients_changed)
157 self._session.draft.called.connect(self._on_op_finished)
158 self._session.draft.sentMessage.connect(self._on_op_finished)
159 self._session.draft.cancelled.connect(self._on_op_finished)
160 self._errorLog = errorLog
162 self._targetLayout = QtGui.QVBoxLayout()
163 self._targetList = QtGui.QWidget()
164 self._targetList.setLayout(self._targetLayout)
165 self._history = QtGui.QTextEdit()
166 self._smsEntry = QtGui.QTextEdit()
167 self._smsEntry.textChanged.connect(self._on_letter_count_changed)
169 self._entryLayout = QtGui.QVBoxLayout()
170 self._entryLayout.addWidget(self._targetList)
171 self._entryLayout.addWidget(self._history)
172 self._entryLayout.addWidget(self._smsEntry)
173 self._entryWidget = QtGui.QWidget()
174 self._entryWidget.setLayout(self._entryLayout)
175 self._scrollEntry = QtGui.QScrollArea()
176 self._scrollEntry.setWidget(self._entryWidget)
177 self._scrollEntry.setAlignment(QtCore.Qt.AlignLeft | QtCore.Qt.AlignBottom)
178 self._scrollEntry.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)
179 self._scrollEntry.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
181 self._characterCountLabel = QtGui.QLabel("Letters: %s" % 0)
182 self._singleNumberSelector = QtGui.QComboBox()
183 self._smsButton = QtGui.QPushButton("SMS")
184 self._smsButton.clicked.connect(self._on_sms_clicked)
185 self._dialButton = QtGui.QPushButton("Dial")
186 self._dialButton.clicked.connect(self._on_call_clicked)
188 self._buttonLayout = QtGui.QHBoxLayout()
189 self._buttonLayout.addWidget(self._characterCountLabel)
190 self._buttonLayout.addWidget(self._singleNumberSelector)
191 self._buttonLayout.addWidget(self._smsButton)
192 self._buttonLayout.addWidget(self._dialButton)
194 self._layout = QtGui.QVBoxLayout()
195 self._layout.addWidget(self._scrollEntry)
196 self._layout.addLayout(self._buttonLayout)
198 centralWidget = QtGui.QWidget()
199 centralWidget.setLayout(self._layout)
201 self._window = QtGui.QMainWindow(parent)
202 qui_utils.set_autorient(self._window, True)
203 qui_utils.set_stackable(self._window, True)
204 self._window.setWindowTitle("Contact")
205 self._window.setCentralWidget(centralWidget)
207 self._closeWindowAction = QtGui.QAction(None)
208 self._closeWindowAction.setText("Close")
209 self._closeWindowAction.setShortcut(QtGui.QKeySequence("CTRL+w"))
210 self._closeWindowAction.triggered.connect(self._on_close_window)
212 fileMenu = self._window.menuBar().addMenu("&File")
213 fileMenu.addAction(self._closeWindowAction)
214 fileMenu.addAction(app.quitAction)
215 viewMenu = self._window.menuBar().addMenu("&View")
216 viewMenu.addAction(app.fullscreenAction)
219 self._update_recipients()
221 def _update_letter_count(self):
222 count = self._smsEntry.toPlainText().size()
223 self._characterCountLabel.setText("Letters: %s" % count)
225 def _update_button_state(self):
226 if len(self._contacts) == 0:
227 self._dialButton.setEnabled(False)
228 self._smsButton.setEnabled(False)
229 elif len(self._contacts) == 1:
230 count = self._smsEntry.toPlainText().size()
232 self._dialButton.setEnabled(True)
233 self._smsButton.setEnabled(False)
235 self._dialButton.setEnabled(False)
236 self._smsButton.setEnabled(True)
238 self._dialButton.setEnabled(False)
239 self._smsButton.setEnabled(True)
241 def _update_recipients(self):
242 draftContactsCount = self._session.draft.get_num_contacts()
243 if draftContactsCount == 0:
245 elif draftContactsCount == 1:
246 (cid, ) = self._session.draft.get_contacts()
247 title = self._session.draft.get_title(cid)
248 description = self._session.draft.get_description(cid)
249 numbers = self._session.draft.get_numbers(cid)
251 self._targetList.setVisible(False)
253 self._history.setHtml(description)
254 self._history.setVisible(True)
256 self._history.setHtml("")
257 self._history.setVisible(False)
258 self._populate_number_selector(self._singleNumberSelector, cid, numbers)
260 self._scroll_to_bottom()
261 self._window.setWindowTitle(title)
264 self._targetList.setVisible(True)
265 while self._targetLayout.count():
266 removedLayoutItem = self._targetLayout.takeAt(self._targetLayout.count()-1)
267 removedWidget = removedLayoutItem.widget()
268 removedWidget.close()
269 for cid in self._session.draft.get_contacts():
270 title = self._session.draft.get_title(cid)
271 description = self._session.draft.get_description(cid)
272 numbers = self._session.draft.get_numbers(cid)
274 titleLabel = QtGui.QLabel(title)
275 numberSelector = QtGui.QComboBox()
276 self._populate_number_selector(numberSelector, cid, numbers)
277 deleteButton = QtGui.QPushButton("Delete")
278 callback = functools.partial(
279 self._on_remove_contact,
282 callback.__name__ = "b"
283 deleteButton.clicked.connect(callback)
285 rowLayout = QtGui.QHBoxLayout()
286 rowLayout.addWidget(titleLabel)
287 rowLayout.addWidget(numberSelector)
288 rowLayout.addWidget(deleteButton)
289 rowWidget = QtGui.QWidget()
290 rowWidget.setLayout(rowLayout)
291 self._targetLayout.addWidget(rowWidget)
292 self._history.setHtml("")
293 self._history.setVisible(False)
294 self._singleNumberSelector.setVisible(False)
296 self._scroll_to_bottom()
297 self._window.setWindowTitle("Contacts")
300 def _populate_number_selector(self, selector, cid, numbers):
301 while 0 < selector.count():
302 selector.removeItem(0)
304 if len(numbers) == 1:
305 numbers, defaultIndex = _get_contact_numbers(self._session, cid, numbers[0])
309 for number, description in numbers:
311 label = "%s - %s" % (number, description)
314 selector.addItem(label)
315 selector.setVisible(True)
317 selector.setEnabled(True)
318 selector.setCurrentIndex(defaultIndex)
320 selector.setEnabled(False)
321 callback = functools.partial(
322 self._on_change_number,
325 callback.__name__ = "thanks partials for not having names and pyqt for requiring them"
326 selector.currentIndexChanged.connect(
327 QtCore.pyqtSlot(int)(callback)
330 def _scroll_to_bottom(self):
331 self._scrollEntry.ensureWidgetVisible(self._smsEntry)
334 @misc_utils.log_exception(_moduleLogger)
335 def _on_sms_clicked(self):
336 message = str(self._smsEntry.toPlainText())
337 self._session.draft.send(message)
338 self._smsEntry.setPlainText("")
341 @misc_utils.log_exception(_moduleLogger)
342 def _on_call_clicked(self):
343 self._session.draft.call()
344 self._smsEntry.setPlainText("")
346 @misc_utils.log_exception(_moduleLogger)
347 def _on_remove_contact(self, cid, toggled):
348 self._session.draft.remove_contact(cid)
350 @misc_utils.log_exception(_moduleLogger)
351 def _on_change_number(self, cid, index):
352 # Exception thrown when the first item is removed
353 numbers = self._session.draft.get_numbers(cid)
354 number = numbers[index][0]
355 self._session.draft.set_selected_number(cid, number)
358 @misc_utils.log_exception(_moduleLogger)
359 def _on_recipients_changed(self):
360 self._update_recipients()
363 @misc_utils.log_exception(_moduleLogger)
364 def _on_op_finished(self):
368 @misc_utils.log_exception(_moduleLogger)
369 def _on_letter_count_changed(self):
370 self._update_letter_count()
371 self._update_button_state()
374 @QtCore.pyqtSlot(bool)
375 @misc_utils.log_exception(_moduleLogger)
376 def _on_close_window(self, checked = True):
380 def _get_contact_numbers(session, contactId, numberDescription):
381 contactPhoneNumbers = []
382 if contactId and contactId != "0":
384 contactDetails = copy.deepcopy(session.get_contacts()[contactId])
385 contactPhoneNumbers = contactDetails["numbers"]
387 contactPhoneNumbers = []
388 contactPhoneNumbers = [
389 (contactPhoneNumber["phoneNumber"], contactPhoneNumber["phoneType"])
390 for contactPhoneNumber in contactPhoneNumbers
392 if contactPhoneNumbers:
393 uglyContactNumbers = (
394 misc_utils.make_ugly(contactNumber)
395 for (contactNumber, _) in contactPhoneNumbers
398 misc_utils.similar_ugly_numbers(numberDescription[0], contactNumber)
399 for contactNumber in uglyContactNumbers
402 defaultIndex = defaultMatches.index(True)
404 contactPhoneNumbers.append(numberDescription)
405 defaultIndex = len(contactPhoneNumbers)-1
407 "Could not find contact %r's number %s among %r" % (
408 contactId, numberDescription, contactPhoneNumbers
412 if not contactPhoneNumbers:
413 contactPhoneNumbers = [numberDescription]
416 return contactPhoneNumbers, defaultIndex