Skip to content
Snippets Groups Projects
Commit 2cb30411 authored by Bert Palm's avatar Bert Palm 🎇
Browse files

review done, minor changes

parent 3dffe2cf
No related branches found
No related tags found
1 merge request!17Transformer bugfix
Pipeline #2899 passed with stage
in 7 minutes and 25 seconds
......@@ -47,7 +47,7 @@ class DslChecker(ast.NodeVisitor):
def visit_Name(self, node):
name = node.id
if name != "this" and name not in self.environment and name not in self.environment["variables"]:
if name not in self.environment and name not in self.environment["variables"]:
raise NameError(f"unknown variable: '{name}'")
self.generic_visit(node)
......
......@@ -8,10 +8,10 @@ from contextlib import contextmanager
from saqc.core.config import Params
class DslTransformer(ast.NodeTransformer):
def __init__(self, environment: Dict[str, Any]):
self.environment = environment
self.arguments = set()
def visit_Call(self, node):
new_args = node.args
......@@ -47,9 +47,10 @@ class DslTransformer(ast.NodeTransformer):
name = node.id
if name == "this":
name = self.environment["field"]
name = self.environment["this"]
if name in self.environment["data"]:
# determine further tree-transformation path by target
if getattr(node, "lookup", True):
value = ast.Constant(value=name)
node = ast.Subscript(
......@@ -58,7 +59,6 @@ class DslTransformer(ast.NodeTransformer):
else:
node = ast.Constant(value=name)
self.arguments.add(name)
return node
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment