# Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence. import filecmp import os import shutil from pathlib import Path from typing import TYPE_CHECKING import pkg_resources from primaite import getLogger, USERS_CONFIG_DIR if TYPE_CHECKING: from logging import Logger _LOGGER: Logger = getLogger(__name__) def run(overwrite_existing: bool = True) -> None: """ Resets the example config files in the users app config directory. :param overwrite_existing: A bool to toggle replacing existing edited config on or off. """ configs_package_data_root = pkg_resources.resource_filename("primaite", "config/_package_data") for subdir, dirs, files in os.walk(configs_package_data_root): for file in files: fp = os.path.join(subdir, file) path_split = os.path.relpath(fp, configs_package_data_root).split(os.sep) target_fp = USERS_CONFIG_DIR / "example_config" / Path(*path_split) target_fp.parent.mkdir(exist_ok=True, parents=True) copy_file = not target_fp.is_file() if overwrite_existing and not copy_file: copy_file = not filecmp.cmp(fp, target_fp) if copy_file: shutil.copy2(fp, target_fp) _LOGGER.info(f"Reset example config: {target_fp}")