golden hour

/opt/imunify360/venv/lib/python3.11/site-packages/imav/patchman

File/Folder	Size	Actions
__init__.py	0 B
__pycache__	-
config.py	1.86 KB
constants.py	908 B
custom_integration.py	5.18 KB
exceptions.py	826 B
feature.py	3.65 KB
fs_scanner	-
license.py	5.29 KB
model.py	2.01 KB
rpc	-
state.py	921 B
utils.py	1.41 KB

Edit: config.py

"""
This program is free software: you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License,
or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. 
See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
 along with this program.  If not, see <https://www.gnu.org/licenses/>.

This software is also available under ImunifyAV commercial license,
see <https://www.imunify360.com/legal/eula>
"""
import configparser
from pathlib import Path
from typing import Optional

class PatchmanConfig:
    PATH = Path("/etc/patchman/patchman.ini")

@classmethod
    def _get_config(cls) -> dict:
        conf = configparser.ConfigParser()
        if cls.PATH.exists():
            conf.read(cls.PATH)
        return conf

@classmethod
    def get(cls, section: str, option: str, default=None) -> Optional[str]:
        """
        Return *option* value in *section* in config if exist,
        None otherwise.
        """
        conf = cls._get_config()
        try:
            return conf.get(section, option)
        except configparser.Error:
            return default

@classmethod
    def set(cls, section: str, option: str, value: str) -> None:
        """
        Set *value* for the *option* of the corresponding *section*
        in the patchman config.
        """
        conf = cls._get_config()
        if not conf.has_section(section):
            conf.add_section(section)
        conf.set(section, option, value)
        with cls.PATH.open("w") as config_file:
            conf.write(config_file)