143 lines
3.3 KiB
Python
143 lines
3.3 KiB
Python
"""
|
|
|
|
"""
|
|
|
|
from functools import partial
|
|
import lark
|
|
from utils import utils, constants as c
|
|
|
|
|
|
class ListingTransformer(lark.Transformer):
|
|
"""
|
|
Transforms content parsed by grammar_listing.lark further.
|
|
Used for the developer and inspirations list.
|
|
"""
|
|
|
|
def unquoted_value(self, x):
|
|
return x[0].strip()
|
|
|
|
def quoted_value(self, x):
|
|
return x[0][1:-1].strip() # remove quotation marks and strip whitespaces
|
|
|
|
def property(self, x):
|
|
"""
|
|
Key is first part, values are following.
|
|
:param x:
|
|
:return:
|
|
"""
|
|
return x[0], x[1:]
|
|
|
|
def name(self, x):
|
|
"""
|
|
The name part is treated as a property with key "Name"
|
|
:param x:
|
|
:return:
|
|
"""
|
|
return 'Name', x[0].strip()
|
|
|
|
def entry(self, x):
|
|
"""
|
|
All (key, value) tuples are inserted into a dictionary.
|
|
:param x:
|
|
:return:
|
|
"""
|
|
d = {}
|
|
for key, value in x:
|
|
if key in d:
|
|
raise RuntimeError('Key in entry appears twice')
|
|
d[key] = value
|
|
return d
|
|
|
|
def start(self, x):
|
|
return x
|
|
|
|
|
|
# transformer
|
|
class EntryTransformer(lark.Transformer):
|
|
|
|
def unquoted_value(self, x):
|
|
return x[0].strip()
|
|
|
|
def quoted_value(self, x):
|
|
return x[0][1:-1].strip() # remove quotation marks
|
|
|
|
def comment_value(self, x):
|
|
return x[0][1:-1].strip() # remove parenthesis
|
|
|
|
def value(self, x):
|
|
"""
|
|
This also stores the comment if needed.
|
|
|
|
:param x:
|
|
:return:
|
|
"""
|
|
if len(x) == 1:
|
|
v = x[0]
|
|
else:
|
|
v = Value(*x)
|
|
return v
|
|
|
|
def property(self, x):
|
|
"""
|
|
The key of a property will be converted to lower case and the value part is the second part
|
|
:param x:
|
|
:return:
|
|
"""
|
|
return x[0].strip(), x[1:]
|
|
|
|
def title(self, x):
|
|
return 'Title', x[0].strip()
|
|
|
|
def note(self, x):
|
|
"""
|
|
Optional
|
|
:param x:
|
|
:return:
|
|
"""
|
|
if not x:
|
|
raise lark.Discard
|
|
return 'Note', ''.join(x)
|
|
|
|
def building(self, x):
|
|
return 'Building', x
|
|
|
|
def start(self, x):
|
|
return x
|
|
|
|
|
|
class Value(str):
|
|
"""
|
|
A value is a string with some additional meta object (a comment) but mostly behaves as a string.
|
|
"""
|
|
|
|
def __new__(cls, value, comment):
|
|
obj = str.__new__(cls, value)
|
|
obj.comment = comment
|
|
return obj
|
|
|
|
def parse(parser, transformer, content):
|
|
tree = parser.parse(content)
|
|
value = transformer.transform(tree)
|
|
return value
|
|
|
|
|
|
def create(grammar, Transformer):
|
|
parser = lark.Lark(grammar, debug=False, parser='lalr')
|
|
transformer = Transformer()
|
|
return partial(parse, parser, transformer)
|
|
|
|
|
|
def read_and_parse(content_file: str, grammar_file: str, Transformer: lark.Transformer):
|
|
"""
|
|
Reads a content file and a grammar file and parses the content with the grammar following by
|
|
transforming the parsed output and returning the transformed result.
|
|
:param content_file:
|
|
:param grammar_file:
|
|
:param transformer:
|
|
:return:
|
|
"""
|
|
grammar = utils.read_text(grammar_file)
|
|
parse = create(grammar, Transformer)
|
|
|
|
content = utils.read_text(content_file)
|
|
return parse(content) |