"""Settings class for ComicTagger app""" # # Copyright 2012-2014 Anthony Beville # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. from __future__ import annotations import configparser import logging import os import pathlib import platform import uuid from collections.abc import Iterator from typing import TextIO, no_type_check from comicapi import utils logger = logging.getLogger(__name__) class ComicTaggerSettings: folder: pathlib.Path | str = "" @staticmethod def get_settings_folder() -> pathlib.Path: if not ComicTaggerSettings.folder: if platform.system() == "Windows": ComicTaggerSettings.folder = pathlib.Path(os.environ["APPDATA"]) / "ComicTagger" else: ComicTaggerSettings.folder = pathlib.Path(os.path.expanduser("~")) / ".ComicTagger" return pathlib.Path(ComicTaggerSettings.folder) @staticmethod def get_source_settings(source_name: str, source_settings: dict): def get_type(section: str, setting: dict): if setting["type"] is str: return config.get(section, setting["name"]) elif setting["type"] is int: return config.getint(section, setting["name"]) elif setting["type"] is float: return config.getfloat(section, setting["name"]) elif setting["type"] is bool: return config.getboolean(section, setting["name"]) settings_dir = ComicTaggerSettings.get_settings_folder() settings_file = os.path.join(settings_dir, "settings") config = configparser.RawConfigParser() try: with open(settings_file, encoding="utf-8") as f: config.read_file(f) for setting in source_settings.values(): setting["value"] = get_type(source_name, setting) return True except Exception: return False def __init__(self, folder: str | pathlib.Path | None) -> None: # General Settings self.rar_exe_path = "" self.allow_cbi_in_rar = True self.check_for_new_version = False self.send_usage_stats = False # automatic settings self.install_id = uuid.uuid4().hex self.last_selected_save_data_style = 0 self.last_selected_load_data_style = 0 self.last_opened_folder = "" self.last_main_window_width = 0 self.last_main_window_height = 0 self.last_main_window_x = 0 self.last_main_window_y = 0 self.last_form_side_width = -1 self.last_list_side_width = -1 self.last_filelist_sorted_column = -1 self.last_filelist_sorted_order = 0 # identifier settings self.id_series_match_search_thresh = 90 self.id_series_match_identify_thresh = 91 self.id_publisher_filter = "Panini Comics, Abril, Planeta DeAgostini, Editorial Televisa, Dino Comics" self.comic_info_source = "comicvine" # Default to CV as should always be present # Show/ask dialog flags self.ask_about_cbi_in_rar = True self.show_disclaimer = True self.dont_notify_about_this_version = "" self.ask_about_usage_stats = True # filename parsing settings self.complicated_parser = False self.remove_c2c = False self.remove_fcbd = False self.remove_publisher = False # Comic source general settings self.clear_form_before_populating = False self.auto_imprint = False self.sort_series_by_year = True self.exact_series_matches_first = True self.always_use_publisher_filter = False # CBL Transform settings self.assume_lone_credit_is_primary = False self.copy_characters_to_tags = False self.copy_teams_to_tags = False self.copy_locations_to_tags = False self.copy_storyarcs_to_tags = False self.copy_notes_to_comments = False self.copy_weblink_to_comments = False self.apply_cbl_transform_on_ct_import = False self.apply_cbl_transform_on_bulk_operation = False # Rename settings self.rename_template = "{series} #{issue} ({year})" self.rename_issue_number_padding = 3 self.rename_use_smart_string_cleanup = True self.rename_extension_based_on_archive = True self.rename_dir = "" self.rename_move_dir = False self.rename_strict = False # Auto-tag stickies self.save_on_low_confidence = False self.dont_use_year_when_identifying = False self.assume_1_if_no_issue_num = False self.ignore_leading_numbers_in_filename = False self.remove_archive_after_successful_match = False self.wait_and_retry_on_rate_limit = False self.config = configparser.RawConfigParser() if folder: ComicTaggerSettings.folder = pathlib.Path(folder) else: ComicTaggerSettings.folder = ComicTaggerSettings.get_settings_folder() if not os.path.exists(ComicTaggerSettings.folder): os.makedirs(ComicTaggerSettings.folder) self.settings_file = os.path.join(ComicTaggerSettings.folder, "settings") # if config file doesn't exist, write one out if not os.path.exists(self.settings_file): self.save() else: self.load() # take a crack at finding rar exe, if not set already if self.rar_exe_path == "": if platform.system() == "Windows": # look in some likely places for Windows machines if os.path.exists(r"C:\Program Files\WinRAR\Rar.exe"): self.rar_exe_path = r"C:\Program Files\WinRAR\Rar.exe" elif os.path.exists(r"C:\Program Files (x86)\WinRAR\Rar.exe"): self.rar_exe_path = r"C:\Program Files (x86)\WinRAR\Rar.exe" else: if os.path.exists("/opt/homebrew/bin"): utils.add_to_path("/opt/homebrew/bin") # see if it's in the path of unix user rarpath = utils.which("rar") if rarpath is not None: self.rar_exe_path = "rar" if self.rar_exe_path != "": self.save() if self.rar_exe_path != "": # make sure rar program is now in the path for the rar class utils.add_to_path(os.path.dirname(self.rar_exe_path)) def reset(self): os.unlink(self.settings_file) self.__init__(ComicTaggerSettings.folder) def load(self) -> None: def readline_generator(f: TextIO) -> Iterator[str]: line = f.readline() while line: yield line line = f.readline() with open(self.settings_file, encoding="utf-8") as f: self.config.read_file(readline_generator(f)) self.rar_exe_path = self.config.get("settings", "rar_exe_path") if self.config.has_option("settings", "check_for_new_version"): self.check_for_new_version = self.config.getboolean("settings", "check_for_new_version") if self.config.has_option("settings", "send_usage_stats"): self.send_usage_stats = self.config.getboolean("settings", "send_usage_stats") if self.config.has_option("auto", "install_id"): self.install_id = self.config.get("auto", "install_id") if self.config.has_option("auto", "last_selected_load_data_style"): self.last_selected_load_data_style = self.config.getint("auto", "last_selected_load_data_style") if self.config.has_option("auto", "last_selected_save_data_style"): self.last_selected_save_data_style = self.config.getint("auto", "last_selected_save_data_style") if self.config.has_option("auto", "last_opened_folder"): self.last_opened_folder = self.config.get("auto", "last_opened_folder") if self.config.has_option("auto", "last_main_window_width"): self.last_main_window_width = self.config.getint("auto", "last_main_window_width") if self.config.has_option("auto", "last_main_window_height"): self.last_main_window_height = self.config.getint("auto", "last_main_window_height") if self.config.has_option("auto", "last_main_window_x"): self.last_main_window_x = self.config.getint("auto", "last_main_window_x") if self.config.has_option("auto", "last_main_window_y"): self.last_main_window_y = self.config.getint("auto", "last_main_window_y") if self.config.has_option("auto", "last_form_side_width"): self.last_form_side_width = self.config.getint("auto", "last_form_side_width") if self.config.has_option("auto", "last_list_side_width"): self.last_list_side_width = self.config.getint("auto", "last_list_side_width") if self.config.has_option("auto", "last_filelist_sorted_column"): self.last_filelist_sorted_column = self.config.getint("auto", "last_filelist_sorted_column") if self.config.has_option("auto", "last_filelist_sorted_order"): self.last_filelist_sorted_order = self.config.getint("auto", "last_filelist_sorted_order") if self.config.has_option("identifier", "id_series_match_search_thresh"): self.id_series_match_search_thresh = self.config.getint("identifier", "id_series_match_search_thresh") if self.config.has_option("identifier", "id_series_match_identify_thresh"): self.id_series_match_identify_thresh = self.config.getint("identifier", "id_series_match_identify_thresh") if self.config.has_option("identifier", "id_publisher_filter"): self.id_publisher_filter = self.config.get("identifier", "id_publisher_filter") if self.config.has_option("identifier", "always_use_publisher_filter"): self.always_use_publisher_filter = self.config.getboolean("identifier", "always_use_publisher_filter") if self.config.has_option("filenameparser", "complicated_parser"): self.complicated_parser = self.config.getboolean("filenameparser", "complicated_parser") if self.config.has_option("filenameparser", "remove_c2c"): self.remove_c2c = self.config.getboolean("filenameparser", "remove_c2c") if self.config.has_option("filenameparser", "remove_fcbd"): self.remove_fcbd = self.config.getboolean("filenameparser", "remove_fcbd") if self.config.has_option("filenameparser", "remove_publisher"): self.remove_publisher = self.config.getboolean("filenameparser", "remove_publisher") if self.config.has_option("dialogflags", "ask_about_cbi_in_rar"): self.ask_about_cbi_in_rar = self.config.getboolean("dialogflags", "ask_about_cbi_in_rar") if self.config.has_option("dialogflags", "show_disclaimer"): self.show_disclaimer = self.config.getboolean("dialogflags", "show_disclaimer") if self.config.has_option("dialogflags", "dont_notify_about_this_version"): self.dont_notify_about_this_version = self.config.get("dialogflags", "dont_notify_about_this_version") if self.config.has_option("dialogflags", "ask_about_usage_stats"): self.ask_about_usage_stats = self.config.getboolean("dialogflags", "ask_about_usage_stats") if self.config.has_option("comic_source_general", "clear_form_before_populating"): self.clear_form_before_populating = self.config.getboolean( "comic_source_general", "clear_form_before_populating" ) if self.config.has_option("comic_source_general", "sort_series_by_year"): self.sort_series_by_year = self.config.getboolean("comic_source_general", "sort_series_by_year") if self.config.has_option("comic_source_general", "exact_series_matches_first"): self.exact_series_matches_first = self.config.getboolean( "comic_source_general", "exact_series_matches_first" ) if self.config.has_option("comic_source_general", "comic_info_source"): self.comic_info_source = self.config.get("identifier", "comic_info_source") if self.config.has_option("cbl_transform", "assume_lone_credit_is_primary"): self.assume_lone_credit_is_primary = self.config.getboolean( "cbl_transform", "assume_lone_credit_is_primary" ) if self.config.has_option("cbl_transform", "copy_characters_to_tags"): self.copy_characters_to_tags = self.config.getboolean("cbl_transform", "copy_characters_to_tags") if self.config.has_option("cbl_transform", "copy_teams_to_tags"): self.copy_teams_to_tags = self.config.getboolean("cbl_transform", "copy_teams_to_tags") if self.config.has_option("cbl_transform", "copy_locations_to_tags"): self.copy_locations_to_tags = self.config.getboolean("cbl_transform", "copy_locations_to_tags") if self.config.has_option("cbl_transform", "copy_notes_to_comments"): self.copy_notes_to_comments = self.config.getboolean("cbl_transform", "copy_notes_to_comments") if self.config.has_option("cbl_transform", "copy_storyarcs_to_tags"): self.copy_storyarcs_to_tags = self.config.getboolean("cbl_transform", "copy_storyarcs_to_tags") if self.config.has_option("cbl_transform", "copy_weblink_to_comments"): self.copy_weblink_to_comments = self.config.getboolean("cbl_transform", "copy_weblink_to_comments") if self.config.has_option("cbl_transform", "apply_cbl_transform_on_ct_import"): self.apply_cbl_transform_on_ct_import = self.config.getboolean( "cbl_transform", "apply_cbl_transform_on_ct_import" ) if self.config.has_option("cbl_transform", "apply_cbl_transform_on_bulk_operation"): self.apply_cbl_transform_on_bulk_operation = self.config.getboolean( "cbl_transform", "apply_cbl_transform_on_bulk_operation" ) if self.config.has_option("rename", "rename_template"): self.rename_template = self.config.get("rename", "rename_template") if self.config.has_option("rename", "rename_issue_number_padding"): self.rename_issue_number_padding = self.config.getint("rename", "rename_issue_number_padding") if self.config.has_option("rename", "rename_use_smart_string_cleanup"): self.rename_use_smart_string_cleanup = self.config.getboolean("rename", "rename_use_smart_string_cleanup") if self.config.has_option("rename", "rename_extension_based_on_archive"): self.rename_extension_based_on_archive = self.config.getboolean( "rename", "rename_extension_based_on_archive" ) if self.config.has_option("rename", "rename_dir"): self.rename_dir = self.config.get("rename", "rename_dir") if self.config.has_option("rename", "rename_move_dir"): self.rename_move_dir = self.config.getboolean("rename", "rename_move_dir") if self.config.has_option("rename", "rename_strict"): self.rename_strict = self.config.getboolean("rename", "rename_strict") if self.config.has_option("autotag", "save_on_low_confidence"): self.save_on_low_confidence = self.config.getboolean("autotag", "save_on_low_confidence") if self.config.has_option("autotag", "dont_use_year_when_identifying"): self.dont_use_year_when_identifying = self.config.getboolean("autotag", "dont_use_year_when_identifying") if self.config.has_option("autotag", "assume_1_if_no_issue_num"): self.assume_1_if_no_issue_num = self.config.getboolean("autotag", "assume_1_if_no_issue_num") if self.config.has_option("autotag", "ignore_leading_numbers_in_filename"): self.ignore_leading_numbers_in_filename = self.config.getboolean( "autotag", "ignore_leading_numbers_in_filename" ) if self.config.has_option("autotag", "remove_archive_after_successful_match"): self.remove_archive_after_successful_match = self.config.getboolean( "autotag", "remove_archive_after_successful_match" ) if self.config.has_option("autotag", "wait_and_retry_on_rate_limit"): self.wait_and_retry_on_rate_limit = self.config.getboolean("autotag", "wait_and_retry_on_rate_limit") if self.config.has_option("autotag", "auto_imprint"): self.auto_imprint = self.config.getboolean("autotag", "auto_imprint") @no_type_check def save(self) -> None: if not self.config.has_section("settings"): self.config.add_section("settings") self.config.set("settings", "check_for_new_version", self.check_for_new_version) self.config.set("settings", "rar_exe_path", self.rar_exe_path) self.config.set("settings", "send_usage_stats", self.send_usage_stats) if not self.config.has_section("auto"): self.config.add_section("auto") self.config.set("auto", "install_id", self.install_id) self.config.set("auto", "last_selected_load_data_style", self.last_selected_load_data_style) self.config.set("auto", "last_selected_save_data_style", self.last_selected_save_data_style) self.config.set("auto", "last_opened_folder", self.last_opened_folder) self.config.set("auto", "last_main_window_width", self.last_main_window_width) self.config.set("auto", "last_main_window_height", self.last_main_window_height) self.config.set("auto", "last_main_window_x", self.last_main_window_x) self.config.set("auto", "last_main_window_y", self.last_main_window_y) self.config.set("auto", "last_form_side_width", self.last_form_side_width) self.config.set("auto", "last_list_side_width", self.last_list_side_width) self.config.set("auto", "last_filelist_sorted_column", self.last_filelist_sorted_column) self.config.set("auto", "last_filelist_sorted_order", self.last_filelist_sorted_order) if not self.config.has_section("identifier"): self.config.add_section("identifier") self.config.set("identifier", "id_series_match_search_thresh", self.id_series_match_search_thresh) self.config.set("identifier", "id_series_match_identify_thresh", self.id_series_match_identify_thresh) self.config.set("identifier", "id_publisher_filter", self.id_publisher_filter) self.config.set("identifier", "always_use_publisher_filter", self.always_use_publisher_filter) if not self.config.has_section("dialogflags"): self.config.add_section("dialogflags") self.config.set("dialogflags", "ask_about_cbi_in_rar", self.ask_about_cbi_in_rar) self.config.set("dialogflags", "show_disclaimer", self.show_disclaimer) self.config.set("dialogflags", "dont_notify_about_this_version", self.dont_notify_about_this_version) self.config.set("dialogflags", "ask_about_usage_stats", self.ask_about_usage_stats) if not self.config.has_section("filenameparser"): self.config.add_section("filenameparser") self.config.set("filenameparser", "complicated_parser", self.complicated_parser) self.config.set("filenameparser", "remove_c2c", self.remove_c2c) self.config.set("filenameparser", "remove_fcbd", self.remove_fcbd) self.config.set("filenameparser", "remove_publisher", self.remove_publisher) if not self.config.has_section("comic_source_general"): self.config.add_section("comic_source_general") self.config.set("comic_source_general", "clear_form_before_populating", self.clear_form_before_populating) self.config.set("comic_source_general", "sort_series_by_year", self.sort_series_by_year) self.config.set("comic_source_general", "exact_series_matches_first", self.exact_series_matches_first) self.config.set("comic_source_general", "comic_info_source", self.comic_info_source) if not self.config.has_section("cbl_transform"): self.config.add_section("cbl_transform") self.config.set("cbl_transform", "assume_lone_credit_is_primary", self.assume_lone_credit_is_primary) self.config.set("cbl_transform", "copy_characters_to_tags", self.copy_characters_to_tags) self.config.set("cbl_transform", "copy_teams_to_tags", self.copy_teams_to_tags) self.config.set("cbl_transform", "copy_locations_to_tags", self.copy_locations_to_tags) self.config.set("cbl_transform", "copy_storyarcs_to_tags", self.copy_storyarcs_to_tags) self.config.set("cbl_transform", "copy_notes_to_comments", self.copy_notes_to_comments) self.config.set("cbl_transform", "copy_weblink_to_comments", self.copy_weblink_to_comments) self.config.set("cbl_transform", "apply_cbl_transform_on_ct_import", self.apply_cbl_transform_on_ct_import) self.config.set( "cbl_transform", "apply_cbl_transform_on_bulk_operation", self.apply_cbl_transform_on_bulk_operation, ) if not self.config.has_section("rename"): self.config.add_section("rename") self.config.set("rename", "rename_template", self.rename_template) self.config.set("rename", "rename_issue_number_padding", self.rename_issue_number_padding) self.config.set("rename", "rename_use_smart_string_cleanup", self.rename_use_smart_string_cleanup) self.config.set("rename", "rename_extension_based_on_archive", self.rename_extension_based_on_archive) self.config.set("rename", "rename_dir", self.rename_dir) self.config.set("rename", "rename_move_dir", self.rename_move_dir) self.config.set("rename", "rename_strict", self.rename_strict) if not self.config.has_section("autotag"): self.config.add_section("autotag") self.config.set("autotag", "save_on_low_confidence", self.save_on_low_confidence) self.config.set("autotag", "dont_use_year_when_identifying", self.dont_use_year_when_identifying) self.config.set("autotag", "assume_1_if_no_issue_num", self.assume_1_if_no_issue_num) self.config.set("autotag", "ignore_leading_numbers_in_filename", self.ignore_leading_numbers_in_filename) self.config.set("autotag", "remove_archive_after_successful_match", self.remove_archive_after_successful_match) self.config.set("autotag", "wait_and_retry_on_rate_limit", self.wait_and_retry_on_rate_limit) self.config.set("autotag", "auto_imprint", self.auto_imprint) # NOTE: Source settings are added in settingswindow module with open(self.settings_file, "w", encoding="utf-8") as configfile: self.config.write(configfile)