From 21274856edf707dbe9743c7946d4747fa5176ef8 Mon Sep 17 00:00:00 2001 From: CodeByMrFinchum Date: Mon, 30 Dec 2024 23:36:18 +0100 Subject: [PATCH] Split the old main into optima35 and gui --- gui.py | 264 ++++++++++++++++++++++++++++++++++++++++++++++++++++ optima35.py | 109 ++++++++++++++++++++++ 2 files changed, 373 insertions(+) create mode 100644 gui.py create mode 100644 optima35.py diff --git a/gui.py b/gui.py new file mode 100644 index 0000000..aeaccd0 --- /dev/null +++ b/gui.py @@ -0,0 +1,264 @@ +import sys +import os +from datetime import datetime + +from optima35 import OPTIMA35 +from utils.utility import Utilities +from ui.main_window import Ui_MainWindow +from ui.exif_handler_window import ExifEditor + +from PySide6 import QtWidgets +from PySide6.QtWidgets import ( + QMessageBox, + QApplication, + QMainWindow, + QWidget, + QVBoxLayout, + QLabel, + QLineEdit, + QPushButton, + QCheckBox, + QFileDialog, + QHBoxLayout, + QSpinBox, +) + +class Optima35GUI(QMainWindow, Ui_MainWindow): + def __init__(self, exif_file): + super(Optima35GUI, self).__init__() + self.ui = Ui_MainWindow() + self.ui.setupUi(self) + self.o = OPTIMA35() + self.u = Utilities() + self.exif_file = exif_file + self.exif_data = None + + self.setWindowTitle(f"{self.o.name} {self.o.version}") + self.default_ui_layout() + self.define_gui_interaction() + + def default_ui_layout(self): + self.ui.png_quality_spinBox.setVisible(False) + + def define_gui_interaction(self): + self.ui.input_folder_button.clicked.connect(self.browse_input_folder) + self.ui.output_folder_button.clicked.connect(self.browse_output_folder) + self.ui.start_button.clicked.connect(self.process) + self.ui.image_type.currentIndexChanged.connect(self.update_quality_options) + + self.ui.exif_checkbox.stateChanged.connect( + lambda state: self.handle_checkbox_state(state, 2, self.populate_exif) + ) + self.ui.tabWidget.currentChanged.connect(self.on_tab_changed) + self.ui.edit_exif_button.clicked.connect(self.open_exif_editor) + self.ui.restart_button.clicked.connect(self.restart_app) + + def process(self): + #self.ui.start_button.setEnabled(False) + #self.ui.restart_button.setEnabled(False) + self.check_options() # Get all user selected data + input_folder_valid = os.path.exists(self.o.settings["input_folder"]) + output_folder_valid = os.path.exists(self.o.settings["output_folder"]) + if not input_folder_valid or not output_folder_valid: + QMessageBox.warning(self, "Warning", f"Input location {input_folder_valid}\nOutput folder {output_folder_valid}...") + return + + input_folder = self.o.settings["input_folder"] + output_folder = self.o.settings["output_folder"] + + image_files = [ + f for f in os.listdir(input_folder) if f.lower().endswith((".png", ".jpg", ".jpeg", ".webp")) + ] + i = 1 + for image_file in image_files: + input_path = os.path.join(input_folder, image_file) + if self.o.settings["new_file_names"] != False: + image_name = self.o.name_images(self.o.settings["new_file_names"], i, len(image_files), self.o.settings["invert_image_order"]) + else: + image_name = os.path.splitext(image_file)[0] + output_path = os.path.join(output_folder, image_name) + + self.o.process(input_path, output_path) + self.handle_qprogressbar(i, len(image_files)) + i += 1 + + QMessageBox.information(self, "Information", "Finished") + #self.ui.start_button.setEnabled(True) + #self.ui.restart_button.setEnabled(True) + self.ui.progressBar.setValue(0) + + def open_exif_editor(self): + """Open the EXIF Editor.""" + self.exif_editor = ExifEditor(self.exif_data) + self.exif_editor.exif_data_updated.connect(self.update_exif_data) + self.exif_editor.show() + + def update_exif_data(self, updated_exif_data): + """Update the EXIF data.""" + self.exif_data = updated_exif_data + self.populate_exif() + + def handle_checkbox_state(self, state, desired_state, action): + """Perform an action based on the checkbox state and a desired state. Have to use lambda when calling.""" + if state == desired_state: + action() + + def on_tab_changed(self, index): + """Handle tab changes.""" + # chatgpt + if index == 1: # EXIF Tab + self.handle_exif_file("read") + elif index == 0: # Main Tab + self.handle_exif_file("write") + + def handle_exif_file(self, do): + if do == "read": + self.exif_data = self.u.read_yaml(self.exif_file) + elif do == "write": + self.u.write_yaml(self.exif_file, self.exif_data) + + def populate_exif(self): + # partly chatGPT + # Mapping of EXIF fields to comboboxes in the UI + combo_mapping = { + "make": self.ui.make_comboBox, + "model": self.ui.model_comboBox, + "lens": self.ui.lens_comboBox, + "iso": self.ui.iso_comboBox, + "image_description": self.ui.image_description_comboBox, + "user_comment": self.ui.user_comment_comboBox, + "artist": self.ui.artist_comboBox, + "copyright_info": self.ui.copyright_info_comboBox, + } + self.populate_comboboxes(combo_mapping) + + def populate_comboboxes(self, combo_mapping): + """Populate comboboxes with EXIF data.""" + # ChatGPT + for field, comboBox in combo_mapping.items(): + comboBox.clear() # Clear existing items + comboBox.addItems(map(str, self.exif_data.get(field, []))) + + def update_quality_options(self): + """Update visibility of quality settings based on selected format.""" + # ChatGPT + selected_format = self.ui.image_type.currentText() + # Hide all quality settings + self.ui.png_quality_spinBox.setVisible(False) + self.ui.jpg_quality_spinBox.setVisible(False) + # Show relevant settings + if selected_format == "jpg": + self.ui.jpg_quality_spinBox.setVisible(True) + elif selected_format == "webp": + self.ui.jpg_quality_spinBox.setVisible(True) + elif selected_format == "png": + self.ui.png_quality_spinBox.setVisible(True) + + def browse_input_folder(self): + folder = QFileDialog.getExistingDirectory(self, "Select Input Folder") + if folder: + self.ui.input_path.setText(folder) + + def browse_output_folder(self): + folder = QFileDialog.getExistingDirectory(self, "Select Output Folder") + if folder: + self.ui.output_path.setText(folder) + + def change_statusbar(self, msg, timeout = 500): + self.ui.statusBar.showMessage(msg, timeout) + + def handle_qprogressbar(self, current, total): + progress = int((100 / total) * current) + self.ui.progressBar.setValue(progress) + + def check_options(self): + try: + self.o.settings["input_folder"] = self.ui.input_path.text() + self.o.settings["output_folder"] = self.ui.output_path.text() + self.o.settings["file_format"] = self.ui.image_type.currentText() + self.o.settings["jpg_quality"] = int(self.ui.jpg_quality_spinBox.text()) + self.o.settings["png_compression"] = int(self.ui.png_quality_spinBox.text()) + self.o.settings["invert_image_order"] = self.ui.revert_checkbox.isChecked() + self.o.settings["grayscale"] = self.ui.grayscale_checkBox.isChecked() + self.o.settings["copy_exif"] = self.ui.exif_copy_checkBox.isChecked() + self.o.settings["own_exif"] = self.ui.exif_checkbox.isChecked() + self.o.settings["font_size"] = self.ui.font_size_comboBox.currentIndex() + 1 + self.o.settings["optimize"] = self.ui.optimize_checkBox.isChecked() + self.o.settings["own_date"] = self.ui.add_date_checkBox.isChecked() + + if self.ui.resize_checkbox.isChecked(): + self.o.settings["resize"] = int(self.ui.resize_spinBox.text()) + + if self.ui.brightness_checkbox.isChecked(): + self.o.settings["brightness"] = int(self.ui.brightness_spinBox.text()) + + if self.ui.contrast_checkbox.isChecked(): + self.o.settings["contrast"] = int(self.ui.contrast_spinBox.text()) + + if self.ui.rename_checkbox.isChecked(): + if self.ui.filename.text() != "": + self.o.settings["new_file_names"] = self.ui.filename.text() + else: + self.o.settings["new_file_names"] = False + else: + self.o.settings["new_file_names"] = False + + + if self.ui.watermark_checkbox.isChecked(): + if self.ui.watermark_lineEdit.text() != "": + self.o.settings["watermark"] = self.ui.watermark_lineEdit.text() + else: + self.o.settings["watermark"] = False + else: + self.o.settings["watermark"] = False + + if self.o.settings["own_exif"]: + self.o.selected_exif = self.collect_selected_exif() + if self.ui.add_date_checkBox.isChecked(): + self.o.selected_exif["date_time_original"] = self.get_date() + if self.ui.gps_checkBox.isChecked(): + self.o.settings["gps"] = [self.ui.lat_lineEdit.text(), self.ui.long_lineEdit.text()] + else: + self.o.settings["gps"] = False + + except Exception as e: + print(f"Whoops: {e}") + + def get_date(self): + date_input = self.ui.dateEdit.date().toString("yyyy-MM-dd") + new_date = datetime.strptime(date_input, "%Y-%m-%d") + return new_date.strftime("%Y:%m:%d 00:00:00") + + def collect_selected_exif(self): + user_data = {} + user_data["make"] = self.ui.make_comboBox.currentText() + user_data["model"] = self.ui.model_comboBox.currentText() + user_data["lens"] = self.ui.lens_comboBox.currentText() + user_data["iso"] = self.ui.iso_comboBox.currentText() + user_data["image_description"] = self.ui.image_description_comboBox.currentText() + user_data["user_comment"] = self.ui.user_comment_comboBox.currentText() + user_data["artist"] = self.ui.artist_comboBox.currentText() + user_data["copyright_info"] = self.ui.copyright_info_comboBox.currentText() + user_data["software"] = f"{self.o.name} {self.o.version}" + return user_data + + def rebuild_ui(self): + # Define the bash script to execute + bash_script = "/home/sam/git/gitlab_public/optima-35/rebuild_ui.sh" + os.system(bash_script) + + def restart_app(self): + """Restarts the application.""" + self.rebuild_ui() + # chatGPT + python = sys.executable # Path to the Python interpreter + os.execv(python, [python] + sys.argv) + +def main(exif_file="local_files/exif.yaml"): + app = QtWidgets.QApplication(sys.argv) + window = Optima35GUI(exif_file=exif_file) + window.show() + app.exec() + +if __name__ == "__main__": + main() diff --git a/optima35.py b/optima35.py new file mode 100644 index 0000000..1b8b6c8 --- /dev/null +++ b/optima35.py @@ -0,0 +1,109 @@ +import re +from datetime import datetime +#from utils.utility import Utilities +from utils.image_handler import ImageProcessor, ExifHandler + +class OPTIMA35: + def __init__(self): + self.name = "OPTIMA-35" + self.version = "0.4.0" + #self.utilities = Utilities() + self.image_processor = ImageProcessor() + self.exif_handler = ExifHandler() + self.settings = { + "input_folder": None, + "output_folder": None, + "file_format": None, + "resize": False, + "contrast": False, + "brightness": False, + "new_file_names": False, + "invert_image_order": False, + "copy_exif": False, + "own_exif": False, + "watermark": False, + "grayscale": False, + "jpg_quality": None, + "png_compression": None, + "font_size": None, + "optimize": False, + "gps": False + } + self.selected_exif = None + + def modify_timestamp_in_exif(self, exif_data, filename): + """"Takes exif data and adjust time to fit ending of filename.""" + try: + last_tree = filename[-3:len(filename)] + total_seconds = int(re.sub(r'\D+', '', last_tree)) + minutes = total_seconds // 60 + seconds = total_seconds % 60 + time = datetime.strptime(exif_data["date_time_original"], "%Y:%m:%d %H:%M:%S") # change date time string back to an time object for modification + new_time = time.replace(hour=12, minute=minutes, second=seconds) + exif_data["date_time_original"] = new_time.strftime("%Y:%m:%d %H:%M:%S") + return exif_data + + except ValueError: + print("Modifying date went wrong, exiting...") + exit() + + def process(self, image_input_file, image_output_file): + with self.image_processor.open_image(image_input_file) as img: + processed_img = img + + if self.settings["resize"] != False: + processed_img = self.image_processor.resize_image( + image = processed_img, percent = self.settings["resize"] + ) + if self.settings["watermark"] != False: + processed_img = self.image_processor.add_watermark(processed_img, self.settings["watermark"], int(self.settings["font_size"])) + if self.settings["grayscale"] != False: # There is a problem, if we first to grayscale and then watermark it braeks + processed_img = self.image_processor.grayscale(processed_img) + if self.settings["brightness"] != False: # Does the order of brightness and contrast matter? + processed_img = self.image_processor.change_brightness(processed_img, self.settings["brightness"]) + if self.settings["contrast"] != False: # Does the order of brightness and contrast matter? + processed_img = self.image_processor.change_contrast(processed_img, self.settings["contrast"]) + + if self.settings["own_exif"] != False: + selected_exif = self.selected_exif + if "date_time_original" in self.selected_exif: + selected_exif = self.modify_timestamp_in_exif(selected_exif, image_name) + exif_data = self.exif_handler.build_exif_dict(selected_exif, self.image_processor.get_image_size(processed_img)) + if self.settings["gps"] != False: + latitude = float(self.settings["gps"][0]) + longitude = float(self.settings["gps"][1]) + exif_data = self.exif_handler.add_geolocation_to_exif(exif_data, latitude, longitude) + + elif self.settings["copy_exif"] == True: + # When copying exif from original, make sure to change Piexel X & Y Dimension to fit new size + try: + og_exif = self.exif_handler.get_exif_info(img) + og_exif["Exif"][40962], og_exif["Exif"][40963] = self.image_processor.get_image_size(processed_img) + exif_data = og_exif + except Exception: + # If an error happends it is because the picture does not have exif data + print("Copying EXIF data selected, but no EXIF data is available in the original image file.") + exif_data = None + elif self.settings["copy_exif"] == False: + exif_data = None + + self.image_processor.save_image( + image = processed_img, + path = image_output_file, + exif_data = exif_data, + file_type = self.settings["file_format"], + jpg_quality = self.settings["jpg_quality"], + png_compressing = self.settings["png_compression"], + optimize = self.settings["optimize"] + ) + + + def name_images(self, base_name, current_image, total_images, invert): + """"Returns name, combination of base_name and ending number.""" + total_digits = len(str(total_images)) + if invert: + ending_number = total_images - (current_image - 1) + else: + ending_number = current_image + ending = f"{ending_number:0{total_digits}}" + return f"{base_name}_{ending}"