Automatically set the output file name based on the input BRF value.

Author: Michael Whapples

src/convert2ebrl/brf_to_ebrf.py

@@ -12,23 +12,22 @@
 
 from PySide6.QtCore import QObject, Slot, Signal, QThreadPool, QSettings, QCoreApplication
 from PySide6.QtGui import QAction
-from PySide6.QtWidgets import QWidget, QDialog, QDialogButtonBox, QVBoxLayout, \
+from PySide6.QtWidgets import QWidget, QDialogButtonBox, QVBoxLayout, \
     QProgressDialog, QMessageBox, QTabWidget, QFileDialog, QComboBox, QHBoxLayout, QMenu, QPushButton, QLabel, \
     QInputDialog
 from brf2ebrl.common import PageLayout
 from brf2ebrl.parser import EBrailleParserOptions
-from brf2ebrl.plugin import  find_plugins
+from brf2ebrl.plugin import find_plugins
 
 from convert2ebrl.convert_task import ConvertTask, Notification
 from convert2ebrl.settings import SettingsProfile
 from convert2ebrl.settings.defaults import DEFAULT_SETTINGS_PROFILES_LIST
-from convert2ebrl.tabs.general_tab import ConversionGeneralSettingsWidget
+from convert2ebrl.tabs.general_tab import ConversionGeneralSettingsWidget, expand_input_brfs
 from convert2ebrl.tabs.metadata_tab import MetadataWidget
 from convert2ebrl.tabs.page_settings_tab import ConversionPageSettingsWidget
 from convert2ebrl.utils import RunnableAdapter, load_settings_profiles, save_settings_profiles, load_settings_profile, \
     save_settings_profile
 
-
 DISCOVERED_PARSER_PLUGINS = find_plugins()
 
 class SettingsProfilesWidget(QWidget):
@@ -210,9 +209,7 @@ def _update_validity(self):
     @Slot()
     def on_apply(self):
         number_of_steps = 1000
-        brf_list = [brf for f in self._brf2ebrf_form.input_brfs for brf in
-                    ([os.path.join(f, b) for b in os.listdir(f) if
-                      os.path.splitext(b)[1].lower() == ".brf"] if os.path.isdir(f) else [f])]
+        brf_list = expand_input_brfs(self._brf2ebrf_form.input_brfs)
         num_of_inputs = len(brf_list)
         output_ebrf = self._brf2ebrf_form.output_ebrf
         if os.path.exists(output_ebrf):

src/convert2ebrl/tabs/general_tab.py

@@ -7,7 +7,7 @@
 
 import os
 
-from PySide6.QtCore import Signal, QObject, QSettings, Slot
+from PySide6.QtCore import Signal, QObject, QSettings, Slot, QStandardPaths
 from PySide6.QtWidgets import QWidget, QFormLayout, QComboBox, QCheckBox, QFileDialog
 
 from convert2ebrl.settings.defaults import CONVERSION_LAST_DIR as DEFAULT_LAST_DIR
@@ -68,9 +68,21 @@ def on_input_type_changed(index):
         self._include_images_checkbox.toggled.connect(self._update_include_images_state)
         self._input_type_combo.currentIndexChanged.connect(self._clear_input_brf)
         self._input_brf_edit.fileChanged.connect(self.inputBrfChanged.emit)
+        self._input_brf_edit.fileChanged.connect(self._update_output_based_on_input)
         self._image_dir_edit.fileChanged.connect(self.imagesDirectoryChanged.emit)
         self._output_ebrf_edit.fileChanged.connect(self.outputEbrfChanged.emit)
 
+    def _update_output_based_on_input(self):
+        brf_list = [os.path.splitext(os.path.basename(x))[0] for x in expand_input_brfs(self.input_brfs)]
+        if brf_list:
+            output_prefix = os.path.commonprefix(brf_list)
+            output_file = (output_prefix if output_prefix else brf_list[0]) + ".ebrl"
+            output_dir = os.path.join(QStandardPaths.writableLocation(QStandardPaths.StandardLocation.DocumentsLocation) if QStandardPaths.writableLocation(QStandardPaths.StandardLocation.DocumentsLocation) else os.path.expanduser("~"), "ebraille")
+            os.makedirs(output_dir, exist_ok=True)
+            self.output_ebrf = os.path.join(output_dir, output_file)
+        else:
+            self.output_ebrf = ""
+
     @Slot(bool)
     def _update_include_images_state(self, checked: bool):
         self._image_dir_edit.setEnabled(checked)
@@ -126,3 +138,9 @@ def output_ebrf(self) -> str:
     @output_ebrf.setter
     def output_ebrf(self, value: str):
         self._output_ebrf_edit.file_name = value
+
+
+def expand_input_brfs(input_brfs: list[str]) -> list[str]:
+    return [brf for f in input_brfs for brf in
+            ([os.path.join(f, b) for b in os.listdir(f) if
+              os.path.splitext(b)[1].lower() == ".brf"] if os.path.isdir(f) else [f])]