class PropertyFileEditor(QDialog):
def __init__(self, subFormats, parent = None):
super().__init__(parent)
self._settings = SubSettings()
self.__createPropertyFilesDirectory()
self.__initSubFormats(subFormats)
self.__initGui()
def __createPropertyFilesDirectory(self):
pfileDir = self._settings.getPropertyFilesPath()
try:
os.makedirs(pfileDir)
except OSError as exc:
if exc.errno == errno.EEXIST and os.path.isdir(pfileDir):
pass
else: raise
def __initSubFormats(self, formats):
self._formats = {}
for f in formats:
self._formats[f.NAME] = f
def __initGui(self):
layout = QVBoxLayout()
layout.setSpacing(10)
layout.addWidget(self._createFpsBox())
layout.addWidget(self._createEncodingBox())
layout.addWidget(self._createFormatBox())
layout.addWidget(self._createButtons())
self.setLayout(layout)
self.setWindowTitle(_("Subtitle Properties Editor"))
self.setModal(True)
# Some signals
self._closeButton.clicked.connect(self.close)
self._openButton.clicked.connect(self.openProperties)
self._saveButton.clicked.connect(self.saveProperties)
self._autoEncoding.toggled.connect(self._inputEncoding.setDisabled)
self._changeEncoding.toggled.connect(self._outputEncoding.setEnabled)
def _createFpsBox(self):
groupbox = QGroupBox(_("FPS"))
layout = QHBoxLayout()
self._autoFps = QCheckBox(_("Auto FPS"), self)
self._fps = QComboBox(self)
self._fps.addItems(["23.976", "24", "25", "29.97", "30"])
self._fps.setEditable(True)
layout.addWidget(self._autoFps)
layout.addWidget(self._fps)
groupbox.setLayout(layout)
return groupbox
def _createEncodingBox(self):
groupbox = QGroupBox(_("File Encoding"))
layout = QGridLayout()
self._autoEncoding = QCheckBox(_("Auto input encoding"), self)
self._inputEncoding = QComboBox(self)
self._inputEncoding.addItems(ALL_ENCODINGS)
self._inputEncoding.setDisabled(self._autoEncoding.isChecked())
inputLabel = QLabel(_("Input encoding"))
self._changeEncoding = QCheckBox(_("Change encoding on save"), self)
self._outputEncoding = QComboBox(self)
self._outputEncoding.addItems(ALL_ENCODINGS)
self._outputEncoding.setEnabled(self._changeEncoding.isChecked())
outputLabel = QLabel(_("Output encoding"))
layout.addWidget(self._autoEncoding, 0, 0)
layout.addWidget(self._inputEncoding, 1, 0)
layout.addWidget(inputLabel, 1, 1)
layout.addWidget(self._changeEncoding, 2, 0)
layout.addWidget(self._outputEncoding, 3, 0)
layout.addWidget(outputLabel, 3, 1)
groupbox.setLayout(layout)
return groupbox
def _createFormatBox(self):
groupbox = QGroupBox(_("Subtitle format"))
layout = QGridLayout()
displayedFormats = list(self._formats.keys())
displayedFormats.sort()
self._outputFormat = QComboBox(self)
self._outputFormat.addItems(displayedFormats)
formatLabel = QLabel(_("Output format"))
layout.addWidget(self._outputFormat, 0, 0)
layout.addWidget(formatLabel, 0, 1)
groupbox.setLayout(layout)
return groupbox
def _createButtons(self):
widget = QWidget(self)
layout = QHBoxLayout()
self._openButton = QPushButton(_("Open"))
self._saveButton = QPushButton(_("Save"))
self._closeButton = QPushButton(_("Close"))
layout.addWidget(self._openButton)
layout.addWidget(self._saveButton)
layout.addWidget(self._closeButton)
widget.setLayout(layout)
return widget
def _createSubtitleProperties(self):
subProperties = SubtitleProperties(list(self._formats.values()))
subProperties.autoFps = self._autoFps.isChecked()
subProperties.fps = self._fps.currentText()
subProperties.autoInputEncoding = self._autoEncoding.isChecked()
subProperties.changeEncoding = self._changeEncoding.isChecked()
subProperties.inputEncoding = self._inputEncoding.currentText()
subProperties.outputEncoding = self._outputEncoding.currentText()
subProperties.outputFormat = self._formats.get(self._outputFormat.currentText())
return subProperties
def changeProperties(self, subProperties):
self._autoFps.setChecked(subProperties.autoFps)
self._fps.setEditText(str(subProperties.fps))
self._autoEncoding.setChecked(subProperties.autoInputEncoding)
self._changeEncoding.setChecked(subProperties.changeEncoding)
self._inputEncoding.setCurrentIndex(
self._inputEncoding.findText(subProperties.inputEncoding))
self._outputEncoding.setCurrentIndex(
self._outputEncoding.findText(subProperties.outputEncoding))
if self._formats.get(subProperties.outputFormat.NAME) is subProperties.outputFormat:
self._outputFormat.setCurrentIndex(
self._outputFormat.findText(subProperties.outputFormat.NAME))
else:
self.close()
raise RuntimeError(_("Subtitle format (%s) doesn't match any of known formats!") %
subProperties.outputFormat.NAME)
def saveProperties(self):
subProperties = None
try:
subProperties = self._createSubtitleProperties()
except Exception as e:
dialog = QMessageBox(self)
dialog.setIcon(QMessageBox.Critical)
dialog.setWindowTitle(_("Incorrect value"))
dialog.setText(_("Could not save SPF file because of incorrect parameters."));
dialog.setDetailedText(str(e));
dialog.exec()
return
fileDialog = FileDialog(
parent = self,
caption = _('Save Subtitle Properties'),
directory = self._settings.getPropertyFilesPath()
)
fileDialog.setAcceptMode(QFileDialog.AcceptSave)
fileDialog.setFileMode(QFileDialog.AnyFile)
if fileDialog.exec():
filename = fileDialog.selectedFiles()[0]
if not filename.endswith(".spf"):
filename = "%s%s" % (filename, ".spf")
self._settings.setPropertyFilesPath(os.path.dirname(filename))
subProperties.save(filename)
self._settings.addPropertyFile(filename)
self.close()
def openProperties(self):
fileDialog = FileDialog(
parent = self,
caption = _("Open Subtitle Properties"),
directory = self._settings.getPropertyFilesPath(),
filter = _("Subtitle Properties (*.spf);;All files (*)")
)
fileDialog.setFileMode(QFileDialog.ExistingFile)
if fileDialog.exec():
filename = fileDialog.selectedFiles()[0]
self._settings.setPropertyFilesPath(os.path.dirname(filename))
subProperties = SubtitleProperties(list(self._formats.values()), filename)
self.changeProperties(subProperties)
