40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
# 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}")
|