|
1
|
|
|
# |
|
2
|
|
|
# Copyright (C) 2011 - 2017 Satoru SATOH <ssato @ redhat.com> |
|
3
|
|
|
# License: MIT |
|
4
|
|
|
# |
|
5
|
|
|
# pylint: disable=unused-argument |
|
6
|
|
|
"""INI or INI like config files backend. |
|
7
|
|
|
|
|
8
|
|
|
- Format to support: INI or INI like ones |
|
9
|
|
|
- Requirements: It should be available always. |
|
10
|
|
|
|
|
11
|
|
|
- ConfigParser in python 2 standard library: |
|
12
|
|
|
https://docs.python.org/2.7/library/configparser.html |
|
13
|
|
|
|
|
14
|
|
|
- configparser in python 3 standard library: |
|
15
|
|
|
https://docs.python.org/3/library/configparser.html |
|
16
|
|
|
|
|
17
|
|
|
- Development Status :: 4 - Beta |
|
18
|
|
|
- Limitations: It cannot process nested configuration dicts correctly due to |
|
19
|
|
|
the limitation of configparser module. |
|
20
|
|
|
|
|
21
|
|
|
- Special options: |
|
22
|
|
|
|
|
23
|
|
|
- Use 'ac_parse_value' boolean keyword option if you want to parse values by |
|
24
|
|
|
custom parser, anyconfig.backend.ini._parse. |
|
25
|
|
|
|
|
26
|
|
|
History: |
|
27
|
|
|
|
|
28
|
|
|
.. versionchanged:: 0.3 |
|
29
|
|
|
Introduce 'ac_parse_value' keyword option to switch behaviors, same as |
|
30
|
|
|
original configparser and rich backend parsing each parameter values. |
|
31
|
|
|
""" |
|
32
|
|
|
from __future__ import absolute_import |
|
33
|
|
|
|
|
34
|
|
|
import anyconfig.backend.base |
|
35
|
|
|
import anyconfig.parser as P |
|
36
|
|
|
import anyconfig.utils |
|
37
|
|
|
|
|
38
|
|
|
from anyconfig.compat import configparser, iteritems, OrderedDict |
|
39
|
|
|
from anyconfig.backend.base import mk_opt_args |
|
40
|
|
|
|
|
41
|
|
|
|
|
42
|
|
|
_SEP = ',' |
|
43
|
|
|
|
|
44
|
|
|
|
|
45
|
|
|
def _noop(val, *args, **kwargs): |
|
46
|
|
|
""" |
|
47
|
|
|
Parser does nothing. |
|
48
|
|
|
""" |
|
49
|
|
|
# It means nothing but can suppress 'Unused argument' pylint warns. |
|
50
|
|
|
# (val, args, kwargs)[0] |
|
51
|
|
|
return val |
|
52
|
|
|
|
|
53
|
|
|
|
|
54
|
|
|
def _parse(val_s, sep=_SEP): |
|
55
|
|
|
""" |
|
56
|
|
|
FIXME: May be too naive implementation. |
|
57
|
|
|
|
|
58
|
|
|
:param val_s: A string represents some value to parse |
|
59
|
|
|
:param sep: separator between values |
|
60
|
|
|
|
|
61
|
|
|
>>> _parse(r'"foo string"') |
|
62
|
|
|
'foo string' |
|
63
|
|
|
>>> _parse("a, b, c") |
|
64
|
|
|
['a', 'b', 'c'] |
|
65
|
|
|
>>> _parse("aaa") |
|
66
|
|
|
'aaa' |
|
67
|
|
|
""" |
|
68
|
|
|
if (val_s.startswith('"') and val_s.endswith('"')) or \ |
|
69
|
|
|
(val_s.startswith("'") and val_s.endswith("'")): |
|
70
|
|
|
return val_s[1:-1] |
|
71
|
|
|
elif sep in val_s: |
|
72
|
|
|
return [P.parse(x) for x in P.parse_list(val_s)] |
|
73
|
|
|
else: |
|
74
|
|
|
return P.parse(val_s) |
|
75
|
|
|
|
|
76
|
|
|
|
|
77
|
|
|
def _to_s(val, sep=", "): |
|
78
|
|
|
"""Convert any to string. |
|
79
|
|
|
|
|
80
|
|
|
:param val: An object |
|
81
|
|
|
:param sep: separator between values |
|
82
|
|
|
|
|
83
|
|
|
>>> _to_s([1, 2, 3]) |
|
84
|
|
|
'1, 2, 3' |
|
85
|
|
|
>>> _to_s("aaa") |
|
86
|
|
|
'aaa' |
|
87
|
|
|
""" |
|
88
|
|
|
if anyconfig.utils.is_iterable(val): |
|
89
|
|
|
return sep.join(str(x) for x in val) |
|
90
|
|
|
else: |
|
91
|
|
|
return str(val) |
|
92
|
|
|
|
|
93
|
|
|
|
|
94
|
|
|
def _make_parser(to_container, **kwargs): |
|
95
|
|
|
""" |
|
96
|
|
|
:param to_container: any callable to make container |
|
97
|
|
|
:return: (to_container, keyword args to be used, parser object) |
|
98
|
|
|
""" |
|
99
|
|
|
if kwargs.get("ac_ordered", False) or kwargs.get("dict_type", False): |
|
100
|
|
|
kwargs["dict_type"] = to_container = OrderedDict |
|
101
|
|
|
if "dict_type" not in kwargs and not kwargs.get("ac_ordered", True): |
|
102
|
|
|
kwargs["dict_type"] = to_container |
|
103
|
|
|
|
|
104
|
|
|
# Optional arguements for configparser.SafeConfigParser{,readfp} |
|
105
|
|
|
kwargs_0 = mk_opt_args(("defaults", "dict_type", "allow_no_value"), kwargs) |
|
106
|
|
|
kwargs_1 = mk_opt_args(("filename", ), kwargs) |
|
107
|
|
|
|
|
108
|
|
|
try: |
|
109
|
|
|
parser = configparser.SafeConfigParser(**kwargs_0) |
|
110
|
|
|
except TypeError: |
|
111
|
|
|
# .. note:: |
|
112
|
|
|
# It seems ConfigPaser.*ConfigParser in python 2.6 does not support |
|
113
|
|
|
# 'allow_no_value' option parameter, and TypeError will be thrown. |
|
114
|
|
|
kwargs_0 = mk_opt_args(("defaults", "dict_type"), kwargs) |
|
115
|
|
|
parser = configparser.SafeConfigParser(**kwargs_0) |
|
116
|
|
|
|
|
117
|
|
|
return (to_container, kwargs_1, parser) |
|
118
|
|
|
|
|
119
|
|
|
|
|
120
|
|
|
def _load(stream, to_container=dict, sep=_SEP, **kwargs): |
|
121
|
|
|
""" |
|
122
|
|
|
:param stream: File or file-like object provides ini-style conf |
|
123
|
|
|
:param to_container: any callable to make container |
|
124
|
|
|
:param sep: Seprator string |
|
125
|
|
|
|
|
126
|
|
|
:return: Dict or dict-like object represents config values |
|
127
|
|
|
""" |
|
128
|
|
|
(to_container, kwargs_1, parser) = _make_parser(to_container, **kwargs) |
|
129
|
|
|
_parse_val = _parse if kwargs.get("ac_parse_value", False) else _noop |
|
130
|
|
|
|
|
131
|
|
|
cnf = to_container() |
|
132
|
|
|
parser.readfp(stream, **kwargs_1) |
|
133
|
|
|
|
|
134
|
|
|
# .. note:: Process DEFAULT config parameters as special ones. |
|
135
|
|
|
defaults = parser.defaults() |
|
136
|
|
|
if defaults: |
|
137
|
|
|
cnf["DEFAULT"] = to_container() |
|
138
|
|
|
for key, val in iteritems(defaults): |
|
139
|
|
|
cnf["DEFAULT"][key] = _parse_val(val, sep) |
|
140
|
|
|
|
|
141
|
|
|
for sect in parser.sections(): |
|
142
|
|
|
cnf[sect] = to_container() |
|
143
|
|
|
for key, val in parser.items(sect): |
|
144
|
|
|
cnf[sect][key] = _parse_val(val, sep) |
|
145
|
|
|
|
|
146
|
|
|
return cnf |
|
147
|
|
|
|
|
148
|
|
|
|
|
149
|
|
|
def _dumps_itr(cnf): |
|
150
|
|
|
""" |
|
151
|
|
|
:param cnf: Configuration data to dump |
|
152
|
|
|
""" |
|
153
|
|
|
dkey = "DEFAULT" |
|
154
|
|
|
for sect, params in iteritems(cnf): |
|
155
|
|
|
yield "[%s]" % sect |
|
156
|
|
|
|
|
157
|
|
|
for key, val in iteritems(params): |
|
158
|
|
|
if sect != dkey and dkey in cnf and cnf[dkey].get(key) == val: |
|
159
|
|
|
continue # It should be in [DEFAULT] section. |
|
160
|
|
|
|
|
161
|
|
|
yield "%s = %s" % (key, _to_s(val)) |
|
162
|
|
|
|
|
163
|
|
|
yield '' # it will be a separator between each sections. |
|
164
|
|
|
|
|
165
|
|
|
|
|
166
|
|
|
def _dumps(cnf, **kwargs): |
|
167
|
|
|
""" |
|
168
|
|
|
:param cnf: Configuration data to dump |
|
169
|
|
|
:param kwargs: optional keyword parameters to be sanitized :: dict |
|
170
|
|
|
|
|
171
|
|
|
:return: String representation of `cnf` object in INI format |
|
172
|
|
|
""" |
|
173
|
|
|
return '\n'.join(l for l in _dumps_itr(cnf)) |
|
174
|
|
|
|
|
175
|
|
|
|
|
176
|
|
|
class Parser(anyconfig.backend.base.FromStreamLoader, |
|
177
|
|
|
anyconfig.backend.base.ToStringDumper): |
|
178
|
|
|
""" |
|
179
|
|
|
Ini config files parser. |
|
180
|
|
|
""" |
|
181
|
|
|
_type = "ini" |
|
182
|
|
|
_extensions = ["ini"] |
|
183
|
|
|
_load_opts = ["defaults", "dict_type", "allow_no_value", "filename", |
|
184
|
|
|
"ac_parse_value"] |
|
185
|
|
|
|
|
186
|
|
|
dump_to_string = anyconfig.backend.base.to_method(_dumps) |
|
187
|
|
|
|
|
188
|
|
|
def load_from_stream(self, stream, to_container, **options): |
|
189
|
|
|
""" |
|
190
|
|
|
Load config from given file like object `stream`. |
|
191
|
|
|
|
|
192
|
|
|
:param stream: Config file or file like object |
|
193
|
|
|
:param to_container: callble to make a container object |
|
194
|
|
|
:param options: optional keyword arguments |
|
195
|
|
|
|
|
196
|
|
|
:return: Dict-like object holding config parameters |
|
197
|
|
|
""" |
|
198
|
|
|
return _load(stream, to_container=to_container, **options) |
|
199
|
|
|
|
|
200
|
|
|
# vim:sw=4:ts=4:et: |
|
201
|
|
|
|