????

Your IP : 3.144.109.104


Current Path : /opt/imunify360/venv/lib64/python3.11/site-packages/imav/patchman/
Upload File :
Current File : //opt/imunify360/venv/lib64/python3.11/site-packages/imav/patchman/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/>.

Copyright © 2019 Cloud Linux Software Inc.

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)