class SupportDialog(QObject): SHORT_FEEDBACK_INTERVAL = 5 * 60 * 1000 DAYS_TO_FEEDBACK = 7 DROPDOWN_BACKGROUND_COLOR = "#f78d1e" DROPDOWN_COLOR = "white" _sending_error = Signal() SUBJECT = { 1: "TECHNICAL", 2: "OTHER", 3: "FEEDBACK", } def __init__(self, parent, parent_window, config, dp=1, selected_index=0): QObject.__init__(self, parent) self._parent = parent self._parent_window = parent_window self._config = config self._dp = dp self._selected_index = selected_index self._dialog = QDialog(parent_window) self._dialog.setWindowFlags(Qt.Dialog) self._dialog.setAttribute(Qt.WA_MacFrameworkScaled) self._is_opened = False self._pipe = None self._feedback_mode = False self._ui = Ui_Dialog() self._ui.setupUi(self._dialog) self._init_ui() self._old_close_event = self._dialog.closeEvent self._feedback_timer = QTimer(self) self._feedback_timer.setSingleShot(True) self._feedback_timer.timeout.connect(self._show_feedback_form) self._parent.service_started.connect(self._check_feedback_needed) self._parent.exit_request.connect(self._on_exit_request) def _init_ui(self): self._ui.pushButton.setEnabled(False) self._ui.comboBox.addItem(tr("---Select Subject---")) self._ui.comboBox.addItem(tr("Technical Question")) self._ui.comboBox.addItem(tr("Other Question")) self._ui.comboBox.addItem(tr("Feedback")) self._ui.comboBox.setCurrentIndex(self._selected_index) palette = self._ui.comboBox.palette() palette.setColor(QPalette.HighlightedText, QColor(self.DROPDOWN_COLOR)) palette.setColor(QPalette.Highlight, QColor(self.DROPDOWN_BACKGROUND_COLOR)) self._ui.comboBox.setPalette(palette) palette = self._ui.comboBox.view().palette() palette.setColor(QPalette.HighlightedText, QColor(self.DROPDOWN_COLOR)) palette.setColor(QPalette.Highlight, QColor(self.DROPDOWN_BACKGROUND_COLOR)) self._ui.comboBox.view().setPalette(palette) self._set_tooltip() self._ui.comboBox.currentIndexChanged.connect(self._on_index_changed) self._ui.plainTextEdit.textChanged.connect(self._set_tooltip) self._ui.pushButton.clicked.connect(self._on_send_clicked) self._ui.text_label.linkActivated.connect(self._on_link_activated) self._sending_error.connect(self._clear_pipe_state) self._set_fonts() def _set_fonts(self): ui = self._ui controls = [ ui.plainTextEdit, ui.pushButton, ui.comboBox, ui.text_label, ui.checkBox ] for control in controls: font = control.font() font_size = control.font().pointSize() * self._dp if font_size > 0: control_font = QFont(font.family(), font_size) control_font.setBold(font.bold()) control.setFont(control_font) def set_selected_index(self, selected_index): self._selected_index = selected_index if self._is_opened: self._ui.comboBox.setCurrentIndex(self._selected_index) def show(self): if self._parent.dialogs_opened(): return self._is_opened = True logger.debug("Support dialog opening...") self._pipe = None self.set_selected_index(self._selected_index) self._ui.checkBox.setChecked(False) self._ui.comboBox.setEnabled(not self._feedback_mode) self._dialog.exec_() logger.debug("Support dialog closed") if self._pipe: try: self._pipe.stop() self._clear_pipe_state() except Exception as e: logger.error("Unexpected error stopping pipe: (%s)", e) self._is_opened = False self._selected_index = 0 self._ui.plainTextEdit.document().clear() self._ui.checkBox.setChecked(False) def dialog_opened(self): return self._is_opened def close(self): self._dialog.close() def _set_tooltip(self): if not self._selected_index: tooltip = tr("Please select subject") self._ui.pushButton.setEnabled(False) elif not self._ui.plainTextEdit.document().toPlainText(): tooltip = tr("Message can't be empty") self._ui.pushButton.setEnabled(False) else: tooltip = tr("Click to send message") self._ui.pushButton.setEnabled(True) self._ui.pushButton.setToolTip(tooltip) def _on_index_changed(self, selected_index): self._selected_index = selected_index self._set_tooltip() def _on_send_clicked(self): self._dialog.setEnabled(False) self._pipe = ProgressPipe(self, self._ui.pushButton, timeout=1000, final_text=tr("Sent"), final_timeout=500) self._pipe.pipe_finished.connect(self._on_pipe_finished) if self._ui.checkBox.isChecked(): self._pipe.add_task(tr("Compressing"), self._archive_logs()) self._pipe.add_task(tr("Uploading"), self._upload_file()) self._pipe.add_task(tr("Sending"), self._send_message()) self._pipe.start() def _on_pipe_finished(self): self._clear_feedback_flag() self.close() def _clear_pipe_state(self): self._dialog.setEnabled(True) try: self._pipe.pipe_finished.disconnect(self._on_pipe_finished) except Exception as e: logger.warning("Can't disconnect signal: %s", e) self._ui.pushButton.setText(tr("SEND")) def _send_message(self): def send(log_file_name=""): logger.debug("Support compressed log_file_name %s", log_file_name) if self._selected_index not in self.SUBJECT: logger.warning("Attempt to send message to support " "with invalid subject") return subject = self.SUBJECT[self._selected_index] res = self._parent.web_api.send_support_message( subject, self._ui.plainTextEdit.document().toPlainText(), log_file_name) was_error = False msg = tr("Can't send message to support") if res and "result" in res: if res["result"] != "success": was_error = True msg = str(res.get("info", msg)) else: was_error = True if was_error: self._parent.show_tray_notification(msg) self._sending_error.emit() raise SendingError(msg) return send def _archive_logs(self): def archive(): # uses function attributes to track progress # archive.size, archive.progress, archive.stop logs_dir = get_bases_dir(self._config.sync_directory) log_files = glob("{}{}*.log".format(logs_dir, os.sep)) log_sizes = list(map(os.path.getsize, log_files)) # mark overall size archive.size = sum(log_sizes) old_archives = glob("{}{}2*_logs.zip".format(logs_dir, os.sep)) try: list(map(remove_file, old_archives)) except Exception as e: logger.warning("Can't delete old archives. Reason: (%s)", e) if get_free_space(logs_dir) < archive.size // 5: # archive.size // 5 is approx future archive size msg = tr("Insufficient disk space to archive logs. " "Please clean disk") self._parent.show_tray_notification(msg) self._sending_error.emit() raise SendingError(msg) archive_name = time.strftime('%Y%m%d_%H%M%S_logs.zip') archive_path = "{}{}{}".format(logs_dir, os.sep, archive_name) archive_dir = op.dirname(archive_path) f = zipfile.ZipFile(archive_path, "w", compression=zipfile.ZIP_DEFLATED, compresslevel=9) try: with cwd(archive_dir): for i, log_file in enumerate(log_files): if not op.isfile(log_file): continue f.write(op.basename(log_file)) # mark progress archive.progress += log_sizes[i] if archive.stop: return except Exception as e: msg = tr("Can't archive logs.") logger.warning(msg + " Reason: (%s)", e) self._parent.show_tray_notification(msg) self._sending_error.emit() raise SendingError(msg) finally: f.close() if archive.stop: remove_file(archive_path) return archive_path return archive def _upload_file(self): def upload(path): # uses function attributes to track progress # upload.size, upload.progress, upload.stop upload.size = op.getsize(path) res = self._parent.web_api.upload_file(path, "application/zip", callback) was_error = False msg = tr("Can't upload archive file") if res and "result" in res: if res["result"] == "success": filename = res.get("file_name", "") else: was_error = True msg = str(res.get("info", msg)) else: was_error = True if was_error and not upload.stop: self._parent.show_tray_notification(msg) self._sending_error.emit() raise SendingError(msg) remove_file(path) return filename def callback(monitor): upload.progress = monitor.bytes_read if upload.stop: raise SendingError("Stopped") return upload def _on_link_activated(self): open_link(self._parent.get_help_uri())() self.close() def _check_feedback_needed(self): if self._feedback_timer.isActive(): return start_date = get_init_done() now = datetime.datetime.now() logger.debug("Start date is %s", start_date) if start_date is None: return interval = (start_date - now) \ + datetime.timedelta(days=self.DAYS_TO_FEEDBACK) if interval.total_seconds() <= 0: logger.debug("Feedback form date time is now") self._show_feedback_form() else: self._feedback_timer.setInterval(interval.seconds * 1000) self._feedback_timer.start() logger.debug("Feedback form date time is %s", now + interval) def _show_feedback_form(self): if self._is_opened: self._feedback_timer.setInterval(self.SHORT_FEEDBACK_INTERVAL) self._feedback_timer.start() return self._feedback_mode = True self._selected_index = 3 self._dialog.closeEvent = self._close_event window_title = self._dialog.windowTitle() label_text = self._ui.text_label.text() feedback_text = tr("Please leave your feedback for Pvtbox") self._ui.text_label.setText( "<html><head/><body><p>{}</p></body></html>".format(feedback_text)) self._dialog.setWindowTitle(tr("Feedback")) self.show() self._dialog.setWindowTitle(window_title) self._ui.text_label.setText(label_text) def _close_event(self, event): if event.spontaneous(): self._clear_feedback_flag() self._old_close_event(event) def _clear_feedback_flag(self): if self._feedback_mode: logger.debug("Feedback flag cleared") clear_init_done() self._feedback_mode = False self._dialog.closeEvent = self._old_close_event def _on_exit_request(self): if self._is_opened: self.close()
class InsertLinkDialog(object): def __init__(self, parent, dp=None, signal_server_address=''): self._dialog = QDialog(parent) self._dp = dp self._parent = parent self._link = '' self._password = '' self._is_shared = True self._password_mode = False self._signal_server_address = signal_server_address self._dialog.setWindowIcon(QIcon(':/images/icon.png')) self._ui = Ui_insert_link_dialog() self._ui.setupUi(self._dialog) self._init_ui() self._cant_validate = tr("Cannot validate share link") def _init_ui(self): self._dialog.setAttribute(Qt.WA_MacFrameworkScaled) self._hide_error() self._ok_button = self._ui.ok_button self._ui.cancel_button.clicked.connect(self._dialog.reject) self._ok_button.clicked.connect(self._ok_clicked) self._ui.link_line_edit.textChanged.connect(self._text_changed) self._set_fonts() def _set_fonts(self): controls = [] controls.extend([c for c in self._dialog.findChildren(QLabel)]) controls.extend( [c for c in self._dialog.findChildren(QLineEdit)]) controls.extend( [c for c in self._dialog.findChildren(QPushButton)]) for control in controls: font = control.font() font_size = control.font().pointSize() * self._dp if font_size > 0: control.setFont(QFont(font.family(), font_size)) def _ok_clicked(self): validated = self._validate() if validated is None: self._change_mode() elif validated: if self._password_mode: self._password = self._ui.link_line_edit.text() else: self._link = self._ui.link_line_edit.text() self._dialog.accept() def _text_changed(self, *args, **kwargs): self._hide_error() def _validate(self): if self._is_shared: return self._validate_shared() else: return self._validate_network_file() def _validate_shared(self): if self._password_mode: return self._validate_password() if not self._validate_scheme(): return False return self._check_share_link() def _validate_scheme(self): share_url = self._ui.link_line_edit.text() pr = urlparse(share_url) success = pr.scheme in ('http', 'https', 'pvtbox') and pr.path share_hash = pr.path.split('/')[-1] success = success and share_hash and len(share_hash) == 32 if not success: self._show_error() return success def _check_share_link(self): self._lock_screen() share_url = self._link if self._password_mode \ else self._ui.link_line_edit.text() pr = urlparse(share_url) share_hash = pr.path.split('/')[-1] param_str = '' if self._password_mode: password = self._ui.link_line_edit.text() password = base64.b64encode( bytes(password, 'utf-8')).decode('utf-8') params = {"passwd": password} query = urlencode(params, encoding='utf-8') param_str = '?{}'.format(query) url = 'https://{}/ws/webshare/{}{}'.format( self._signal_server_address, share_hash, param_str) logger.debug("url %s", url) error = '' try: response = urlopen(url, timeout=1) status = response.status except HTTPError as e: logger.warning("Request to signal server returned error %s", e) status = e.code response = str(e.read(), encoding='utf-8') except URLError as e: logger.warning("Request to signal server returned url error %s", e) self._show_error(self._cant_validate) self._unlock_screen() return False logger.debug("request status %s", status) if status == 400: if self._password_mode: self._link += param_str success = True else: success, error = self._parse_response(response) if success is False: self._show_error(error) self._unlock_screen() return success def _parse_response(self, response): try: data = json.loads(response) err_code = data.get("errcode", '') info = data.get("info", '') if err_code == 'SHARE_WRONG_PASSWORD': success = None if not self._password_mode else False error = '' elif err_code == 'LOCKED_CAUSE_TOO_MANY_BAD_LOGIN': success = False error = tr('Locked after too many incorrect attempts') elif err_code == 'SHARE_NOT_FOUND': success = False error = '' else: success = False error = info if info else self._cant_validate except Exception as e: logger.warning("Can't parse response (%s). reason: %s", response, e) success = False error = self._cant_validate return success, error def _validate_password(self): if not self._ui.link_line_edit.text(): self._show_error() return False return self._check_share_link() def _validate_network_file(self): # place code to validate network file link here return False def _change_mode(self): assert not self._password_mode, \ "Must not be in password mode in changing mode" logger.debug("Changing to password mode") self._password_mode = True self._dialog.setWindowTitle(tr("Insert password")) self._link = self._ui.link_line_edit.text() self._ui.link_line_edit.setText('') self._ui.link_line_edit.setPlaceholderText(tr("Insert password here")) self._ui.link_line_edit.setEchoMode(QLineEdit.Password) self._hide_error() def _show_error(self, error_text=''): if not error_text: link_text = self._ui.link_line_edit.text() error_text = tr("Please insert share link") \ if not self._password_mode and not link_text \ else tr("Invalid link") if not self._password_mode \ else tr("Password can not be empty") if not link_text \ else tr("Wrong password") self._ui.error_label.setText(error_text) self._ui.link_line_edit.setFocus() def _hide_error(self): self._ui.error_label.setText('') self._ui.link_line_edit.setFocus() def _lock_screen(self): self._ok_button.setText(tr("Processing...")) self._dialog.setEnabled(False) self._dialog.repaint() def _unlock_screen(self): self._ok_button.setText(tr("Ok")) self._dialog.setEnabled(True) self._dialog.repaint() def show(self): logger.debug("Opening insert link dialog") if self._dialog.exec_() == QDialog.Rejected: self._link = '' self._password = '' self._is_shared = True logger.verbose("link (%s), password (%s)", self._link, self._password) return self._link, self._is_shared