Files
picard/picard/ui/scripteditor.py
2021-06-16 12:22:36 -06:00

1211 lines
49 KiB
Python

# -*- coding: utf-8 -*-
#
# Picard, the next-generation MusicBrainz tagger
#
# Copyright (C) 2021 Bob Swift
# Copyright (C) 2021 Laurent Monin
# Copyright (C) 2021 Philipp Wolfer
#
# 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 2
# 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, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
from functools import partial
import os.path
from PyQt5 import (
QtCore,
QtGui,
QtWidgets,
)
from PyQt5.QtGui import QPalette
from picard import log
from picard.config import (
BoolOption,
Option,
TextOption,
get_config,
)
from picard.const import (
DEFAULT_FILE_NAMING_FORMAT,
PICARD_URLS,
)
from picard.file import File
from picard.script import (
ScriptError,
ScriptParser,
get_file_naming_script,
get_file_naming_script_presets,
)
from picard.script.serializer import (
FileNamingScript,
ScriptImportError,
)
from picard.util import (
icontheme,
webbrowser2,
)
from picard.util.settingsoverride import SettingsOverride
from picard.ui import PicardDialog
from picard.ui.options import OptionsPage
from picard.ui.options.scripting import (
OptionsCheckError,
ScriptCheckError,
)
from picard.ui.ui_scripteditor import Ui_ScriptEditor
from picard.ui.ui_scripteditor_details import Ui_ScriptDetails
from picard.ui.widgets.scriptdocumentation import ScriptingDocumentationWidget
class ScriptFileError(OptionsCheckError):
pass
class ScriptEditorExamples():
"""File naming script examples.
"""
max_samples = 10 # pick up to 10 samples
def __init__(self, tagger):
"""File naming script examples.
Args:
tagger (object): The main window tagger object.
"""
self.tagger = tagger
self._sampled_example_files = []
config = get_config()
self.settings = config.setting
self.example_list = []
self.script_text = get_file_naming_script(self.settings)
def update_sample_example_files(self):
"""Get a new sample of randomly selected / loaded files to use as renaming examples.
"""
import random
if self.tagger.window.selected_objects:
# If files/albums/tracks are selected, sample example files from them
files = self.tagger.get_files_from_objects(self.tagger.window.selected_objects)
length = min(self.max_samples, len(files))
files = [file for file in random.sample(files, k=length)]
else:
# If files/albums/tracks are not selected, sample example files from the pool of loaded files
files = self.tagger.files
length = min(self.max_samples, len(files))
files = [files[key] for key in random.sample(files.keys(), k=length)]
if not files:
# If no file has been loaded, use generic examples
files = list(self.default_examples())
self._sampled_example_files = files
self.update_examples()
def update_examples(self, override=None, script_text=None):
"""Update the before and after file naming examples list.
Args:
override (dict, optional): Dictionary of settings overrides to apply. Defaults to None.
script_text (str, optional): Text of the file naming script to use. Defaults to None.
"""
if override and isinstance(override, dict):
self.settings = SettingsOverride(self.settings, override)
if script_text and isinstance(script_text, str):
self.script_text = script_text
if self.settings["move_files"] or self.settings["rename_files"]:
if not self._sampled_example_files:
self.update_sample_example_files()
self.example_list = [self._example_to_filename(example) for example in self._sampled_example_files]
else:
err_text = _("Renaming options are disabled")
self.example_list = [[err_text, err_text]]
def _example_to_filename(self, file):
"""Produce the before and after file naming example tuple for the specified file.
Args:
file (File): File to produce example before and after names
Returns:
tuple: Example before and after names for the specified file
"""
try:
if self.settings["enable_tagger_scripts"]:
for s_pos, s_name, s_enabled, s_text in self.settings["list_of_scripts"]:
if s_enabled and s_text:
parser = ScriptParser()
parser.eval(s_text, file.metadata)
filename_before = file.filename
filename_after = file.make_filename(filename_before, file.metadata, self.settings, self.script_text)
if not self.settings["move_files"]:
return os.path.basename(filename_before), os.path.basename(filename_after)
return filename_before, filename_after
except ScriptError:
return "", ""
except TypeError:
return "", ""
def update_example_listboxes(self, before_listbox, after_listbox):
"""Update the contents of the file naming examples before and after listboxes.
Args:
before_listbox (QListBox): The before listbox
after_listbox (QListBox): The after listbox
"""
before_listbox.clear()
after_listbox.clear()
for before, after in sorted(self.get_examples(), key=lambda x: x[1]):
before_listbox.addItem(before)
after_listbox.addItem(after)
def get_examples(self):
"""Get the list of examples.
Returns:
[list]: List of the before and after file name example tuples
"""
return self.example_list
@staticmethod
def synchronize_selected_example_lines(current_row, source, target):
"""Sets the current row in the target to match the current row in the source.
Args:
current_row (int): Currently selected row
source (QListView): Source list
target (QListView): Target list
"""
if source.currentRow() != current_row:
current_row = source.currentRow()
target.blockSignals(True)
target.setCurrentRow(current_row)
target.blockSignals(False)
@classmethod
def get_notes_text(cls):
"""Provides usage notes text suitable for display on the dialog.
Returns:
str: Notes text
"""
return _(
"If you select files from the Cluster pane or Album pane prior to opening the Options screen, "
"up to %u files will be randomly chosen from your selection as file naming examples. If you "
"have not selected any files, then some default examples will be provided.") % cls.max_samples
@classmethod
def get_tooltip_text(cls):
"""Provides tooltip text suitable for display on the dialog.
Returns:
str: Tooltip text
"""
return _("Reload up to %u items chosen at random from files selected in the main window") % cls.max_samples
@staticmethod
def default_examples():
"""Generator for default example files.
Yields:
File: the next example File object
"""
# example 1
efile = File("ticket_to_ride.mp3")
efile.state = File.NORMAL
efile.metadata.update({
'album': 'Help!',
'title': 'Ticket to Ride',
'~releasecomment': '2014 mono remaster',
'artist': 'The Beatles',
'artistsort': 'Beatles, The',
'albumartist': 'The Beatles',
'albumartistsort': 'Beatles, The',
'tracknumber': '7',
'totaltracks': '14',
'discnumber': '1',
'totaldiscs': '1',
'originaldate': '1965-08-06',
'originalyear': '1965',
'date': '2014-09-08',
'releasetype': ['album', 'soundtrack'],
'~primaryreleasetype': ['album'],
'~secondaryreleasetype': ['soundtrack'],
'releasestatus': 'official',
'releasecountry': 'US',
'~extension': 'mp3',
'musicbrainz_albumid': 'd7fbbb0a-1348-40ad-8eef-cd438d4cd203',
'musicbrainz_albumartistid': 'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d',
'musicbrainz_artistid': 'b10bbbfc-cf9e-42e0-be17-e2c3e1d2600d',
'musicbrainz_recordingid': 'ed052ae1-c950-47f2-8d2b-46e1b58ab76c',
'musicbrainz_releasetrackid': '392639f5-5629-477e-b04b-93bffa703405',
})
yield efile
# example 2
config = get_config()
efile = File("track05.mp3")
efile.state = File.NORMAL
efile.metadata.update({
'album': "Coup d'État, Volume 1: Ku De Ta / Prologue",
'title': "I've Got to Learn the Mambo",
'artist': "Snowboy feat. James Hunter",
'artistsort': "Snowboy feat. Hunter, James",
'albumartist': config.setting['va_name'],
'albumartistsort': config.setting['va_name'],
'tracknumber': '5',
'totaltracks': '13',
'discnumber': '2',
'totaldiscs': '2',
'discsubtitle': "Beat Up",
'originaldate': '2005-07-04',
'originalyear': '2005',
'date': '2005-07-04',
'releasetype': ['album', 'compilation'],
'~primaryreleasetype': 'album',
'~secondaryreleasetype': 'compilation',
'releasestatus': 'official',
'releasecountry': 'AU',
'compilation': '1',
'~multiartist': '1',
'~extension': 'mp3',
'musicbrainz_albumid': '4b50c71e-0a07-46ac-82e4-cb85dc0c9bdd',
'musicbrainz_recordingid': 'b3c487cb-0e55-477d-8df3-01ec6590f099',
'musicbrainz_releasetrackid': 'f8649a05-da39-39ba-957c-7abf8f9012be',
'musicbrainz_albumartistid': '89ad4ac3-39f7-470e-963a-56509c546377',
'musicbrainz_artistid': ['7b593455-d207-482c-8c6f-19ce22c94679', '9e082466-2390-40d1-891e-4803531f43fd'],
})
yield efile
def confirmation_dialog(parent, message):
"""Displays a confirmation dialog.
Args:
parent (object): Parent object / window making the call to set modality
message (str): Message to be displayed
Returns:
bool: True if accepted, otherwise False.
"""
dialog = QtWidgets.QMessageBox(
QtWidgets.QMessageBox.Warning,
_('Confirm'),
message,
QtWidgets.QMessageBox.Ok | QtWidgets.QMessageBox.Cancel,
parent
)
return dialog.exec_() == QtWidgets.QMessageBox.Ok
def synchronize_vertical_scrollbars(widgets):
"""Synchronize position of vertical scrollbars and selections for listed widgets.
Args:
widgets (list): List of QListView widgets to synchronize
"""
# Set highlight colors for selected list items
example_style = widgets[0].palette()
highlight_bg = example_style.color(QPalette.Active, QPalette.Highlight)
highlight_fg = example_style.color(QPalette.Active, QPalette.HighlightedText)
stylesheet = "QListView::item:selected { color: " + highlight_fg.name() + "; background-color: " + highlight_bg.name() + "; }"
def _sync_scrollbar_vert(widget, value):
widget.blockSignals(True)
widget.verticalScrollBar().setValue(value)
widget.blockSignals(False)
widgets = set(widgets)
for widget in widgets:
for other in widgets - {widget}:
widget.verticalScrollBar().valueChanged.connect(partial(_sync_scrollbar_vert, other))
widget.setStyleSheet(stylesheet)
def user_script_title(title):
"""Make a user script title for the combo box identifying the script as a user-defined script.
Args:
title (str): Base script title
Returns:
str: Base script title with "User:" identification added
"""
return _("User: %s") % title
def populate_script_selection_combo_box(naming_scripts, selected_script_id, combo_box):
"""Populate the specified script selection combo box and identify the selected script.
Args:
naming_scripts (dict): Dictionary of available user-defined naming scripts as script dictionaries as produced by FileNamingScript().to_dict()
selected_script_id (str): ID code for the currently selected script
combo_box (QComboBox): Combo box object to populate
Returns:
int: The index of the currently selected script
"""
combo_box.blockSignals(True)
combo_box.clear()
def _add_and_check(idx, count, title, item):
combo_box.addItem(title, item)
if item['id'] == selected_script_id:
idx = count
return idx
idx = 0
for count, (id, naming_script) in enumerate(sorted(naming_scripts.items(), key=lambda item: item[1]['title'])):
naming_script["deletable"] = True
naming_script["readonly"] = False
idx = _add_and_check(idx, count, user_script_title(naming_script['title']), naming_script)
# Add preset scripts not provided in the user-defined scripts list.
for count, script_item in enumerate(get_file_naming_script_presets(), start=count+1):
naming_script = script_item.to_dict()
naming_script["deletable"] = False
naming_script["readonly"] = True
idx = _add_and_check(idx, count, naming_script['title'], naming_script)
combo_box.setCurrentIndex(idx)
combo_box.blockSignals(False)
return idx
class ScriptEditorDialog(PicardDialog):
"""File Naming Script Editor Page
"""
TITLE = N_("File naming script editor")
STYLESHEET_ERROR = OptionsPage.STYLESHEET_ERROR
help_url = PICARD_URLS["doc_naming_script_edit"]
options = [
BoolOption('persist', 'script_editor_show_documentation', False),
Option("setting", "file_renaming_scripts", {}),
TextOption("setting", "selected_file_naming_script_id", ""),
]
signal_save = QtCore.pyqtSignal()
signal_update = QtCore.pyqtSignal()
signal_selection_changed = QtCore.pyqtSignal()
signal_update_scripts_list = QtCore.pyqtSignal()
signal_index_changed = QtCore.pyqtSignal()
default_script_directory = os.path.normpath(QtCore.QStandardPaths.writableLocation(QtCore.QStandardPaths.DocumentsLocation))
default_script_filename = "picard_naming_script.ptsp"
def __init__(self, parent=None, examples=None):
"""Stand-alone file naming script editor.
Args:
parent (QMainWindow or OptionsPage, optional): Parent object. Defaults to None.
examples (ScriptEditorExamples, required): Object containing examples to display. Defaults to None.
"""
super().__init__(parent)
self.examples = examples
self.FILE_TYPE_ALL = _("All Files") + " (*)"
self.FILE_TYPE_SCRIPT = _("Picard Script Files") + " (*.pts *.txt)"
self.FILE_TYPE_PACKAGE = _("Picard Naming Script Package") + " (*.ptsp *.yaml)"
self.setWindowTitle(self.TITLE)
self.displaying = False
self.loading = True
self.ui = Ui_ScriptEditor()
self.ui.setupUi(self)
self.make_menu()
self.ui.label.setWordWrap(False)
self.installEventFilter(self)
# Dialog buttons
self.reset_button = QtWidgets.QPushButton(_('Revert'))
self.reset_button.setToolTip(self.reset_action.toolTip())
self.reset_button.clicked.connect(self.reset_script)
self.ui.buttonbox.addButton(self.reset_button, QtWidgets.QDialogButtonBox.ActionRole)
self.save_button = self.ui.buttonbox.addButton(QtWidgets.QDialogButtonBox.Save)
self.save_button.setToolTip(self.save_action.toolTip())
self.ui.buttonbox.accepted.connect(self.save_script)
self.close_button = self.ui.buttonbox.addButton(QtWidgets.QDialogButtonBox.Close)
self.close_button.setToolTip(self.close_action.toolTip())
self.ui.buttonbox.rejected.connect(self.close)
self.ui.buttonbox.addButton(QtWidgets.QDialogButtonBox.Help)
self.ui.buttonbox.helpRequested.connect(self.show_help)
self.ui.file_naming_format.setEnabled(True)
# Add scripting documentation to parent frame.
doc_widget = ScriptingDocumentationWidget(self, include_link=False)
self.ui.documentation_frame_layout.addWidget(doc_widget)
self.ui.file_naming_format.textChanged.connect(self.check_formats)
self._sampled_example_files = []
self.ui.example_filename_after.itemSelectionChanged.connect(self.match_before_to_after)
self.ui.example_filename_before.itemSelectionChanged.connect(self.match_after_to_before)
self.ui.preset_naming_scripts.currentIndexChanged.connect(partial(self.select_script, skip_check=False))
synchronize_vertical_scrollbars((self.ui.example_filename_before, self.ui.example_filename_after))
self.toggle_documentation() # Force update to display
self.examples_current_row = -1
self.script_metadata_changed = False
self.selected_script_index = 0
self.restore_selected_script_index = 0
self.current_item_dict = None
self.load()
self.loading = False
def make_menu(self):
"""Build the menu bar.
"""
config = get_config()
main_menu = QtWidgets.QMenuBar()
# File menu settings
file_menu = main_menu.addMenu(_('&File'))
file_menu.setToolTipsVisible(True)
self.import_action = QtWidgets.QAction(_("&Import a script file"), self)
self.import_action.setToolTip(_("Import a file as a new script"))
self.import_action.setIcon(icontheme.lookup('document-open'))
self.import_action.triggered.connect(self.import_script)
file_menu.addAction(self.import_action)
self.export_action = QtWidgets.QAction(_("&Export a script file"), self)
self.export_action.setToolTip(_("Export the script to a file"))
self.export_action.setIcon(icontheme.lookup('document-save'))
self.export_action.triggered.connect(self.export_script)
file_menu.addAction(self.export_action)
self.close_action = QtWidgets.QAction(_("E&xit / Close editor"), self)
self.close_action.setToolTip(_("Close the script editor"))
self.close_action.triggered.connect(self.close)
file_menu.addAction(self.close_action)
# Script menu settings
script_menu = main_menu.addMenu(_('&Script'))
script_menu.setToolTipsVisible(True)
self.details_action = QtWidgets.QAction(_("Edit Script &Metadata"), self)
self.details_action.setToolTip(_("Display the details for the script"))
self.details_action.triggered.connect(self.view_script_details)
self.details_action.setShortcut(QtGui.QKeySequence(_("Ctrl+M")))
script_menu.addAction(self.details_action)
self.add_action = QtWidgets.QAction(_("Add a &new script"), self)
self.add_action.setToolTip(_("Create a new file naming script"))
self.add_action.setIcon(icontheme.lookup('add-item'))
self.add_action.triggered.connect(self.new_script)
script_menu.addAction(self.add_action)
self.copy_action = QtWidgets.QAction(_("&Copy the current script"), self)
self.copy_action.setToolTip(_("Save a copy of the script as a new script"))
self.copy_action.setIcon(icontheme.lookup('edit-copy'))
self.copy_action.triggered.connect(self.copy_script)
script_menu.addAction(self.copy_action)
self.delete_action = QtWidgets.QAction(_("&Delete the current script"), self)
self.delete_action.setToolTip(_("Delete the script"))
self.delete_action.setIcon(icontheme.lookup('list-remove'))
self.delete_action.triggered.connect(self.delete_script)
script_menu.addAction(self.delete_action)
self.reset_action = QtWidgets.QAction(_("&Revert the current script"), self)
self.reset_action.setToolTip(_("Revert the script to the last saved value"))
self.reset_action.setIcon(icontheme.lookup('view-refresh'))
self.reset_action.triggered.connect(self.reset_script)
script_menu.addAction(self.reset_action)
self.save_action = QtWidgets.QAction(_("&Save the current script"), self)
self.save_action.setToolTip(_("Save changes to the script"))
self.save_action.setIcon(icontheme.lookup('document-save'))
self.save_action.setShortcut(QtGui.QKeySequence(_("Ctrl+S")))
self.save_action.triggered.connect(self.save_script)
script_menu.addAction(self.save_action)
# Display menu settings
display_menu = main_menu.addMenu(_('&View'))
display_menu.setToolTipsVisible(True)
self.examples_action = QtWidgets.QAction(_("&Reload random example files"), self)
self.examples_action.setToolTip(self.examples.get_tooltip_text())
self.examples_action.setIcon(icontheme.lookup('view-refresh'))
self.examples_action.triggered.connect(self.update_example_files)
display_menu.addAction(self.examples_action)
display_menu.addAction(self.ui.file_naming_format.wordwrap_action)
display_menu.addAction(self.ui.file_naming_format.show_tooltips_action)
self.docs_action = QtWidgets.QAction(_("&Show documentation"), self)
self.docs_action.setToolTip(_("View the scripting documentation in a sidebar"))
self.docs_action.triggered.connect(self.toggle_documentation)
self.docs_action.setShortcut(QtGui.QKeySequence(_("Ctrl+H")))
self.docs_action.setCheckable(True)
self.docs_action.setChecked(config.persist["script_editor_show_documentation"])
display_menu.addAction(self.docs_action)
# Help menu settings
help_menu = main_menu.addMenu(_('&Help'))
help_menu.setToolTipsVisible(True)
self.help_action = QtWidgets.QAction(_("&Help..."), self)
self.help_action.setShortcut(QtGui.QKeySequence.HelpContents)
self.help_action.triggered.connect(self.show_help)
help_menu.addAction(self.help_action)
self.scripting_docs_action = QtWidgets.QAction(_("&Scripting documentation..."), self)
self.scripting_docs_action.setToolTip(_("Open the scripting documentation in your browser"))
self.scripting_docs_action.triggered.connect(self.docs_browser)
help_menu.addAction(self.scripting_docs_action)
self.ui.layout_for_menubar.addWidget(main_menu)
def load(self):
"""Load initial configuration.
"""
config = get_config()
self.examples.settings = config.setting
self.naming_scripts = config.setting["file_renaming_scripts"]
self.selected_script_id = config.setting["selected_file_naming_script_id"]
self.selected_script_index = 0
self.restore_selected_script_index = 0
self.populate_script_selector()
if not self.loading:
self.select_script(skip_check=True)
def docs_browser(self):
"""Open the scriping documentation in a browser.
"""
webbrowser2.open('doc_scripting')
def eventFilter(self, object, event):
"""Process selected events.
"""
evtype = event.type()
if evtype in {QtCore.QEvent.WindowActivate, QtCore.QEvent.FocusIn}:
self.update_examples()
return False
def closeEvent(self, event):
"""Custom close event handler to check for unsaved changes.
"""
if self.unsaved_changes_confirmation():
if self.has_changed():
self.select_script(skip_check=True)
super().closeEvent(event)
else:
event.ignore()
def populate_script_selector(self):
"""Populate the script selection combo box.
"""
idx = populate_script_selection_combo_box(self.naming_scripts, self.selected_script_id, self.ui.preset_naming_scripts)
self.update_scripts_list()
self.set_selected_script_index(idx)
self.restore_selected_script_index = idx
def toggle_documentation(self):
"""Toggle the display of the scripting documentation sidebar.
"""
checked = self.docs_action.isChecked()
config = get_config()
config.persist["script_editor_show_documentation"] = checked
self.ui.documentation_frame.setVisible(checked)
def view_script_details(self):
"""View and edit (if not readonly) the metadata associated with the script.
"""
self.current_item_dict = self.get_selected_item()
details_page = ScriptDetailsEditor(self, self.current_item_dict)
details_page.signal_save.connect(self.update_from_details)
details_page.show()
details_page.raise_()
details_page.activateWindow()
def has_changed(self):
"""Check if the current script has pending edits to the title, script or metadata that have not been saved.
Returns:
bool: True if there are unsaved changes, otherwise false.
"""
script_item = self.ui.preset_naming_scripts.itemData(self.restore_selected_script_index)
return self.ui.script_title.text().strip() != script_item['title'] or \
self.get_script() != script_item['script'] or \
self.script_metadata_changed
def update_from_details(self):
"""Update the script selection combo box and script list after updates from the script details dialog.
"""
self.update_combo_box_item(self.ui.preset_naming_scripts.currentIndex(), self.current_item_dict)
self.ui.script_title.setText(self.current_item_dict['title'])
self.script_metadata_changed = True
def _set_combobox_index(self, idx):
"""Sets the index of the script selector combo box.
Args:
idx (int): New index position
"""
self.restore_selected_script_index = self.selected_script_index
self.ui.preset_naming_scripts.blockSignals(True)
self.ui.preset_naming_scripts.setCurrentIndex(idx)
self.ui.preset_naming_scripts.blockSignals(False)
self.selected_script_index = idx
self.signal_index_changed.emit()
def _insert_item(self, script_item):
"""Insert a new item into the script selection combo box and update the script list in the settings.
Args:
script_item (dict): File naming script to insert as produced by FileNamingScript().to_dict()
"""
script_item["readonly"] = False
script_item["deletable"] = True
self.selected_script_id = script_item["id"]
self.naming_scripts[self.selected_script_id] = script_item
idx = populate_script_selection_combo_box(self.naming_scripts, self.selected_script_id, self.ui.preset_naming_scripts)
self._set_combobox_index(idx)
self.update_scripts_list()
self.select_script(skip_check=True)
self.restore_selected_script_index = idx
def new_script(self):
"""Add a new (empty) script to the script selection combo box and script list.
"""
if self.unsaved_changes_confirmation():
script_item = FileNamingScript(script=DEFAULT_FILE_NAMING_FORMAT).to_dict()
self._insert_item(script_item)
def copy_script(self):
"""Add a copy of the script as a new editable script to the script selection combo box and script list.
"""
if self.unsaved_changes_confirmation():
selected = self.ui.preset_naming_scripts.currentIndex()
script_item = self.ui.preset_naming_scripts.itemData(selected)
new_item = FileNamingScript.create_from_dict(script_dict=script_item).copy().to_dict()
self._insert_item(new_item)
def update_script_in_settings(self):
"""Sends a save signal to trigger processing in the parent.
"""
if not self.loading:
self.signal_save.emit()
def update_scripts_list(self):
"""Refresh the script list in the settings based on the contents of the script selection combo box.
"""
self.naming_scripts = {}
for idx in range(self.ui.preset_naming_scripts.count()):
script_item = self.ui.preset_naming_scripts.itemData(idx)
# Only add items that can be removed -- no presets
if script_item["deletable"]:
self.naming_scripts[script_item["id"]] = script_item
self.signal_update_scripts_list.emit()
def get_selected_item(self, idx=None):
"""Get the specified item from the script selection combo box.
Args:
idx (int, optional): Index of the combo box item to retrieve. Defaults to None.
Returns:
dict: File naming script dictionary as produced by FileNamingScript().to_dict()
"""
if idx is None:
idx = self.ui.preset_naming_scripts.currentIndex()
return self.ui.preset_naming_scripts.itemData(idx)
def unsaved_changes_confirmation(self):
"""Check if there are unsaved changes and ask the user to confirm the action resulting in their loss.
Returns:
bool: True if no unsaved changes or user confirms the action, otherwise False.
"""
if not self.loading and self.has_changed() and not confirmation_dialog(self,
_("There are unsaved changes to the current script. Do you want to continue and lose these changes?")
):
self.selected_script_index = self.restore_selected_script_index
self._set_combobox_index(self.restore_selected_script_index)
return False
return True
def set_selected_script_id(self, id, skip_check=True):
"""Select the script with the specified ID.
Args:
id (str): ID of the script to select
skip_check (bool, optional): Skip the check for unsaved edits. Defaults to True.
"""
idx = 0
for i in range(self.ui.preset_naming_scripts.count()):
script_item = self.ui.preset_naming_scripts.itemData(i)
if script_item["id"] == id:
idx = i
break
self.set_selected_script_index(idx, skip_check=skip_check)
def set_selected_script_index(self, idx, skip_check=True):
"""Select the script at the specified combo box index.
Args:
idx (int): Index of the script to select
skip_check (bool, optional): Skip the check for unsaved edits. Defaults to True.
"""
self._set_combobox_index(idx)
self.select_script(skip_check=skip_check)
def select_script(self, skip_check=False):
"""Load the current script from the combo box into the editor.
Args:
skip_check (bool): Skip the check for unsaved edits. Defaults to False.
"""
self.selected_script_index = self.ui.preset_naming_scripts.currentIndex()
if self.loading or skip_check or self.unsaved_changes_confirmation():
script_item = self.get_selected_item()
self.ui.script_title.setText(script_item['title'])
self.set_script(script_item['script'])
self.selected_script_id = script_item['id']
self.restore_selected_script_index = self.selected_script_index
self.script_metadata_changed = False
if not self.loading:
self.update_script_in_settings()
self.set_button_states()
self.update_examples()
if not self.loading:
self.signal_selection_changed.emit()
def update_combo_box_item(self, idx, script_item):
"""Update the title and item data for the specified script selection combo box item.
Args:
idx (int): Index of the item to update
script_item (dict): Updated file naming script information as produced by FileNamingScript().to_dict()
"""
self.ui.preset_naming_scripts.setItemData(idx, script_item)
self.ui.preset_naming_scripts.setItemText(idx, user_script_title(script_item['title']))
self.update_scripts_list()
self.update_script_in_settings()
def set_button_states(self, save_enabled=True):
"""Set the button states based on the readonly and deletable attributes of the currently selected
item in the script selection combo box.
Args:
save_enabled (bool, optional): Allow updates to be saved to this item. Defaults to True.
"""
selected = self.ui.preset_naming_scripts.currentIndex()
if selected < 0:
return
script_item = self.get_selected_item()
readonly = script_item['readonly']
self.ui.script_title.setReadOnly(readonly or selected < 1)
# Buttons
self.ui.file_naming_format.setReadOnly(readonly)
self.save_button.setEnabled(save_enabled and not readonly)
self.reset_button.setEnabled(not readonly)
# Menu items
self.save_action.setEnabled(save_enabled and not readonly)
self.reset_action.setEnabled(not readonly)
self.add_action.setEnabled(save_enabled)
self.copy_action.setEnabled(save_enabled)
self.delete_action.setEnabled(script_item['deletable'] and save_enabled)
self.import_action.setEnabled(save_enabled)
self.export_action.setEnabled(save_enabled)
def match_after_to_before(self):
"""Sets the selected item in the 'after' list to the corresponding item in the 'before' list.
"""
self.examples.synchronize_selected_example_lines(self.examples_current_row, self.ui.example_filename_before, self.ui.example_filename_after)
def match_before_to_after(self):
"""Sets the selected item in the 'before' list to the corresponding item in the 'after' list.
"""
self.examples.synchronize_selected_example_lines(self.examples_current_row, self.ui.example_filename_after, self.ui.example_filename_before)
def delete_script(self):
"""Removes the currently selected script from the script selection combo box and script list.
"""
if confirmation_dialog(self, _('Are you sure that you want to delete the script?')):
idx = self.ui.preset_naming_scripts.currentIndex()
self.ui.preset_naming_scripts.blockSignals(True)
self.ui.preset_naming_scripts.removeItem(idx)
self.ui.preset_naming_scripts.blockSignals(False)
if idx >= self.ui.preset_naming_scripts.count():
idx = self.ui.preset_naming_scripts.count() - 1
self._set_combobox_index(idx)
self.selected_script_index = idx
self.update_scripts_list()
self.select_script(skip_check=True)
def save_script(self):
"""Saves changes to the current script to the script list and combo box item.
"""
selected = self.ui.preset_naming_scripts.currentIndex()
title = str(self.ui.script_title.text()).strip()
if title:
script_item = self.ui.preset_naming_scripts.itemData(selected)
script_item["title"] = title
script_item["script"] = self.get_script()
self.update_combo_box_item(selected, script_item)
dialog = QtWidgets.QMessageBox(
QtWidgets.QMessageBox.Information,
_("Save Script"),
_("Changes to the script have been saved."),
QtWidgets.QMessageBox.Ok,
self
)
dialog.exec_()
else:
self.display_error(OptionsCheckError(_("Error"), _("The script title must not be empty.")))
def get_script(self):
"""Provides the text of the file naming script currently loaded into the editor.
Returns:
str: File naming script
"""
return str(self.ui.file_naming_format.toPlainText()).strip()
def set_script(self, script_text):
"""Sets the text of the file naming script into the editor.
Args:
script_text (str): File naming script text to set in the editor.
"""
self.ui.file_naming_format.setPlainText(str(script_text).strip())
def update_example_files(self):
"""Update the before and after file naming examples list.
"""
self.examples.update_sample_example_files()
self.display_examples()
def update_examples(self):
"""Update the before and after file naming examples using the current file naming script in the editor.
"""
self.examples.update_examples(script_text=self.get_script())
self.display_examples()
def display_examples(self):
"""Update the display of the before and after file naming examples.
Args:
send_signal (bool, optional): Determines whether the update signal should be emitted. Defaults to True.
"""
self.examples_current_row = -1
self.examples.update_example_listboxes(self.ui.example_filename_before, self.ui.example_filename_after)
if not self.loading:
self.signal_update.emit()
def output_error(self, title, fmt, filename, msg):
"""Log error and display error message dialog.
Args:
title (str): Title to display on the error dialog box
fmt (str): Format for the error type being displayed
filename (str): Name of the file being imported or exported
msg (str): Error message to display
"""
log.error(fmt, filename, msg)
error_message = _(fmt) % (filename, _(msg))
self.display_error(ScriptFileError(_(title), error_message))
def output_file_error(self, fmt, filename, msg):
"""Log file error and display error message dialog.
Args:
fmt (str): Format for the error type being displayed
filename (str): Name of the file being imported or exported
msg (str): Error message to display
"""
self.output_error(_("File Error"), fmt, filename, msg)
def import_script(self):
"""Import from an external text file to a new script. Import can be either a plain text script or
a naming script package.
"""
FILE_ERROR_IMPORT = N_('Error importing "%s". %s.')
FILE_ERROR_DECODE = N_('Error decoding "%s". %s.')
if not self.unsaved_changes_confirmation():
return
dialog_title = _("Import Script File")
dialog_file_types = self._get_dialog_filetypes()
options = QtWidgets.QFileDialog.Options()
options |= QtWidgets.QFileDialog.DontUseNativeDialog
filename, file_type = QtWidgets.QFileDialog.getOpenFileName(self, dialog_title, self.default_script_directory, dialog_file_types, options=options)
if filename:
log.debug('Importing naming script file: %s' % filename)
try:
with open(filename, 'r', encoding='utf8') as i_file:
file_content = i_file.read()
except OSError as error:
self.output_file_error(FILE_ERROR_IMPORT, filename, error.strerror)
return
if not file_content.strip():
self.output_file_error(FILE_ERROR_IMPORT, filename, _('The file was empty'))
return
if file_type == self.FILE_TYPE_PACKAGE:
try:
script_item = FileNamingScript().create_from_yaml(file_content)
except ScriptImportError as error:
self.output_file_error(FILE_ERROR_DECODE, filename, error)
return
else:
script_item = FileNamingScript(
title=_("Imported from %s") % filename,
script=file_content.strip()
)
self._insert_item(script_item.to_dict())
def export_script(self):
"""Export the current script to an external file. Export can be either as a plain text
script or a naming script package.
"""
FILE_ERROR_EXPORT = N_('Error exporting file "%s". %s.')
script_item = FileNamingScript.create_from_dict(script_dict=self.get_selected_item(), create_new_id=False)
script_text = self.get_script()
if script_text:
default_path = os.path.normpath(os.path.join(self.default_script_directory, self.default_script_filename))
dialog_title = _("Export Script File")
dialog_file_types = self._get_dialog_filetypes()
options = QtWidgets.QFileDialog.Options()
options |= QtWidgets.QFileDialog.DontUseNativeDialog
filename, file_type = QtWidgets.QFileDialog.getSaveFileName(self, dialog_title, default_path, dialog_file_types, options=options)
if filename:
# Fix issue where Qt may set the extension twice
(name, ext) = os.path.splitext(filename)
if ext and str(name).endswith('.' + ext):
filename = name
log.debug('Exporting naming script file: %s' % filename)
if file_type == self.FILE_TYPE_PACKAGE:
script_text = script_item.to_yaml()
try:
with open(filename, 'w', encoding='utf8') as o_file:
o_file.write(script_text)
except OSError as error:
self.output_file_error(FILE_ERROR_EXPORT, filename, error.strerror)
else:
dialog = QtWidgets.QMessageBox(
QtWidgets.QMessageBox.Information,
_("Export Script"),
_("Script successfully exported to %s") % filename,
QtWidgets.QMessageBox.Ok,
self
)
dialog.exec_()
def _get_dialog_filetypes(self):
"""Helper function to build file type string used in the file dialogs.
Returns:
str: File type selection string
"""
return ";;".join((
self.FILE_TYPE_PACKAGE,
self.FILE_TYPE_SCRIPT,
self.FILE_TYPE_ALL,
))
def reset_script(self):
"""Reset the script to the last saved value.
"""
if self.has_changed():
if confirmation_dialog(self, _("Are you sure that you want to reset the script to its last saved value?")):
self.select_script(skip_check=True)
else:
dialog = QtWidgets.QMessageBox(
QtWidgets.QMessageBox.Information,
_("Revert Script"),
_("There have been no changes made since the last time the script was saved."),
QtWidgets.QMessageBox.Ok,
self
)
dialog.exec_()
def check_formats(self):
"""Checks for valid file naming script and settings, and updates the examples.
"""
self.test()
self.update_examples()
def check_format(self):
"""Parse the file naming script and check for errors.
"""
config = get_config()
parser = ScriptParser()
script_text = self.get_script()
try:
parser.eval(script_text)
except Exception as e:
raise ScriptCheckError("", str(e))
if config.setting["rename_files"]:
if not self.get_script():
raise ScriptCheckError("", _("The file naming format must not be empty."))
def display_error(self, error):
"""Display an error message for the specified error.
Args:
error (Exception): The exception to display.
"""
# Ignore scripting errors, those are handled inline
if not isinstance(error, ScriptCheckError):
dialog = QtWidgets.QMessageBox(QtWidgets.QMessageBox.Warning, error.title, error.info, QtWidgets.QMessageBox.Ok, self)
dialog.exec_()
def test(self):
"""Parse the script and display any errors.
"""
self.ui.renaming_error.setStyleSheet("")
self.ui.renaming_error.setText("")
save_enabled = True
try:
self.check_format()
except ScriptCheckError as e:
self.ui.renaming_error.setStyleSheet(self.STYLESHEET_ERROR)
self.ui.renaming_error.setText(e.info)
save_enabled = False
self.set_button_states(save_enabled=save_enabled)
class ScriptDetailsEditor(PicardDialog):
"""View / edit the metadata details for a script.
"""
NAME = 'scriptdetails'
TITLE = N_('Script Details')
signal_save = QtCore.pyqtSignal()
def __init__(self, parent, script_item):
"""Script metadata viewer / editor.
Args:
parent (ScriptEditorDialog): The page used for editing the scripts
script_item (dict): The script whose metadata is displayed
"""
super().__init__(parent=parent)
self.script_item = script_item
self.readonly = script_item["readonly"]
self.setWindowTitle(self.TITLE)
self.displaying = False
self.ui = Ui_ScriptDetails()
self.ui.setupUi(self)
self.ui.buttonBox.accepted.connect(self.save_changes)
self.ui.buttonBox.rejected.connect(self.close_window)
self.ui.last_updated_now.clicked.connect(self.set_last_updated)
self.ui.script_title.setText(self.script_item['title'])
self.ui.script_author.setText(self.script_item['author'])
self.ui.script_version.setText(self.script_item['version'])
self.ui.script_last_updated.setText(self.script_item['last_updated'])
self.ui.script_license.setText(self.script_item['license'])
self.ui.script_description.setPlainText(self.script_item['description'])
self.ui.script_last_updated.setReadOnly(self.readonly)
self.ui.script_author.setReadOnly(self.readonly)
self.ui.script_version.setReadOnly(self.readonly)
self.ui.script_license.setReadOnly(self.readonly)
self.ui.script_description.setReadOnly(self.readonly)
self.ui.buttonBox.button(QtWidgets.QDialogButtonBox.Close).setVisible(self.readonly)
self.ui.buttonBox.button(QtWidgets.QDialogButtonBox.Cancel).setVisible(not self.readonly)
self.ui.buttonBox.button(QtWidgets.QDialogButtonBox.Save).setVisible(not self.readonly)
self.ui.buttonBox.setFocus()
self.setModal(True)
self.skip_change_check = False
def has_changed(self):
"""Check if the current script metadata has pending edits that have not been saved.
Returns:
bool: True if there are unsaved changes, otherwise false.
"""
return self.script_item['title'] != self.ui.script_title.text().strip() or \
self.script_item['author'] != self.ui.script_author.text().strip() or \
self.script_item['version'] != self.ui.script_version.text().strip() or \
self.script_item['license'] != self.ui.script_license.text().strip() or \
self.script_item['description'] != self.ui.script_description.toPlainText().strip()
def change_check(self):
"""Confirm whether the unsaved changes should be lost.
Returns:
bool: True if changes can be lost, otherwise False.
"""
return confirmation_dialog(
self,
_("There are unsaved changes to the current metadata. Do you want to continue and lose these changes?"),
)
def set_last_updated(self):
"""Set the last updated value to the current timestamp.
"""
self.ui.script_last_updated.setText(FileNamingScript.make_last_updated())
self.ui.script_last_updated.setModified(True)
def save_changes(self):
"""Update the script object with any changes to the metadata.
"""
title = self.ui.script_title.text().strip()
if not title:
QtWidgets.QMessageBox(
QtWidgets.QMessageBox.Critical,
_("Error"),
_("The script title must not be empty."),
QtWidgets.QMessageBox.Ok,
self
).exec_()
return
if self.has_changed():
if not self.ui.script_last_updated.isModified() or not self.ui.script_last_updated.text().strip():
self.set_last_updated()
self.script_item["title"] = self.ui.script_title.text().strip()
self.script_item["author"] = self.ui.script_author.text().strip()
self.script_item["version"] = self.ui.script_version.text().strip()
self.script_item["license"] = self.ui.script_license.text().strip()
self.script_item["description"] = self.ui.script_description.toPlainText().strip()
self.script_item["last_updated"] = self.ui.script_last_updated.text().strip()
self.signal_save.emit()
self.skip_change_check = True
self.close_window()
def close_window(self):
"""Close the script metadata editor window.
"""
self.close()
def closeEvent(self, event):
"""Custom close event handler to check for unsaved changes.
"""
if self.skip_change_check or not self.has_changed() or (self.has_changed() and self.change_check()):
event.accept()
else:
event.ignore()