From 1efd094c840b0ac66835988470fa21a0325b32da Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Fri, 15 Apr 2016 20:11:59 +0000 Subject: [PATCH] add excpetion_handler option to Graphcore.query --- graphcore/graphcore.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/graphcore/graphcore.py b/graphcore/graphcore.py index 8a1e1bb..1e3c1cd 100644 --- a/graphcore/graphcore.py +++ b/graphcore/graphcore.py @@ -7,6 +7,7 @@ from . import call_graph from .query_planner import QueryPlanner from .equality_mixin import HashMixin, EqualityMixin +from .result_set import default_exception_handler class QuerySearchIterator(object): @@ -399,7 +400,8 @@ def optimize(self, query_search): from .optimize_constrain_sql_queries import constrain_sql_queries constrain_sql_queries(query_search.call_graph) - def query(self, query, limit=None): + def query(self, query, limit=None, + exception_handler=default_exception_handler): query_search = QuerySearch(self, query) query_search.backward() @@ -412,7 +414,9 @@ def query(self, query, limit=None): ) query_plan = query_planner.plan_query() - return query_plan.execute(limit=limit) + return query_plan.execute( + limit=limit, exception_handler=exception_handler + ) def explain(self, query): query_search = QuerySearch(self, query)