diff options
Diffstat (limited to 'src/gui/presetmanager.py')
| -rw-r--r-- | src/gui/presetmanager.py | 349 |
1 files changed, 0 insertions, 349 deletions
diff --git a/src/gui/presetmanager.py b/src/gui/presetmanager.py deleted file mode 100644 index 980a969..0000000 --- a/src/gui/presetmanager.py +++ /dev/null @@ -1,349 +0,0 @@ -""" -Preset manager object handles all interactions with presets, including -the context menu accessed from MainWindow. -""" - -from PyQt6 import QtCore, QtWidgets, uic -import string -import os -import logging - -from ..toolkit import badName -from ..core import Core -from .actions import * - - -log = logging.getLogger("AVP.Gui.PresetManager") - - -class PresetManager(QtWidgets.QDialog): - def __init__(self, parent): - super().__init__() - uic.loadUi(os.path.join(Core.wd, "gui", "presetmanager.ui"), self) - self.parent = parent - self.core = parent.core - self.settings = parent.settings - self.presetDir = parent.presetDir - if not self.settings.value("presetDir"): - self.settings.setValue( - "presetDir", os.path.join(parent.dataDir, "projects") - ) - - self.findPresets() - - # window - self.lastFilter = "*" - self.presetRows = [] # list of (comp, vers, name) tuples - - self.setWindowFlags(QtCore.Qt.WindowType.WindowStaysOnTopHint) - - # connect button signals - self.pushButton_delete.clicked.connect(self.openDeletePresetDialog) - self.pushButton_rename.clicked.connect(self.openRenamePresetDialog) - self.pushButton_import.clicked.connect(self.openImportDialog) - self.pushButton_export.clicked.connect(self.openExportDialog) - self.pushButton_close.clicked.connect(self.close) - - # create filter box and preset list - self.drawFilterList() - self.comboBox_filter.currentIndexChanged.connect( - lambda: self.drawPresetList( - self.comboBox_filter.currentText(), self.lineEdit_search.text() - ) - ) - - # make auto-completion for search bar - self.autocomplete = QtCore.QStringListModel() - completer = QtWidgets.QCompleter() - completer.setModel(self.autocomplete) - self.lineEdit_search.setCompleter(completer) - self.lineEdit_search.textChanged.connect( - lambda: self.drawPresetList( - self.comboBox_filter.currentText(), self.lineEdit_search.text() - ) - ) - self.drawPresetList("*") - - def show_(self): - """Open a new preset manager window from the mainwindow""" - self.findPresets() - self.drawFilterList() - self.drawPresetList("*") - self.show() - - def findPresets(self): - log.debug("Searching %s for presets", self.presetDir) - parseList = [] - for dirpath, dirnames, filenames in os.walk(self.presetDir): - # anything without a subdirectory must be a preset folder - if dirnames: - continue - for preset in filenames: - compName = os.path.basename(os.path.dirname(dirpath)) - if compName not in self.core.compNames: - continue - compVers = os.path.basename(dirpath) - try: - parseList.append((compName, int(compVers), preset)) - except ValueError: - continue - self.presets = { - compName: [ - (vers, preset) for name, vers, preset in parseList if name == compName - ] - for compName, _, __ in parseList - } - - def drawPresetList(self, compFilter=None, presetFilter=""): - self.listWidget_presets.clear() - if compFilter: - self.lastFilter = str(compFilter) - else: - compFilter = str(self.lastFilter) - self.presetRows = [] - presetNames = [] - for component, presets in self.presets.items(): - if compFilter != "*" and component != compFilter: - continue - for vers, preset in presets: - if not presetFilter or presetFilter in preset: - self.listWidget_presets.addItem("%s: %s" % (component, preset)) - self.presetRows.append((component, vers, preset)) - if preset not in presetNames: - presetNames.append(preset) - self.autocomplete.setStringList(presetNames) - - def drawFilterList(self): - self.comboBox_filter.clear() - self.comboBox_filter.addItem("*") - for component in self.presets: - self.comboBox_filter.addItem(component) - - def clearPreset(self, compI=None): - """Functions on mainwindow level from the context menu""" - compI = self.parent.listWidget_componentList.currentRow() - action = ClearPreset(self.parent, compI) - self.parent.undoStack.push(action) - - def openSavePresetDialog(self): - """Functions on mainwindow level from the context menu""" - selectedComponents = self.core.selectedComponents - componentList = self.parent.listWidget_componentList - - if componentList.currentRow() == -1: - return - while True: - index = componentList.currentRow() - currentPreset = selectedComponents[index].currentPreset - newName, OK = QtWidgets.QInputDialog.getText( - self.parent, - "Audio Visualizer", - "New Preset Name:", - QtWidgets.QLineEdit.EchoMode.Normal, - currentPreset, - ) - if OK: - if badName(newName): - self.warnMessage(self.parent) - continue - if newName: - if index != -1: - selectedComponents[index].currentPreset = newName - saveValueStore = selectedComponents[index].savePreset() - saveValueStore["preset"] = newName - componentName = str(selectedComponents[index]).strip() - vers = selectedComponents[index].version - self.createNewPreset( - componentName, - vers, - newName, - saveValueStore, - window=self.parent, - ) - self.findPresets() - self.drawPresetList() - self.openPreset(newName, index) - break - - def createNewPreset(self, compName, vers, filename, saveValueStore, **kwargs): - path = os.path.join(self.presetDir, compName, str(vers), filename) - if self.presetExists(path, **kwargs): - return - self.core.createPresetFile(compName, vers, filename, saveValueStore) - - def presetExists(self, path, **kwargs): - if os.path.exists(path): - window = kwargs.get("window", self) - ch = self.parent.showMessage( - msg="%s already exists! Overwrite it?" % os.path.basename(path), - showCancel=True, - icon="Warning", - parent=window, - ) - if not ch: - # user clicked cancel - return True - - return False - - def openPreset(self, presetName, compPos=None): - componentList = self.parent.listWidget_componentList - index = compPos if compPos is not None else componentList.currentRow() - if index == -1: - return - action = OpenPreset(self, presetName, index) - self.parent.undoStack.push(action) - - def _openPreset(self, presetName, index): - selectedComponents = self.core.selectedComponents - - componentName = selectedComponents[index].name.strip() - version = selectedComponents[index].version - dirname = os.path.join(self.presetDir, componentName, str(version)) - filepath = os.path.join(dirname, presetName) - self.core.openPreset(filepath, index, presetName) - - self.parent.updateComponentTitle(index) - self.parent.drawPreview() - - def openDeletePresetDialog(self): - row = self.getPresetRow() - if row == -1: - return - comp, vers, name = self.presetRows[row] - ch = self.parent.showMessage( - msg="Really delete %s?" % name, - showCancel=True, - icon="Warning", - parent=self, - ) - if not ch: - return - self.deletePreset(comp, vers, name) - - def deletePreset(self, comp, vers, name): - action = DeletePreset(self, comp, vers, name) - self.parent.undoStack.push(action) - - def warnMessage(self, window=None): - self.parent.showMessage( - msg="Preset names must contain only letters, " "numbers, and spaces.", - parent=window if window else self, - ) - - def getPresetRow(self): - row = self.listWidget_presets.currentRow() - if row > -1: - return row - - # check if component selected in MainWindow has preset loaded - componentList = self.parent.listWidget_componentList - compIndex = componentList.currentRow() - if compIndex == -1: - return compIndex - - preset = self.core.selectedComponents[compIndex].currentPreset - if preset is None: - return -1 - else: - rowTuple = ( - self.core.selectedComponents[compIndex].name, - self.core.selectedComponents[compIndex].version, - preset, - ) - for i, tup in enumerate(self.presetRows): - if rowTuple == tup: - index = i - break - else: - return -1 - return index - - def openRenamePresetDialog(self): - presetList = self.listWidget_presets - index = self.getPresetRow() - if index == -1: - return - - while True: - newName, OK = QtWidgets.QInputDialog.getText( - self, - "Preset Manager", - "Rename Preset:", - QtWidgets.QLineEdit.EchoMode.Normal, - self.presetRows[index][2], - ) - if OK: - if badName(newName): - self.warnMessage() - continue - if newName: - comp, vers, oldName = self.presetRows[index] - path = os.path.join(self.presetDir, comp, str(vers)) - newPath = os.path.join(path, newName) - if self.presetExists(newPath): - return - action = RenamePreset(self, path, oldName, newName) - self.parent.undoStack.push(action) - break - - def renamePreset(self, path, oldName, newName): - oldPath = os.path.join(path, oldName) - newPath = os.path.join(path, newName) - if os.path.exists(newPath): - os.remove(newPath) - os.rename(oldPath, newPath) - self.findPresets() - self.drawPresetList() - path = os.path.dirname(newPath) - for i, comp in enumerate(self.core.selectedComponents): - if self.core.getPresetDir(comp) == path and comp.currentPreset == oldName: - self.core.openPreset(newPath, i, newName) - self.parent.updateComponentTitle(i, False) - self.parent.drawPreview() - - def openImportDialog(self): - filename, _ = QtWidgets.QFileDialog.getOpenFileName( - self, - "Import Preset File", - self.settings.value("presetDir"), - "Preset Files (*.avl)", - ) - if filename: - # get installed path & ask user to overwrite if needed - path = "" - while True: - if path: - if self.presetExists(path): - break - else: - if os.path.exists(path): - os.remove(path) - success, path = self.core.importPreset(filename) - if success: - break - - self.findPresets() - self.drawPresetList() - self.settings.setValue("presetDir", os.path.dirname(filename)) - - def openExportDialog(self): - index = self.getPresetRow() - if index == -1: - return - filename, _ = QtWidgets.QFileDialog.getSaveFileName( - self, - "Export Preset", - self.settings.value("presetDir"), - "Preset Files (*.avl)", - ) - if filename: - comp, vers, name = self.presetRows[index] - if not self.core.exportPreset(filename, comp, vers, name): - self.parent.showMessage( - msg="Couldn't export %s." % filename, parent=self - ) - self.settings.setValue("presetDir", os.path.dirname(filename)) - - def clearPresetListSelection(self): - self.listWidget_presets.setCurrentRow(-1) |