forked from openlp/openlp
150 lines
6.1 KiB
Python
150 lines
6.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
##########################################################################
|
|
# OpenLP - Open Source Lyrics Projection #
|
|
# ---------------------------------------------------------------------- #
|
|
# Copyright (c) 2008-2021 OpenLP Developers #
|
|
# ---------------------------------------------------------------------- #
|
|
# This program is free software: you can redistribute it and/or modify #
|
|
# it under the terms of the GNU General Public License as published by #
|
|
# the Free Software Foundation, either version 3 of the License, or #
|
|
# (at your option) any later version. #
|
|
# #
|
|
# This program is distributed in the hope that it will be useful, #
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of #
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the #
|
|
# GNU General Public License for more details. #
|
|
# #
|
|
# You should have received a copy of the GNU General Public License #
|
|
# along with this program. If not, see <https://www.gnu.org/licenses/>. #
|
|
##########################################################################
|
|
""" Patch the QFileDialog so it accepts and returns Path objects"""
|
|
from PyQt5 import QtCore, QtWidgets
|
|
|
|
from openlp.core.common.path import path_to_str, replace_params, str_to_path
|
|
from openlp.core.common.i18n import UiStrings, translate
|
|
|
|
|
|
class FileDialog(QtWidgets.QFileDialog):
|
|
@classmethod
|
|
def getExistingDirectory(cls, *args, **kwargs):
|
|
"""
|
|
Wraps `getExistingDirectory` so that it can be called with, and return Path objects
|
|
|
|
:type parent: QtWidgets.QWidget | None
|
|
:type caption: str
|
|
:type directory: pathlib.Path
|
|
:type options: QtWidgets.QFileDialog.Options
|
|
:rtype: pathlib.Path
|
|
"""
|
|
args, kwargs = replace_params(args, kwargs, ((2, 'directory', path_to_str),))
|
|
|
|
return_value = super().getExistingDirectory(*args, **kwargs)
|
|
|
|
# getExistingDirectory returns a str that represents the path. The string is empty if the user cancels the
|
|
# dialog.
|
|
return str_to_path(return_value)
|
|
|
|
@classmethod
|
|
def getOpenFileName(cls, *args, **kwargs):
|
|
"""
|
|
Wraps `getOpenFileName` so that it can be called with, and return Path objects
|
|
|
|
:type parent: QtWidgets.QWidget | None
|
|
:type caption: str
|
|
:type directory: pathlib.Path
|
|
:type filter: str
|
|
:type initialFilter: str
|
|
:type options: QtWidgets.QFileDialog.Options
|
|
:rtype: tuple[pathlib.Path, str]
|
|
"""
|
|
args, kwargs = replace_params(args, kwargs, ((2, 'directory', path_to_str),))
|
|
|
|
file_name, selected_filter = super().getOpenFileName(*args, **kwargs)
|
|
|
|
# getOpenFileName returns a tuple. The first item is a str that represents the path. The string is empty if
|
|
# the user cancels the dialog.
|
|
return str_to_path(file_name), selected_filter
|
|
|
|
@classmethod
|
|
def getOpenFileNames(cls, *args, **kwargs):
|
|
"""
|
|
Wraps `getOpenFileNames` so that it can be called with, and return Path objects
|
|
|
|
:type parent: QtWidgets.QWidget | None
|
|
:type caption: str
|
|
:type directory: pathlib.Path
|
|
:type filter: str
|
|
:type initialFilter: str
|
|
:type options: QtWidgets.QFileDialog.Options
|
|
:rtype: tuple[list[pathlib.Path], str]
|
|
"""
|
|
args, kwargs = replace_params(args, kwargs, ((2, 'directory', path_to_str),))
|
|
|
|
file_names, selected_filter = super().getOpenFileNames(*args, **kwargs)
|
|
|
|
# getSaveFileName returns a tuple. The first item is a list of str's that represents the path. The list is
|
|
# empty if the user cancels the dialog.
|
|
paths = [str_to_path(path) for path in file_names]
|
|
return paths, selected_filter
|
|
|
|
@classmethod
|
|
def getSaveFileName(cls, *args, **kwargs):
|
|
"""
|
|
Wraps `getSaveFileName` so that it can be called with, and return Path objects
|
|
|
|
:type parent: QtWidgets.QWidget | None
|
|
:type caption: str
|
|
:type directory: pathlib.Path
|
|
:type filter: str
|
|
:type initialFilter: str
|
|
:type options: QtWidgets.QFileDialog.Options
|
|
:rtype: tuple[pathlib.Path | None, str]
|
|
"""
|
|
args, kwargs = replace_params(args, kwargs, ((2, 'directory', path_to_str),))
|
|
|
|
file_name, selected_filter = super().getSaveFileName(*args, **kwargs)
|
|
|
|
# getSaveFileName returns a tuple. The first item represents the path as a str. The string is empty if the user
|
|
# cancels the dialog.
|
|
return str_to_path(file_name), selected_filter
|
|
|
|
|
|
class DownloadProgressDialog(QtWidgets.QProgressDialog):
|
|
"""
|
|
Local class to handle download display based and supporting httputils:get_web_page
|
|
"""
|
|
def __init__(self, parent, app):
|
|
super(DownloadProgressDialog, self).__init__(parent)
|
|
self.parent = parent
|
|
self.app = app
|
|
self.setWindowModality(QtCore.Qt.WindowModal)
|
|
self.setWindowTitle(translate('OpenLP.RemotePlugin', 'Importing Website'))
|
|
self.setLabelText(UiStrings().StartingImport)
|
|
self.setCancelButton(None)
|
|
self.setRange(0, 1)
|
|
self.setMinimumDuration(0)
|
|
self.was_cancelled = False
|
|
self.previous_size = 0
|
|
|
|
def update_progress(self, count, block_size):
|
|
"""
|
|
Calculate and display the download progress.
|
|
"""
|
|
increment = (count * block_size) - self.previous_size
|
|
self._increment_progress_bar(None, increment)
|
|
self.previous_size = count * block_size
|
|
|
|
def _increment_progress_bar(self, status_text, increment=1):
|
|
"""
|
|
Update the wizard progress page.
|
|
|
|
:param status_text: Current status information to display.
|
|
:param increment: The value to increment the progress bar by.
|
|
"""
|
|
if status_text:
|
|
self.setText(status_text)
|
|
if increment > 0:
|
|
self.setValue(self.value() + increment)
|
|
self.app.process_events()
|