mininet-sample/core/parameter.py

import logging

class Parameter(object):
    """
    Generic handler for parameters stored in configuration files

    Attributes:
        parameters  dictionary containing the value for configuration parameters
    """
    def __init__(self, parameter_filename):
        self.parameters = {}
        if parameter_filename is None:
            logging.warning("No parameter file provided; using default parameters")
        else:
            logging.info("Extract parameters from file {}".format(parameter_filename))
            self.load_parameter_file(parameter_filename)

    def load_parameter_file(self, parameter_filename):
        with open(parameter_filename) as f:
            for line in f.readlines():
                # Ignore comments
                if line.startswith("#"):
                    continue

                try:
                    # Also get rid of trailing characters
                    key, value = line.strip().split(":")
                    if key in self.parameters:
                        if not isinstance(self.parameters[key], list):
                            self.parameters[key] = [self.parameters[key]]
                        self.parameters[key].append(value)
                    else:
                        self.parameters[key] =  value
                except ValueError as e:
                    logging.warning(
                        "Got error '{}' for line '{}'; ignore it".format(e, line))

    def get(self, key):
        """
        Get the parameter with key `key`. If it does not exist, return None
        """
        return self.parameters.get(key)

    def __str__(self):
        return self.parameters.__str__()