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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
from typing import Iterable
from collections import OrderedDict
__all__ = (
"KconfigFile",
)
class KConfigEntry(object):
__slots__ = 'name', 'value', 'comments'
def __init__(self, name, value, comments=None) -> None:
self.name, self.value = name, value
self.comments = comments or []
def __eq__(self, other) -> bool:
return self.name == other.name and self.value == other.value
def __hash__(self) -> int:
return hash(self.name) | hash(self.value)
def __repr__(self) -> str:
return ('<{}({!r}, {!r}, {!r})>'
.format(self.__class__.__name__, self.name, self.value,
self.comments))
def __str__(self) -> str:
return 'CONFIG_{}={}'.format(self.name, self.value)
def write(self) -> Iterable[str]:
for comment in self.comments:
yield '#. ' + comment
yield str(self)
class KConfigEntryTristate(KConfigEntry):
__slots__ = ()
VALUE_NO = False
VALUE_YES = True
VALUE_MOD = object()
def __init__(self, name, value, comments=None) -> None:
if value == 'n' or value is None:
value = self.VALUE_NO
elif value == 'y':
value = self.VALUE_YES
elif value == 'm':
value = self.VALUE_MOD
else:
raise NotImplementedError
super(KConfigEntryTristate, self).__init__(name, value, comments)
def __str__(self) -> str:
if self.value is self.VALUE_MOD:
return 'CONFIG_{}=m'.format(self.name)
if self.value:
return 'CONFIG_{}=y'.format(self.name)
return '# CONFIG_{} is not set'.format(self.name)
class KconfigFile(OrderedDict[str, KConfigEntry]):
def __str__(self) -> str:
ret = []
for i in self.str_iter():
ret.append(i)
return '\n'.join(ret) + '\n'
def read(self, f) -> None:
for line in iter(f.readlines()):
line = line.strip()
if line.startswith("CONFIG_"):
i = line.find('=')
option = line[7:i]
value = line[i + 1:]
self.set(option, value)
elif line.startswith("# CONFIG_"):
option = line[9:-11]
self.set(option, 'n')
elif line.startswith("#") or not line:
pass
else:
raise RuntimeError("Can't recognize %s" % line)
def set(self, key, value) -> None:
if value in ('y', 'm', 'n'):
self[key] = KConfigEntryTristate(key, value)
else:
self[key] = KConfigEntry(key, value)
def str_iter(self) -> Iterable[str]:
for key, value in self.items():
yield str(value)
|
