1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
|
# -*- coding: utf8 -*-
"""
kaptan.handlers.ini_handler
~~~~~~~~~~~~~~~~~~~~~~~~~~~
:copyright: (c) 2013 by the authors and contributors (See AUTHORS file).
:license: BSD, see LICENSE for more details.
"""
from __future__ import print_function, unicode_literals
try:
import ConfigParser as configparser
from StringIO import StringIO
configparser.RawConfigParser.read_file = configparser.RawConfigParser.readfp # NOQA
except ImportError: # Python 3
import configparser
from io import StringIO
from . import BaseHandler
class KaptanIniParser(configparser.RawConfigParser):
def from_dict(self, dictionary):
self._sections = dictionary
def as_dict(self):
d = dict(self._sections)
for k in d:
d[k] = dict(self._defaults, **d[k])
d[k].pop('__name__', None)
return d
class IniHandler(BaseHandler):
def load(self, value):
config = KaptanIniParser()
# ConfigParser.ConfigParser wants to read value as file / IO
config.read_file(StringIO(value))
return config.as_dict()
def dump(self, data, file_=None):
if file_ is None:
raise NotImplementedError("Exporting .ini as string is not supported.")
config = KaptanIniParser()
config.from_dict(data)
with open(file_, 'w') as fp:
config.write(fp)
|
