Как загрузить файл конфигурации в Python и получить доступ к значениям с помощью точечной нотации (атрибутов)? - PullRequest
Новая
       14

Как загрузить файл конфигурации в Python и получить доступ к значениям с помощью точечной нотации (атрибутов)?

2 голосов
/ 19 марта 2012

Я ищу решение, которое позволило бы мне использовать атрибуты для данных файла конфигурации.

Я хотел бы иметь возможность сделать что-то вроде этого: 

config = Config('config.ini')
print config.section1.user
print config.section2.password

Iя знаю, что ConfigParser позволил бы мне сделать что-то вроде config['section1']['user'], но это слишком уродливо, разве мы не можем добиться большего успеха?

Решение должно работать с Python 2.5 или новее.

Ответы [ 2 ]

2 голосов
/ 19 марта 2012

Это не уродливо, это лучше - точечная запись означает, что, возможно, существует некоторый пользовательский класс «объект внутри другого пользовательского класса».Более выполнимый способ - использовать словари (использующие скобочные обозначения).

Но если вы настаиваете, вы, вероятно, можете перевести код следующим образом: 

def config2object(config):
    """
    Convert dictionary into instance allowing access to dictionary keys using
    dot notation (attributes).
    """
    class ConfigObject(dict):
        """
        Represents configuration options' group, works like a dict
        """
        def __init__(self, *args, **kwargs):
            dict.__init__(self, *args, **kwargs)
        def __getattr__(self, name):
            return self[name]
        def __setattr__(self, name, val):
            self[name] = val
    if isinstance(config, dict):
        result = ConfigObject()
        for key in config:
            result[key] = config2object(config[key])
        return result
    else:
        return config

И тесты показывают ожидаемоерезультаты: 

>>> c1 = {
    'conf1': {
        'key1': 'aaa',
        'key2': 12321,
        'key3': False,
        },
    'conf2': 'bbbb',
    }
>>> c1
{'conf2': 'bbbb', 'conf1': {'key3': False, 'key2': 12321, 'key1': 'aaa'}}
>>> c2 = config2object(c1)
>>> c2.conf1
{'key3': False, 'key2': 12321, 'key1': 'aaa'}
>>> c2.conf1.key1
'aaa'
>>> c2.conf1.key3
False
>>> c2.conf2
'bbbb'

РЕДАКТИРОВАТЬ : Свен Марнах отметил, что Config('config.ini') - это некоторый пользовательский экземпляр класса.Это не словарь, у него есть некоторые пользовательские методы, которые могут быть весьма полезны, но могут сделать некоторые параметры конфигурации недоступными (при конфликте имен).Поэтому предпочтительный подход состоит не в том, чтобы использовать решение, которое я упомянул, а в использовании скобочной записи для доступа к параметрам конфигурации.

1 голос
/ 24 мая 2015

Я тоже хотел использовать точечную нотацию для доступа к атрибутам, считываемым из файлов конфигурации ConfigParser. ( Доступно на github ).

Вот моя попытка расширить ConfigParser: 

from ConfigParser import ConfigParser as BaseClass

SPACE = " "
UNDERSCORE = "_"


def internal_name(name, needle=SPACE, replacement=UNDERSCORE):
    return name.replace(needle, replacement)


def reverse_name_internalization(name):
    return internal_name(name, needle=UNDERSCORE, replacement=SPACE)


class DotNotationConfigParser(BaseClass, object):

    def __init__(self, coersion_map=None, *args, **kwargs):
        super(DotNotationConfigParser, self).__init__(*args, **kwargs)

        self.optionxform = internal_name
        self.section_attr = None

    def get_internalized_section(self, section):
        if self.has_section(section):
            return internal_name(section)

    def __set_section_obj(self, internalized_section):
        if self.has_section(internalized_section):
            section = internalized_section
        else:
            section = reverse_name_internalization(internalized_section)

        if self.get_internalized_section(section):
            # set an attr to an object instance with section items
            obj = type('', (), dict(self.items(section)))()
            setattr(self, internalized_section, obj)

    def __getattr__(self, attr):
        try:
            return super(DotNotationConfigParser, self).__getattribute__(attr)
        except AttributeError:
            section = attr
            self.__set_section_obj(section)
            return super(DotNotationConfigParser, self).__getattribute__(attr)


try:
    from cStringIO import StringIO
except ImportError:
    from StringIO import StringIO
configuration_file = """
[section 1]
foo = the foo value
bar = the bar value
[section 2]
index = 3
repeat = False
[section_n]
string = This is some text.
ip = 10.0.1.1
"""
configuration_file = StringIO(configuration_file)

parser = DotNotationConfigParser()
parser.readfp(configuration_file)

assert parser.section_1.foo == 'the foo value'
assert parser.section_1.bar == 'the bar value'
assert type(parser.section_2.index) is not int
for section_name in ('section_1', 'section_2', 'section_n'):
    section = getattr(parser, section_name)
    options = [option for option in dir(section)
               if not option.startswith('__')]
    for option in options:
        print section_name, ": ", getattr(section, option)

print "dot notation", parser.section_1.foo
...