backup/pywb
1
0
Fork 0
mirror of https://github.com/webrecorder/pywb.git synced 2025-03-24 06:59:52 +01:00
Code Issues Projects Releases Activity

urlrewrite: add support for index query

Browse Source
This commit is contained in:
Ilya Kreymer 2016-04-15 04:01:36 +00:00
parent 0370470e68
commit b056acd88e
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
urlrewrite/rewriterapp.py
View File

@ -210,8 +210,12 @@ class RewriterApp(object):
return r 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): def get_host_prefix(self):
return request.urlparts.scheme + '://' + request.urlparts.netloc return request.urlparts.scheme + '://' + request.urlparts.netloc
@ -255,6 +259,9 @@ class RewriterApp(object):
return None return None
def handle_custom_response(self, wb_url, full_prefix, host_prefix, kwargs): 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: if self.framed_replay and wb_url.mod == self.frame_mod:
extra_params = self.get_top_frame_params(wb_url, kwargs) extra_params = self.get_top_frame_params(wb_url, kwargs)
return self.frame_insert_view.get_top_frame(wb_url, return self.frame_insert_view.get_top_frame(wb_url,