config.py

import json
import shutil
import os
from SystemInfo import SystemInfo
from typing import Any
from screenshotters import availableTools

sysInfo = SystemInfo()

defaultConfig: dict[str, Any] = {
    "ENABLED": True,
    "AUTO_DELETE_TYPE": "Disabled",
    "DELETE_AFTER_PERIOD": 1,
    "SCREENSHOT_FREQUENCY_MS": 3_600_000,  # once per hour
    "SAVE_LOCATION": sysInfo.getLocation(sysInfo.dbPath),
    "DATE_FORMAT": r"%d.%m. %Y %H:%M.%S",
    "SCREENSHOT_TOOL": ("Default" if not sysInfo.isWayland else (availableTools[0] if availableTools else "N/A")),
    "USE_PASSWORD": False,
    "BASEKEY": "",
    "SALT": ""
}

config: dict[str, Any] = {}
savePath = f"{sysInfo.dataDir}/settings.json"


def makeConfigIfNotExists() -> None:
    file = None
    try:
        file = open(savePath, "r")
    except:
        if sysInfo.info:
            print("No existing configuration file. (created)")

        file = open(savePath, "w")
        file.write(json.dumps(defaultConfig))
    finally:
        if file:
            file.close()


def loadConfig():
    global config
    data = {}
    makeConfigIfNotExists()
    with open(savePath, "r") as settingsFile:
        settings = settingsFile.read()
        data = json.loads(settings)

    targetKeys = list(defaultConfig.keys())
    for i in targetKeys:
        if i not in data:
            data[i] = defaultConfig[i]

    if data["SCREENSHOT_TOOL"] == "Default" and sysInfo.isWayland:
        data["SCREENSHOT_TOOL"] = defaultConfig["SCREENSHOT_TOOL"]

    config = data


def saveConfig() -> bool:
    shutil.copyfile(savePath, f"{savePath}.bak")
    fileSaved: bool = _saveConfigFile()

    if not fileSaved:
        shutil.move(f"{savePath}.bak", savePath)
        return False
    else:
        os.remove(f"{savePath}.bak")

    return True


def _saveConfigFile() -> bool:
    succ: bool = False

    try:
        with open(savePath, "w") as f:
            f.write(json.dumps(config))
            succ = True
    except:
        succ = False

    return succ


def get(setting: str) -> Any:
    if setting in config:
        return config[setting]


def set(setting: str, value: Any) -> bool:
    config[setting] = value
    return saveConfig()