From b056acd88e149deeefdf1bf10fc323c64f99850c Mon Sep 17 00:00:00 2001 From: Ilya Kreymer Date: Fri, 15 Apr 2016 04:01:36 +0000 Subject: [PATCH] urlrewrite: add support for index query --- urlrewrite/rewriterapp.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/urlrewrite/rewriterapp.py b/urlrewrite/rewriterapp.py index b0d1e8f9..84e536a9 100644 --- a/urlrewrite/rewriterapp.py +++ b/urlrewrite/rewriterapp.py @@ -210,8 +210,12 @@ class RewriterApp(object): return r - - + def do_query(self, wb_url, kwargs): + upstream_url = self.get_upstream_url(wb_url.url, wb_url, 'now', kwargs) + upstream_url = upstream_url.replace('/resource/postreq', '/index') + r = requests.get(upstream_url + '&output=json') + print(r.text) + return r.text def get_host_prefix(self): return request.urlparts.scheme + '://' + request.urlparts.netloc @@ -255,6 +259,9 @@ class RewriterApp(object): return None def handle_custom_response(self, wb_url, full_prefix, host_prefix, kwargs): + if wb_url.is_query(): + return self.do_query(wb_url, kwargs) + if self.framed_replay and wb_url.mod == self.frame_mod: extra_params = self.get_top_frame_params(wb_url, kwargs) return self.frame_insert_view.get_top_frame(wb_url,