45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
import textwrap
|
|
import os
|
|
from os import path
|
|
import json
|
|
import importlib
|
|
from lark import Lark
|
|
|
|
from agtest.ast import *
|
|
from agtest.gen import *
|
|
|
|
grammar = path.join(path.dirname(path.realpath(__file__)), "grammar.lark")
|
|
p = Lark(open(grammar).read(), start="program", parser="lalr")
|
|
|
|
if __name__ == "__main__":
|
|
with open("test/arith.ag") as f:
|
|
data = f.read()
|
|
|
|
cst = p.parse(data)
|
|
|
|
trans = Parser()
|
|
ast = trans.transform(cst)
|
|
|
|
res = gen(ast)
|
|
print("Grammar:")
|
|
print(res.parser_data)
|
|
|
|
if not os.path.exists("gen"):
|
|
os.makedirs("gen")
|
|
|
|
with open("agruntime.tmpl.py", "r") as f:
|
|
fmt_str = f.read()
|
|
|
|
with open("gen/arith.py", "w") as f:
|
|
f.write(
|
|
fmt_str.format(
|
|
pd=res.parser_data,
|
|
ex=res.extra,
|
|
starts=list(res.starts),
|
|
transdef=res.transdef,
|
|
ntmap=json.dumps(res.nonterminal_map),
|
|
)
|
|
)
|
|
|
|
mod = importlib.import_module("gen.arith")
|
|
print(mod.parse("1 + 2 * 3", start="Expr")) # type: ignore
|