def subquery_fields(sub_root, sub_edge_name, exprs):
re_env = {}
exprs = {name: to_expr(obj, re_env) for name, obj in exprs.items()}
ec_env = {func.__fn_name__ for func in re_env.values()}
fn_env = {func.__fn_name__: func.fn for func in re_env.values()}
re_env['this'] = sub_root.fields[sub_edge_name]
re_env.update(sub_root.fields)
ec = ExpressionCompiler(ec_env)
reqs_map = {}
procs_map = {}
for name, expr in exprs.items():
reqs_map[name] = RequirementsExtractor.analyze(re_env, expr)
procs_map[name] = eval(compile(ec.compile_lambda_expr(expr),
'<expr>', 'eval'))
def query_func(queue, task_set, edge, fields, ids):
this_link = Link(THIS_LINK_NAME, None, sub_edge_name, None, True)
reqs = merge(reqs_map[f.name] for f in fields)
# FIXME: implement proper way to handle "this" value
# and query other possible data from sub_root
pattern = reqs.fields['this'].edge
procs = [procs_map[f.name] for f in fields]
query = Query(queue, task_set, sub_root, None)
query._process_link(sub_root, this_link, pattern, None, ids)
return _create_result_proc(query, fn_env, edge, fields, procs, ids)
query_func.__subquery__ = True
return [Field(name, query_func) for name in exprs.keys()]
def subquery_fields(sub_root, sub_edge_name, exprs):
re_env = {}
exprs = {name: to_expr(obj, re_env) for name, obj in exprs.items()}
ec_env = {func.__fn_name__ for func in re_env.values()}
fn_env = {func.__fn_name__: func.fn for func in re_env.values()}
re_env['this'] = sub_root.fields[sub_edge_name]
re_env.update(sub_root.fields)
ec = ExpressionCompiler(ec_env)
reqs_map = {}
procs_map = {}
for name, expr in exprs.items():
reqs_map[name] = RequirementsExtractor.analyze(re_env, expr)
procs_map[name] = eval(
compile(ec.compile_lambda_expr(expr), '<expr>', 'eval'))
def query_func(queue, task_set, edge, fields, ids):
this_link = Link(THIS_LINK_NAME, None, sub_edge_name, None, True)
reqs = merge(reqs_map[f.name] for f in fields)
# FIXME: implement proper way to handle "this" value
# and query other possible data from sub_root
pattern = reqs.fields['this'].edge
procs = [procs_map[f.name] for f in fields]
query = Query(queue, task_set, sub_root, None)
query._process_link(sub_root, this_link, pattern, None, ids)
return _create_result_proc(query, fn_env, edge, fields, procs, ids)
query_func.__subquery__ = True
return [Field(name, query_func) for name in exprs.keys()]
def assertRequires(self, expr, reqs):
fn_reg = {}
expr = to_expr(expr, fn_reg)
env = dict(_ENV, **{fn.__fn_name__: fn for fn in fn_reg.values()})
expr_reqs = RequirementsExtractor.analyze(env, expr)
with reqs_eq_patcher():
self.assertEqual(expr_reqs, reqs)
def assertCompiles(self, expr, code):
ec = ExpressionCompiler({'foo', 'baz'})
py_expr = ec.visit(to_expr(expr, {}))
first = astor.to_source(py_expr)
if not PY3:
first = first.replace("u'", "'")
second = dedent(code).strip()
if first != second:
msg = ('Compiled code is not equal:\n\n{}'.format('\n'.join(
difflib.ndiff(first.splitlines(), second.splitlines()))))
raise self.failureException(msg)
def assertCompiles(self, expr, code):
ec = ExpressionCompiler({'foo', 'baz'})
py_expr = ec.visit(to_expr(expr, {}))
first = astor.to_source(py_expr)
if not PY3:
first = first.replace("u'", "'")
second = dedent(code).strip()
if first != second:
msg = ('Compiled code is not equal:\n\n{}'
.format('\n'.join(difflib.ndiff(first.splitlines(),
second.splitlines()))))
raise self.failureException(msg)
def check(self, expr):
fn_reg = {}
expr = to_expr(expr, fn_reg)
env = dict(_ENV, **{fn.__fn_name__: fn for fn in fn_reg.values()})
return Checker(env).visit(expr)