aswi2022bug-thugs-gitlab/client/config_manager.py @ 474551ae

from sys import exit
from configparser import RawConfigParser


class Config:

    def __init__(self, filepath):
        self.config = RawConfigParser()
        if not self.config.read(filepath):
            print(f"Failed to parse the config file {filepath}. Make sure you entered a valid path.")
            exit(1)

        self._parse_usb_detector_section()
        self._parse_server_section()
        self._parse_logger_section()
        self._parse_cache_section()

    def _parse_usb_detector_section(self):
        section_name = "usb_detector"
        self.scan_period_seconds = float(self.config[section_name]["scan_period_seconds"])
        self.connected_devices_filename = self.config[section_name]["connected_devices_filename"]

    def _parse_server_section(self):
        section_name = "server"
        self.server_url = self.config[section_name]["url"]
        self.server_port = self.config[section_name]["port"]
        self.server_endpoint = self.config[section_name]["end_point"]

    def _parse_logger_section(self):
        section_name = "logger"
        self.logger_config_file = self.config[section_name]["config_file"]

    def _parse_cache_section(self):
        section_name = "cache"
        self.cache_dir = self.config[section_name]["directory"]
        self.cache_max_entries = int(self.config[section_name]["max_entries"])
        self.cache_max_retries = int(self.config[section_name]["max_retries"])
        self.cache_retry_period_seconds = float(self.config[section_name]["retry_period_seconds"])