config.py

import configparser
import sys


class Config:
    def __init__(self, datadir, program):
        self.config = configparser.ConfigParser()
        self.dataset = self.config.read(f"{datadir}/{program}.ini")
        if self.dataset == []:
            self.dataset = self.config.read("config.ini")
        self.fixstrings = ["account_name", "prefix", "subprefix", "suffix"]
        self.datadir = datadir
        self.program = program

    def get(self, attribute, defaultvalue="", section: str = None):
        data = ""

        if len(self.dataset) != 1:
            sys.tracebacklimit = 0
            sys.exit(
                f"Cannot read {self.datadir}/{self.program}.ini or config.ini! - Please make sure it exists in the folder where 3cqsbot.py is executed."
            )

        sections = self.config.sections()

        if section == None:
            for section in sections:
                if self.config.has_option(section, attribute):
                    raw_value = self.config[section].get(attribute)

                    if raw_value:
                        if attribute in self.fixstrings:
                            data = raw_value
                        else:
                            data = self.check_type(raw_value)
                        break
        elif self.config.has_option(section, attribute):
            raw_value = self.config[section].get(attribute)
            if raw_value:
                if attribute in self.fixstrings:
                    data = raw_value
                else:
                    data = self.check_type(raw_value)

        if data == "" and str(defaultvalue):
            data = defaultvalue
        elif data == "" and defaultvalue == "" and not attribute == "botid":
            sys.tracebacklimit = 0
            sys.exit(
                "Make sure that section ["
                + section
                + "] is defined and mandatory attribute '"
                + attribute
                + "' is set. Please check the readme for configuration. Exiting script!"
            )

        return data

    def isfloat(self, element):
        try:
            float(element)
            return True
        except ValueError:
            return False

    def check_type(self, raw_value):
        data = ""

        if raw_value.isdigit():
            data = int(raw_value)
        elif raw_value.lower() == "true":
            data = True
        elif raw_value.lower() == "false":
            data = False
        elif self.isfloat(raw_value):
            data = float(raw_value)
        else:
            data = str(raw_value)

        return data