mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-28 00:25:21 +01:00
6461af030b
use BlockLoader().load for StaticHandler static file resolving update static paths to point to pywb/static instead of static
128 lines
4.2 KiB
Python
128 lines
4.2 KiB
Python
import pkgutil
|
|
import mimetypes
|
|
import time
|
|
|
|
from pywb.utils.wbexception import NotFoundException
|
|
from pywb.utils.loaders import BlockLoader
|
|
|
|
from pywb.framework.basehandlers import BaseHandler, WbUrlHandler
|
|
from pywb.framework.wbrequestresponse import WbResponse
|
|
|
|
from views import TextCapturesView
|
|
|
|
|
|
#=================================================================
|
|
# Standard WB Handler
|
|
#=================================================================
|
|
class WBHandler(WbUrlHandler):
|
|
def __init__(self, index_reader, replay,
|
|
html_view=None, search_view=None):
|
|
|
|
self.index_reader = index_reader
|
|
|
|
self.replay = replay
|
|
|
|
self.text_query_view = TextCapturesView()
|
|
|
|
self.query_view = html_view
|
|
if not self.query_view:
|
|
self.query_view = text_query_view
|
|
|
|
self.search_view = search_view
|
|
|
|
def __call__(self, wbrequest):
|
|
if wbrequest.wb_url_str == '/':
|
|
return self.render_search_page(wbrequest)
|
|
|
|
with PerfTimer(wbrequest.env.get('X_PERF'), 'query') as t:
|
|
cdx_lines = self.index_reader.load_for_request(wbrequest)
|
|
|
|
# new special modifier to always show cdx index
|
|
if wbrequest.wb_url.mod == 'cdx_':
|
|
return self.text_query_view.render_response(wbrequest, cdx_lines)
|
|
|
|
if ((wbrequest.wb_url.type == wbrequest.wb_url.QUERY) or
|
|
(wbrequest.wb_url.type == wbrequest.wb_url.URL_QUERY)):
|
|
return self.query_view.render_response(wbrequest, cdx_lines)
|
|
|
|
with PerfTimer(wbrequest.env.get('X_PERF'), 'replay') as t:
|
|
return self.replay(wbrequest,
|
|
cdx_lines,
|
|
self.index_reader.cdx_load_callback(wbrequest))
|
|
|
|
def render_search_page(self, wbrequest):
|
|
if self.search_view:
|
|
return self.search_view.render_response(wbrequest=wbrequest)
|
|
else:
|
|
return WbResponse.text_response('No Lookup Url Specified')
|
|
|
|
def __str__(self):
|
|
return 'WBHandler: ' + str(self.index_reader) + ', ' + str(self.replay)
|
|
|
|
|
|
#=================================================================
|
|
# Static Content Handler
|
|
#=================================================================
|
|
class StaticHandler(BaseHandler):
|
|
def __init__(self, static_path, pkg='pywb'):
|
|
mimetypes.init()
|
|
|
|
self.static_path = static_path
|
|
self.pkg = pkg
|
|
|
|
def __call__(self, wbrequest):
|
|
full_path = self.static_path + wbrequest.wb_url_str
|
|
|
|
try:
|
|
#if full_path.startswith('.') or full_path.startswith('file://'):
|
|
# data = open(full_path, 'rb')
|
|
#else:
|
|
# data = pkgutil.get_data(self.pkg, full_path)
|
|
data = BlockLoader().load(full_path)
|
|
|
|
if 'wsgi.file_wrapper' in wbrequest.env:
|
|
reader = wbrequest.env['wsgi.file_wrapper'](data)
|
|
else:
|
|
reader = iter(lambda: data.read(), '')
|
|
|
|
content_type, _ = mimetypes.guess_type(full_path)
|
|
|
|
return WbResponse.text_stream(data, content_type=content_type)
|
|
|
|
except IOError:
|
|
raise NotFoundException('Static File Not Found: ' +
|
|
wbrequest.wb_url_str)
|
|
|
|
def __str__(self):
|
|
return 'Static files from ' + self.static_path
|
|
|
|
|
|
#=================================================================
|
|
# Debug Handlers
|
|
#=================================================================
|
|
class DebugEchoEnvHandler(BaseHandler):
|
|
def __call__(self, wbrequest):
|
|
return WbResponse.text_response(str(wbrequest.env))
|
|
|
|
|
|
#=================================================================
|
|
class DebugEchoHandler(BaseHandler):
|
|
def __call__(self, wbrequest):
|
|
return WbResponse.text_response(str(wbrequest))
|
|
|
|
|
|
#=================================================================
|
|
class PerfTimer:
|
|
def __init__(self, perfdict, name):
|
|
self.perfdict = perfdict
|
|
self.name = name
|
|
|
|
def __enter__(self):
|
|
self.start = time.clock()
|
|
return self
|
|
|
|
def __exit__(self, *args):
|
|
self.end = time.clock()
|
|
if self.perfdict is not None:
|
|
self.perfdict[self.name] = str(self.end - self.start)
|