def _parse_EVAL_tag(self, node: Node):
args, kw = self._parse_options(node)
if self.context.get("ALL_FLOATS"):
self.flags["floats"] = True
for arg in args:
if arg.isdigit():
self.flags["round"] = int(arg)
elif arg == ".":
self.flags["."] = True
elif arg in ("floats", "float"):
self.flags["floats"] = True
elif arg == "str":
self.flags["str"] = True
else:
raise ValueError("Unknown flag: " + repr(arg))
# XXX: support options round, float, (sympy, python,) select and rand
code = node.arg(0)
assert isinstance(code, str), type(code)
try:
txt = self._eval_and_format_python_expr(code)
except Exception:
print("ERROR: Can't evaluate this: " + repr(code))
raise
# Tags must be cleared *before* calling .write(txt), since .write(txt)
# add '+', '-' and '\times ' before txt if corresponding flags are set,
# and ._eval_and_format_python_expr() has already do this.
self.flags.clear()
self.write(txt)
def _parse_TEST_tag(self, node: Node) -> None:
try:
if eval(node.arg(0), self.context):
self._parse_children(node.children[1].children)
else:
self._parse_children(node.children[2].children)
except Exception:
print(node.display(color=False))
raise
def _parse_ASSERT_tag(self, node: Node):
code = node.arg(0)
test = eval(code, self.context)
if not test:
print("Error in assertion (NUM=%s):" % self.NUM)
print("***")
print(code)
print("***")
assert test
def _parse_PRINT_tag(self, node: Node) -> None:
print(node.arg(0))
def _parse_CALL_tag(self, node: Node):
"""Calling a macro."""
name = node.arg(0)
if name not in self.macros:
raise NameError(f"Error: MACRO {name!r} undefined.")
self._parse_children(self.macros[name])
def _parse_MACRO_tag(self, node: Node):
name = node.arg(0)
self.macros[name] = node.children[1:]
def _parse_IMPORT_tag(self, node: Node):
exec("from %s import *" % node.arg(0), self.context)
def _parse_CASE_tag(self, node: Node):
test = eval(node.arg(0), self.context) == self.NUM
if test:
self._parse_children(node.children[1:])
return test
def _parse_IFNUM_tag(self, node: Node):
if eval(node.arg(0), self.context) == self.NUM:
self._parse_children(node.children[1].children)