devtools/repotest.py
changeset 9445 65d93a4fd11c
parent 9098 5467fb901931
child 9448 3e7cad3967c5
equal deleted inserted replaced
9444:6012cf57f48e 9445:65d93a4fd11c
   409             except AttributeError:
   409             except AttributeError:
   410                 # const
   410                 # const
   411                 return x.value
   411                 return x.value
   412     return _orig_choose_term(self, source, DumbOrderedDict2(sourceterms, get_key))
   412     return _orig_choose_term(self, source, DumbOrderedDict2(sourceterms, get_key))
   413 
   413 
   414 from cubicweb.server.sources.pyrorql import PyroRQLSource
       
   415 _orig_syntax_tree_search = PyroRQLSource.syntax_tree_search
       
   416 
       
   417 def _syntax_tree_search(*args, **kwargs):
       
   418     return deepcopy(_orig_syntax_tree_search(*args, **kwargs))
       
   419 
   414 
   420 def _ordered_iter_relations(stinfo):
   415 def _ordered_iter_relations(stinfo):
   421     return sorted(_orig_iter_relations(stinfo), key=lambda x:x.r_type)
   416     return sorted(_orig_iter_relations(stinfo), key=lambda x:x.r_type)
   422 
   417 
   423 def do_monkey_patch():
   418 def do_monkey_patch():
   427     ExecutionPlan._check_permissions = _check_permissions
   422     ExecutionPlan._check_permissions = _check_permissions
   428     ExecutionPlan.tablesinorder = None
   423     ExecutionPlan.tablesinorder = None
   429     ExecutionPlan.init_temp_table = _init_temp_table
   424     ExecutionPlan.init_temp_table = _init_temp_table
   430     PartPlanInformation.merge_input_maps = _merge_input_maps
   425     PartPlanInformation.merge_input_maps = _merge_input_maps
   431     PartPlanInformation._choose_term = _choose_term
   426     PartPlanInformation._choose_term = _choose_term
   432     PyroRQLSource.syntax_tree_search = _syntax_tree_search
       
   433 
   427 
   434 def undo_monkey_patch():
   428 def undo_monkey_patch():
   435     rqlrewrite.iter_relations = _orig_iter_relations
   429     rqlrewrite.iter_relations = _orig_iter_relations
   436     rqlrewrite.RQLRewriter.insert_snippets = _orig_insert_snippets
   430     rqlrewrite.RQLRewriter.insert_snippets = _orig_insert_snippets
   437     rqlrewrite.RQLRewriter.build_variantes = _orig_build_variantes
   431     rqlrewrite.RQLRewriter.build_variantes = _orig_build_variantes
   438     ExecutionPlan._check_permissions = _orig_check_permissions
   432     ExecutionPlan._check_permissions = _orig_check_permissions
   439     ExecutionPlan.init_temp_table = _orig_init_temp_table
   433     ExecutionPlan.init_temp_table = _orig_init_temp_table
   440     PartPlanInformation.merge_input_maps = _orig_merge_input_maps
   434     PartPlanInformation.merge_input_maps = _orig_merge_input_maps
   441     PartPlanInformation._choose_term = _orig_choose_term
   435     PartPlanInformation._choose_term = _orig_choose_term
   442     PyroRQLSource.syntax_tree_search = _orig_syntax_tree_search