mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-31 03:04:12 +02:00
304ddbec84
* Refactor views class to support more Jinja2 views (J2Template) * Add a home page, collection search page, and error pages, all optional * all exceptions appear on error page * wbrequest supports a request with an empty or / wb_url
82 lines
2.5 KiB
Python
82 lines
2.5 KiB
Python
import indexreader
|
|
import utils
|
|
import wbrequestresponse
|
|
import wbexceptions
|
|
import time
|
|
|
|
from os import path
|
|
from itertools import imap
|
|
from jinja2 import Environment, FileSystemLoader
|
|
|
|
|
|
#=================================================================
|
|
class StaticTextView:
|
|
def __init__(self, text):
|
|
self.text = text
|
|
|
|
def render_to_string(self, **kwargs):
|
|
return self.text
|
|
|
|
def render_response(self, **kwargs):
|
|
return wbrequestresponse.WbResponse.text_stream(self.text)
|
|
|
|
#=================================================================
|
|
class J2TemplateView:
|
|
def __init__(self, filename):
|
|
template_dir, template_file = path.split(filename)
|
|
|
|
self.template_file = template_file
|
|
|
|
self.jinja_env = self.make_jinja_env(template_dir)
|
|
|
|
|
|
def make_jinja_env(self, template_dir):
|
|
jinja_env = Environment(loader = FileSystemLoader(template_dir), trim_blocks = True)
|
|
jinja_env.filters['format_ts'] = J2TemplateView.format_ts
|
|
return jinja_env
|
|
|
|
def render_to_string(self, **kwargs):
|
|
template = self.jinja_env.get_template(self.template_file)
|
|
|
|
template_result = template.render(**kwargs)
|
|
|
|
return template_result
|
|
|
|
def render_response(self, **kwargs):
|
|
template_result = self.render_to_string(**kwargs)
|
|
return wbrequestresponse.WbResponse.text_response(str(template_result), content_type = 'text/html; charset=utf-8')
|
|
|
|
|
|
# Filters
|
|
@staticmethod
|
|
def format_ts(value, format='%a, %b %d %Y %H:%M:%S'):
|
|
value = utils.timestamp_to_datetime(value)
|
|
return time.strftime(format, value)
|
|
|
|
|
|
|
|
|
|
# cdx index view
|
|
|
|
#=================================================================
|
|
# html captures 'calendar' view
|
|
#=================================================================
|
|
class J2HtmlCapturesView(J2TemplateView):
|
|
def render_response(self, wbrequest, cdx_lines):
|
|
return J2TemplateView.render_response(self,
|
|
cdx_lines = list(cdx_lines),
|
|
url = wbrequest.wb_url.url,
|
|
prefix = wbrequest.wb_prefix)
|
|
|
|
|
|
#=================================================================
|
|
# stream raw cdx text
|
|
#=================================================================
|
|
class TextCapturesView:
|
|
def render_response(self, wbrequest, cdx_lines):
|
|
cdx_lines = imap(lambda x: str(x) + '\n', cdx_lines)
|
|
return wbrequestresponse.WbResponse.text_stream(cdx_lines)
|
|
|
|
|
|
|