mirror of
https://github.com/dw-0/kiauh.git
synced 2025-12-23 15:53:36 +05:00
86 lines
3.2 KiB
Python
86 lines
3.2 KiB
Python
#!/usr/bin/env python3
|
|
|
|
# ======================================================================= #
|
|
# Copyright (C) 2020 - 2023 Dominik Willner <th33xitus@gmail.com> #
|
|
# #
|
|
# This file is part of KIAUH - Klipper Installation And Update Helper #
|
|
# https://github.com/dw-0/kiauh #
|
|
# #
|
|
# This file may be distributed under the terms of the GNU GPLv3 license #
|
|
# ======================================================================= #
|
|
|
|
import configparser
|
|
from pathlib import Path
|
|
from typing import Union
|
|
|
|
from kiauh.utils.logger import Logger
|
|
|
|
|
|
# noinspection PyMethodMayBeStatic
|
|
class ConfigManager:
|
|
def __init__(self, cfg_file: Path):
|
|
self.config_file = cfg_file
|
|
self.config = CustomConfigParser()
|
|
|
|
if cfg_file.is_file():
|
|
self.read_config()
|
|
|
|
def read_config(self) -> None:
|
|
if not self.config_file:
|
|
Logger.print_error("Unable to read config file. File not found.")
|
|
return
|
|
|
|
self.config.read_file(open(self.config_file, "r"))
|
|
|
|
def write_config(self) -> None:
|
|
with open(self.config_file, "w") as cfg:
|
|
self.config.write(cfg)
|
|
|
|
def get_value(self, section: str, key: str, silent=True) -> Union[str, bool, None]:
|
|
if not self.config.has_section(section):
|
|
if not silent:
|
|
log = f"Section not defined. Unable to read section: [{section}]."
|
|
Logger.print_error(log)
|
|
return None
|
|
|
|
if not self.config.has_option(section, key):
|
|
if not silent:
|
|
log = f"Option not defined in section [{section}]. Unable to read option: '{key}'."
|
|
Logger.print_error(log)
|
|
return None
|
|
|
|
value = self.config.get(section, key)
|
|
if value == "True" or value == "true":
|
|
return True
|
|
elif value == "False" or value == "false":
|
|
return False
|
|
else:
|
|
return value
|
|
|
|
def set_value(self, section: str, key: str, value: str):
|
|
self.config.set(section, key, value)
|
|
|
|
|
|
class CustomConfigParser(configparser.ConfigParser):
|
|
"""
|
|
A custom ConfigParser class overwriting the write() method of configparser.Configparser.
|
|
Key and value will be delimited by a ": ".
|
|
Note the whitespace AFTER the colon, which is the whole reason for that overwrite.
|
|
"""
|
|
|
|
def write(self, fp, space_around_delimiters=False):
|
|
if self._defaults:
|
|
fp.write("[%s]\n" % configparser.DEFAULTSECT)
|
|
for key, value in self._defaults.items():
|
|
fp.write("%s: %s\n" % (key, str(value).replace("\n", "\n\t")))
|
|
fp.write("\n")
|
|
for section in self._sections:
|
|
fp.write("[%s]\n" % section)
|
|
for key, value in self._sections[section].items():
|
|
if key == "__name__":
|
|
continue
|
|
if (value is not None) or (self._optcre == self.OPTCRE):
|
|
key = ": ".join((key, str(value).replace("\n", "\n\t")))
|
|
fp.write("%s\n" % key)
|
|
fp.write("\n")
|