K2PKConfig.py

from configparser import ConfigParser


def read_db_config(filename='config.ini', section='mysql'):
    """ Read database configuration file and return a dictionary object.

    :param filename: name of the configuration file
    :param section: section of database configuration
    :return: a dictionary of database parameters
    """
    # create parser and read ini configuration file
    parser = ConfigParser()
    parser.read(filename)

    # get section, default to mysql
    db = {}
    if parser.has_section(section):
        items = parser.items(section)
        for item in items:
            db[item[0]] = item[1]
    else:
        raise Exception(
            '{0} not found in the {1} file'.format(section, filename))

    return db


def read_octopart_config(filename='config.ini', section='octopart'):
    """ Read database configuration file and return a dictionary object.

    :param filename: name of the configuration file
    :param section: section of database configuration
    :return: a dictionary of Octopart parameters
    """
    # create parser and read ini configuration file
    parser = ConfigParser()
    parser.read(filename)

    # get section, default to mysql
    octopart = {}
    if parser.has_section(section):
        items = parser.items(section)
        for item in items:
            octopart[item[0]] = item[1]
    else:
        raise Exception(
            '{0} not found in the {1} file'.format(section, filename))

    return octopart


def read_distributors_config(filename='config.ini', section='distributors'):
    """ Read database configuration file and return a dictionary object.

    :param filename: name of the configuration file
    :param section: section of database configuration
    :return: a dictionary of database parameters
    """
    # create parser and read ini configuration file
    parser = ConfigParser()
    parser.read(filename)

    # get section, default to mysql
    distributors = {}
    if parser.has_section(section):
        items = parser.items(section)
        for item in items:
            distributors[item[0]] = item[1]
    else:
        raise Exception(
            '{0} not found in the {1} file'.format(section, filename))

    return distributors


def read_currency_config(filename='config.ini', section='currency'):
    """ Read database configuration file and return a dictionary object.

    :param filename: name of the configuration file
    :param section: section of database configuration
    :return: a dictionary of database parameters
    """
    # create parser and read ini configuration file
    parser = ConfigParser()
    parser.read(filename)

    # get section, default to mysql
    currency = {}
    if parser.has_section(section):
        items = parser.items(section)
        for item in items:
            currency[item[0]] = item[1]
    else:
        raise Exception(
            '{0} not found in the {1} file'.format(section, filename))

    return currency