mirror of
https://github.com/nzp-team/tools.git
synced 2024-11-10 06:31:34 +00:00
258 lines
9.5 KiB
Python
258 lines
9.5 KiB
Python
# vim:ts=4:et
|
|
# ##### BEGIN GPL LICENSE BLOCK #####
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; either version 2
|
|
# of the License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software Foundation,
|
|
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
# ##### END GPL LICENSE BLOCK #####
|
|
|
|
# <pep8 compliant>
|
|
|
|
quotables = ("0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
+ "abcdefghijklmnopqrstuvwxyz!#$%&*+-./:?@|~_^")
|
|
|
|
class PListError(Exception):
|
|
def __init__(self, line, message):
|
|
Exception.__init__(self, "%d: %s" % (line, message))
|
|
self.line = line
|
|
|
|
class pldata:
|
|
def __init__(self, src = ''):
|
|
self.src = src
|
|
self.pos = 0;
|
|
self.end = len(self.src)
|
|
self.line = 1
|
|
def skip_space(self):
|
|
while self.pos < self.end:
|
|
c = self.src[self.pos]
|
|
if not c.isspace():
|
|
if c == '/' and self.pos < self.end - 1: #comments
|
|
if self.src[self.pos + 1] == '/': # // coment
|
|
self.pos += 2
|
|
while self.pos < self.end:
|
|
c = self.src[self.pos]
|
|
if c == '\n':
|
|
break
|
|
self.pos += 1
|
|
if self.pos >= self.end:
|
|
raise PListError(self.line,
|
|
"Reached end of string in comment")
|
|
elif self.src[self.pos + 1] == '*': # /* comment */
|
|
start_line = self.line
|
|
self.pos += 2
|
|
while self.pos < self.end:
|
|
c = self.src[self.pos]
|
|
if c == '\n':
|
|
self.line += 1
|
|
elif (c == '*' and self.pos < self.end - 1
|
|
and self.src[self.pos + 1] == '/'):
|
|
self.pos += 1
|
|
break
|
|
self.pos += 1
|
|
if self.pos >= self.end:
|
|
raise PListError(start_line,
|
|
"Reached end of string in comment")
|
|
else:
|
|
return True
|
|
else:
|
|
return True
|
|
if c == '\n':
|
|
self.line += 1
|
|
self.pos += 1
|
|
raise PListError(self.line, "Reached end of string")
|
|
def parse_quoted_string(self):
|
|
start_line = self.line
|
|
long_string = False
|
|
escaped = 0
|
|
shrink = 0
|
|
hexa = False
|
|
self.pos += 1
|
|
start = self.pos
|
|
if (self.pos < self.end - 1 and self.src[self.pos] == '"'
|
|
and self.src[self.pos + 1] == '"'):
|
|
self.pos += 2
|
|
long_string = True
|
|
start += 2
|
|
while self.pos < self.end:
|
|
c = self.src[self.pos]
|
|
if escaped:
|
|
if escaped == 1 and c == '0':
|
|
escaped += 1
|
|
hexa = False
|
|
elif escaped > 1:
|
|
if escaped == 2 and c == 'x':
|
|
hexa = True
|
|
shring += 1
|
|
escaped += 1
|
|
elif hex and c.isxdigit():
|
|
shrink += 1
|
|
escaped += 1
|
|
elif c in range(0, 8):
|
|
shrink += 1
|
|
escaped += 1
|
|
else:
|
|
self.pos -= 1
|
|
escaped = 0
|
|
else:
|
|
escaped = 0
|
|
else:
|
|
if c == '\\':
|
|
escaped = 1
|
|
shrink += 1
|
|
elif (c == '"'
|
|
and (not long_string
|
|
or (self.pos < self.end - 2
|
|
and self.src[self.pos + 1] == '"'
|
|
and self.src[self.pos + 2] == '"'))):
|
|
break
|
|
if c == '\n':
|
|
self.line += 1
|
|
self.pos += 1
|
|
if self.pos >= self.end:
|
|
raise PListError(start_line,
|
|
"Reached end of string while parsing quoted string")
|
|
if self.pos - start - shrink == 0:
|
|
return ""
|
|
s = self.src[start:self.pos]
|
|
self.pos += 1
|
|
if long_string:
|
|
self.pos += 2
|
|
return eval('"""' + s + '"""')
|
|
def parse_unquoted_string(self):
|
|
start = self.pos
|
|
while self.pos < self.end:
|
|
if self.src[self.pos] not in quotables:
|
|
break
|
|
self.pos += 1
|
|
return self.src[start:self.pos]
|
|
def parse_data(self):
|
|
start_line = self.line
|
|
self.pos += 1
|
|
start = self.pos
|
|
nibbles = 0
|
|
while self.pos < self.end:
|
|
if self.src[self.pos].isxdigit:
|
|
nibbles += 1
|
|
self.pos += 1
|
|
continue
|
|
if self.src[self.pos] == '>':
|
|
if nibbles & 1:
|
|
raise PListError(self.line,
|
|
"Invalid data, missing nibble")
|
|
s = self.src[start:self.pos]
|
|
self.pos += 1
|
|
return binascii.a2b_hex(s)
|
|
raise PListError(self.line,
|
|
"Invalid character in data")
|
|
raise PListError(start_line,
|
|
"Reached end of string while parsing data")
|
|
def parse(self):
|
|
self.skip_space()
|
|
if self.src[self.pos] == '{':
|
|
item = {}
|
|
self.pos += 1
|
|
while self.skip_space() and self.src[self.pos] != '}':
|
|
key = self.parse()
|
|
if type(key) != str:
|
|
raise PListError(self.line,
|
|
"Key is not a string")
|
|
self.skip_space()
|
|
if self.src[self.pos] != '=':
|
|
raise PListError(self.line,
|
|
"Unexpected character (expected '=')")
|
|
self.pos += 1
|
|
value = self.parse()
|
|
if self.src[self.pos] == ';':
|
|
self.pos += 1
|
|
elif self.src[self.pos] != '}':
|
|
raise PListError(self.line,
|
|
"Unexpected character (wanted ';' or '}')")
|
|
item[key] = value
|
|
if self.pos >= self.end:
|
|
raise PListError(self.line,
|
|
"Unexpected end of string when parsing dictionary")
|
|
self.pos += 1
|
|
return item
|
|
elif self.src[self.pos] == '(':
|
|
item = []
|
|
self.pos += 1
|
|
while self.skip_space() and self.src[self.pos] != ')':
|
|
value = self.parse()
|
|
self.skip_space()
|
|
if self.src[self.pos] == ',':
|
|
self.pos += 1
|
|
elif self.src[self.pos] != ')':
|
|
raise PListError(self.line,
|
|
"Unexpected character (wanted ',' or ')')")
|
|
item.append(value)
|
|
self.pos += 1
|
|
return item
|
|
elif self.src[self.pos] == '<':
|
|
return self.parse_data()
|
|
elif self.src[self.pos] == '"':
|
|
return self.parse_quoted_string()
|
|
else:
|
|
return self.parse_unquoted_string()
|
|
def write_string(self, item):
|
|
quote = False
|
|
for i in item:
|
|
if i not in quotables:
|
|
quote = True
|
|
break
|
|
if quote:
|
|
item = repr(item)
|
|
# repr uses ', we want "
|
|
item = '"' + item[1:-1].replace('"', '\\"') + '"'
|
|
self.data.append(item)
|
|
def write_item(self, item, level):
|
|
if type(item) == dict:
|
|
if not item:
|
|
self.data.append("{ }")
|
|
return
|
|
self.data.append("{\n")
|
|
for i in item.items():
|
|
self.data.append('\t' * (level + 1))
|
|
self.write_string(i[0])
|
|
self.data.append(' = ')
|
|
self.write_item(i[1], level + 1)
|
|
self.data.append(';\n')
|
|
self.data.append('\t' * (level))
|
|
self.data.append("}")
|
|
elif type(item) in(list, tuple):
|
|
if not item:
|
|
self.data.append("( )")
|
|
return
|
|
self.data.append("(\n")
|
|
for n, i in enumerate(item):
|
|
self.data.append('\t' * (level + 1))
|
|
self.write_item(i, level + 1)
|
|
if n < len(item) - 1:
|
|
self.data.append(',\n')
|
|
self.data.append('\n')
|
|
self.data.append('\t' * (level))
|
|
self.data.append(")")
|
|
elif type(item) == bytes:
|
|
self.data.append('<')
|
|
self.data.append(binascii.b2a_hex(item))
|
|
self.data.append('>')
|
|
elif type(item) == str:
|
|
self.write_string(item)
|
|
elif type(item) in [int, float]:
|
|
self.write_string(str(item))
|
|
else:
|
|
raise PListError(0, "unsupported type")
|
|
def write(self, item):
|
|
self.data = []
|
|
self.write_item(item, 0)
|
|
return ''.join(self.data)
|