Source code for megabeast.mbsettings

__all__ = ["mbsettings"]




class mbsettings:
    """
    All of the settings for the MegaBEAST

    Attributes
    ----------
    settings_file : string
        name of the parameter file that contains the settings in python format
        (e.g., supports dictonaries)
    """

    def __init__(self, settings_file=None):
        """
        Parameters
        ----------
        settings_file : string
            input file name
        """
        if settings_file is not None:
            self.settings_file = settings_file
            self.read()
            self.verify()



    def read(self):
        """
        Read in the settings file and set parameters
        """

        # read everything in as strings
        with open(self.settings_file, "r") as f:
            temp_data = f.readlines()
        # remove empty lines and comments
        input_data = [
            line.strip()
            for line in temp_data
            if line.strip() != "" and line.strip()[0] != "#"
        ]
        # remove comments that are mid-line (e.g., "x = 5 #comment")
        for i, line in enumerate(input_data):
            try:
                input_data[i] = line[: line.index("#")]
            except ValueError:
                pass
        # if parameters are defined over multiple lines, combine lines
        for i in reversed(range(len(input_data))):
            if ("import " not in input_data[i]) and ("=" not in input_data[i]):
                input_data[i - 1] += input_data[i]
                del input_data[i]

        # parse it into a dictionary
        beast_params = {}

        for i in range(len(input_data)):
            # execute imports
            if "import " in input_data[i]:
                exec(input_data[i])

            # extract parameter and value (as strings)
            else:
                param = input_data[i].split("=")[0].strip()
                # exec the string to get parameter values accessible
                exec(input_data[i])
                beast_params[param] = eval(param)

        # turn dictionary into attributes
        for key in beast_params:
            setattr(self, key, beast_params[key])




    def verify(self):
        """
        Run the verification code on the settings
        """
        pass


        # verify_beast_settings.verify_input_format(self)