jw-python/tools/python/jwutils/stree/serdes.py
Jan Lindemann 838ac06ebb Fix: StringTree misparsed key = "val # blah"
Signed-off-by: Jan Lindemann <jan@janware.com>
2017-11-26 21:25:01 +01:00

68 lines
1.9 KiB
Python

from StringTree import *
from jwutils.log import *
def _cleanup_line(line):
line = line.strip()
r = ''
in_quote = None
for c in line:
# slog(DEBUG, "in_quote = >" + str(in_quote) + "<")
if in_quote is not None:
if c == in_quote:
in_quote = None
else:
if c in [ '"', "'" ]:
in_quote = c
elif in_quote is None and c == '#':
return r.strip()
r += c
if len(r) >= 2 and r[0] in [ '"', "'" ] and r[-1] == r[0]:
return r[1:-1]
return r
def parse(s): # export
slog(DEBUG, "parsing", s)
root = StringTree('', content='root')
sec = ''
for line in s.splitlines():
slog(DEBUG, "line=", line)
line = _cleanup_line(line)
#slog(DEBUG, "cleaned line=", line)
if not len(line):
continue
if line[0] == '[':
if line[-1] == ']':
sec = line[1:-1]
elif line[-1] == '[':
if len(sec):
sec += '.'
sec += line[1:-1]
else:
raise Exception("failed to parse section line", line)
if root.get(sec) is None:
root.add(sec)
continue
elif line[0] == ']':
assert(len(sec) > 0)
sec = '.'.join(sec.split('.')[0:-1])
continue
lhs = ''
rhs = None
for c in line:
if rhs is None:
if c == '=':
rhs = ''
continue
lhs += c
continue
rhs += c
if rhs is None:
raise Exception("failed to parse assignment", line)
root.add(sec + '.' + cleanup_string(lhs), cleanup_string(rhs))
return root
def read(path): # export
with open(path, 'r') as infile:
s = infile.read()
return parse(s)