piecrust2: piecrust/serving/server.py annotate

author	Ludovic Chabant <ludovic@chabant.com>
date	Thu, 07 May 2015 21:37:38 -0700
parents
children	aade4ea57e7f

rev	line source
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	1 import io
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	2 import os
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	3 import re
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	4 import gzip
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	5 import time
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	6 import os.path
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 import hashlib
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8 import logging
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	9 import datetime
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	10 from werkzeug.exceptions import (
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	11 NotFound, MethodNotAllowed, InternalServerError, HTTPException)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	12 from werkzeug.wrappers import Request, Response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	13 from werkzeug.wsgi import ClosingIterator, wrap_file
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	14 from jinja2 import FileSystemLoader, Environment
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	15 from piecrust.app import PieCrust
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	16 from piecrust.rendering import QualifiedPage, PageRenderingContext, render_page
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	17 from piecrust.sources.base import MODE_PARSING
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	18 from piecrust.uriutil import split_sub_uri
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	19
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	20
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 logger = logging.getLogger(__name__)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	24 class ServeRecord(object):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	25 def __init__(self):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	26 self.entries = {}
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	27
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	28 def addEntry(self, entry):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	29 key = self._makeKey(entry.uri, entry.sub_num)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	30 self.entries[key] = entry
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	31
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	32 def getEntry(self, uri, sub_num):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	33 key = self._makeKey(uri, sub_num)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	34 return self.entries.get(key)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	35
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	36 def _makeKey(self, uri, sub_num):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	37 return "%s:%s" % (uri, sub_num)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	38
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	39
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	40 class ServeRecordPageEntry(object):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	41 def __init__(self, uri, sub_num):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	42 self.uri = uri
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	43 self.sub_num = sub_num
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	44 self.used_source_names = set()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	45
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	46
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	47 class WsgiServerWrapper(object):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	48 def __init__(self, server):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	49 self.server = server
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	50
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	51 def __call__(self, environ, start_response):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	52 return self.server._run_request(environ, start_response)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	53
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	54
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	55 class Server(object):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	56 def __init__(self, root_dir,
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	57 debug=False, sub_cache_dir=None,
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	58 static_preview=True, run_sse_check=None):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	59 self.root_dir = root_dir
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	60 self.debug = debug
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	61 self.sub_cache_dir = sub_cache_dir
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	62 self.run_sse_check = run_sse_check
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	63 self.static_preview = static_preview
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	64 self._out_dir = None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	65 self._page_record = None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	66 self._proc_loop = None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	67 self._mimetype_map = load_mimetype_map()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	68
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	69 def getWsgiApp(self):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	70 # Bake all the assets so we know what we have, and so we can serve
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	71 # them to the client. We need a temp app for this.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	72 app = PieCrust(root_dir=self.root_dir, debug=self.debug)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	73 app._useSubCacheDir(self.sub_cache_dir)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	74 self._out_dir = os.path.join(app.sub_cache_dir, 'server')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	75 self._page_record = ServeRecord()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	76
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	77 if not self.run_sse_check or self.run_sse_check():
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	78 # When using a server with code reloading, some implementations
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	79 # use process forking and we end up going here twice. We only want
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	80 # to start the pipeline loop in the inner process most of the
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	81 # time so we let the implementation tell us if this is OK.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	82 from piecrust.processing.base import ProcessorPipeline
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	83 from piecrust.serving.procloop import ProcessingLoop
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	84 pipeline = ProcessorPipeline(app, self._out_dir)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	85 self._proc_loop = ProcessingLoop(pipeline)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	86 self._proc_loop.start()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	87
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	88 # Run the WSGI app.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	89 wsgi_wrapper = WsgiServerWrapper(self)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	90 return wsgi_wrapper
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	91
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	92 def _run_request(self, environ, start_response):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	93 try:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	94 return self._try_run_request(environ, start_response)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	95 except Exception as ex:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	96 if self.debug:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	97 raise
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	98 return self._handle_error(ex, environ, start_response)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	99
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	100 def _try_run_request(self, environ, start_response):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	101 request = Request(environ)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	102
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	103 # We don't support anything else than GET requests since we're
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	104 # previewing something that will be static later.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	105 if self.static_preview and request.method != 'GET':
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	106 logger.error("Only GET requests are allowed, got %s" %
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	107 request.method)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	108 raise MethodNotAllowed()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	109
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	110 # Handle special requests right away.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	111 response = self._try_special_request(environ, request)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	112 if response is not None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	113 return response(environ, start_response)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	114
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	115 # Also handle requests to a pipeline-built asset right away.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	116 response = self._try_serve_asset(environ, request)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	117 if response is not None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	118 return response(environ, start_response)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	119
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	120 # Create the app for this request.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	121 app = PieCrust(root_dir=self.root_dir, debug=self.debug)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	122 app._useSubCacheDir(self.sub_cache_dir)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	123 app.config.set('site/root', '/')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	124 app.config.set('server/is_serving', True)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	125 if (app.config.get('site/enable_debug_info') and
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	126 '!debug' in request.args):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	127 app.config.set('site/show_debug_info', True)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	128
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	129 # We'll serve page assets directly from where they are.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	130 app.env.base_asset_url_format = '/_asset/%path%'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	131
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	132 # Let's see if it can be a page asset.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	133 response = self._try_serve_page_asset(app, environ, request)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	134 if response is not None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	135 return response(environ, start_response)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	136
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	137 # Nope. Let's see if it's an actual page.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	138 try:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	139 response = self._try_serve_page(app, environ, request)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	140 return response(environ, start_response)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	141 except (RouteNotFoundError, SourceNotFoundError) as ex:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	142 raise NotFound(str(ex)) from ex
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	143 except HTTPException:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	144 raise
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	145 except Exception as ex:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	146 if app.debug:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	147 logger.exception(ex)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	148 raise
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	149 msg = str(ex)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	150 logger.error(msg)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	151 raise InternalServerError(msg) from ex
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	152
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	153 def _try_special_request(self, environ, request):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	154 static_mount = '/__piecrust_static/'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	155 if request.path.startswith(static_mount):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	156 rel_req_path = request.path[len(static_mount):]
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	157 mount = os.path.join(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	158 os.path.dirname(__file__),
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	159 'resources', 'server')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	160 full_path = os.path.join(mount, rel_req_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	161 try:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	162 response = self._make_wrapped_file_response(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	163 environ, request, full_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	164 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	165 except OSError:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	166 pass
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	167
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	168 debug_mount = '/__piecrust_debug/'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	169 if request.path.startswith(debug_mount):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	170 rel_req_path = request.path[len(debug_mount):]
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	171 if rel_req_path == 'pipeline_status':
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	172 from piecrust.server.procloop import (
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	173 PipelineStatusServerSideEventProducer)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	174 provider = PipelineStatusServerSideEventProducer(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	175 self._proc_loop.status_queue)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	176 it = ClosingIterator(provider.run(), [provider.close])
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	177 response = Response(it)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	178 response.headers['Cache-Control'] = 'no-cache'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	179 if 'text/event-stream' in request.accept_mimetypes:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	180 response.mimetype = 'text/event-stream'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	181 response.direct_passthrough = True
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	182 response.implicit_sequence_conversion = False
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	183 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	184
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	185 return None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	186
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	187 def _try_serve_asset(self, environ, request):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	188 rel_req_path = request.path.lstrip('/').replace('/', os.sep)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	189 if request.path.startswith('/_cache/'):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	190 # Some stuff needs to be served directly from the cache directory,
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	191 # like LESS CSS map files.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	192 full_path = os.path.join(self.root_dir, rel_req_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	193 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	194 full_path = os.path.join(self._out_dir, rel_req_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	195
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	196 try:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	197 response = self._make_wrapped_file_response(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	198 environ, request, full_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	199 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	200 except OSError:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	201 pass
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	202 return None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	203
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	204 def _try_serve_page_asset(self, app, environ, request):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	205 if not request.path.startswith('/_asset/'):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	206 return None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	207
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	208 full_path = os.path.join(app.root_dir, request.path[len('/_asset/'):])
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	209 if not os.path.isfile(full_path):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	210 return None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	211
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	212 return self._make_wrapped_file_response(environ, request, full_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	213
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	214 def _try_serve_page(self, app, environ, request):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	215 # Try to find what matches the requested URL.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	216 req_path, page_num = split_sub_uri(app, request.path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	217
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	218 routes = find_routes(app.routes, req_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	219 if len(routes) == 0:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	220 raise RouteNotFoundError("Can't find route for: %s" % req_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	221
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	222 rendered_page = None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	223 first_not_found = None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	224 for route, route_metadata in routes:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	225 try:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	226 logger.debug("Trying to render match from source '%s'." %
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	227 route.source_name)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	228 rendered_page = self._try_render_page(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	229 app, route, route_metadata, page_num, req_path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	230 if rendered_page is not None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	231 break
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	232 except NotFound as nfe:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	233 if first_not_found is None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	234 first_not_found = nfe
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	235 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	236 raise SourceNotFoundError(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	237 "Can't find path for: %s (looked in: %s)" %
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	238 (req_path, [r.source_name for r, _ in routes]))
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	239
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	240 # If we haven't found any good match, raise whatever exception we
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	241 # first got. Otherwise, raise a generic exception.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	242 if rendered_page is None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	243 first_not_found = first_not_found or NotFound(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	244 "This page couldn't be found.")
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	245 raise first_not_found
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	246
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	247 # Start doing stuff.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	248 page = rendered_page.page
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	249 rp_content = rendered_page.content
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	250
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	251 # Profiling.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	252 if app.config.get('site/show_debug_info'):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	253 now_time = time.clock()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	254 timing_info = (
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	255 '%8.1f ms' %
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	256 ((now_time - app.env.start_time) * 1000.0))
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	257 rp_content = rp_content.replace(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	258 '__PIECRUST_TIMING_INFORMATION__', timing_info)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	259
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	260 # Build the response.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	261 response = Response()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	262
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	263 etag = hashlib.md5(rp_content.encode('utf8')).hexdigest()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	264 if not app.debug and etag in request.if_none_match:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	265 response.status_code = 304
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	266 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	267
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	268 response.set_etag(etag)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	269 response.content_md5 = etag
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	270
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	271 cache_control = response.cache_control
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	272 if app.debug:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	273 cache_control.no_cache = True
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	274 cache_control.must_revalidate = True
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	275 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	276 cache_time = (page.config.get('cache_time') or
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	277 app.config.get('site/cache_time'))
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	278 if cache_time:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	279 cache_control.public = True
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	280 cache_control.max_age = cache_time
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	281
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	282 content_type = page.config.get('content_type')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	283 if content_type and '/' not in content_type:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	284 mimetype = content_type_map.get(content_type, content_type)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	285 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	286 mimetype = content_type
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	287 if mimetype:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	288 response.mimetype = mimetype
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	289
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	290 if ('gzip' in request.accept_encodings and
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	291 app.config.get('site/enable_gzip')):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	292 try:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	293 with io.BytesIO() as gzip_buffer:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	294 with gzip.open(gzip_buffer, mode='wt',
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	295 encoding='utf8') as gzip_file:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	296 gzip_file.write(rp_content)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	297 rp_content = gzip_buffer.getvalue()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	298 response.content_encoding = 'gzip'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	299 except Exception:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	300 logger.exception("Error compressing response, "
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	301 "falling back to uncompressed.")
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	302 response.set_data(rp_content)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	303
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	304 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	305
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	306 def _try_render_page(self, app, route, route_metadata, page_num, req_path):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	307 # Match the route to an actual factory.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	308 taxonomy_info = None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	309 source = app.getSource(route.source_name)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	310 if route.taxonomy_name is None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	311 factory = source.findPageFactory(route_metadata, MODE_PARSING)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	312 if factory is None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	313 return None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	314 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	315 taxonomy = app.getTaxonomy(route.taxonomy_name)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	316 route_terms = route_metadata.get(taxonomy.term_name)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	317 if route_terms is None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	318 return None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	319
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	320 tax_page_ref = taxonomy.getPageRef(source.name)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	321 factory = tax_page_ref.getFactory()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	322 tax_terms = route.unslugifyTaxonomyTerm(route_terms)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	323 route_metadata[taxonomy.term_name] = tax_terms
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	324 taxonomy_info = (taxonomy, tax_terms)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	325
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	326 # Build the page.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	327 page = factory.buildPage()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	328 # We force the rendering of the page because it could not have
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	329 # changed, but include pages that did change.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	330 qp = QualifiedPage(page, route, route_metadata)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	331 render_ctx = PageRenderingContext(qp,
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	332 page_num=page_num,
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	333 force_render=True)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	334 if taxonomy_info is not None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	335 taxonomy, tax_terms = taxonomy_info
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	336 render_ctx.setTaxonomyFilter(taxonomy, tax_terms)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	337
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	338 # See if this page is known to use sources. If that's the case,
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	339 # just don't use cached rendered segments for that page (but still
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	340 # use them for pages that are included in it).
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	341 uri = qp.getUri()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	342 assert uri == req_path
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	343 entry = self._page_record.getEntry(uri, page_num)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	344 if (taxonomy_info is not None or entry is None or
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	345 entry.used_source_names):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	346 cache_key = '%s:%s' % (uri, page_num)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	347 app.env.rendered_segments_repository.invalidate(cache_key)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	348
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	349 # Render the page.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	350 rendered_page = render_page(render_ctx)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	351
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	352 # Check if this page is a taxonomy page that actually doesn't match
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	353 # anything.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	354 if taxonomy_info is not None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	355 paginator = rendered_page.data.get('pagination')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	356 if (paginator and paginator.is_loaded and
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	357 len(paginator.items) == 0):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	358 taxonomy = taxonomy_info[0]
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	359 message = ("This URL matched a route for taxonomy '%s' but "
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	360 "no pages have been found to have it. This page "
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	361 "won't be generated by a bake." % taxonomy.name)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	362 raise NotFound(message)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	363
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	364 # Remember stuff for next time.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	365 if entry is None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	366 entry = ServeRecordPageEntry(req_path, page_num)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	367 self._page_record.addEntry(entry)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	368 for p, pinfo in render_ctx.render_passes.items():
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	369 entry.used_source_names \|= pinfo.used_source_names
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	370
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	371 # Ok all good.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	372 return rendered_page
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	373
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	374 def _make_wrapped_file_response(self, environ, request, path):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	375 logger.debug("Serving %s" % path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	376
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	377 # Check if we can return a 304 status code.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	378 mtime = os.path.getmtime(path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	379 etag_str = '%s$$%s' % (path, mtime)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	380 etag = hashlib.md5(etag_str.encode('utf8')).hexdigest()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	381 if etag in request.if_none_match:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	382 response = Response()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	383 response.status_code = 304
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	384 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	385
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	386 wrapper = wrap_file(environ, open(path, 'rb'))
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	387 response = Response(wrapper)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	388 _, ext = os.path.splitext(path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	389 response.set_etag(etag)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	390 response.last_modified = datetime.datetime.fromtimestamp(mtime)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	391 response.mimetype = self._mimetype_map.get(
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	392 ext.lstrip('.'), 'text/plain')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	393 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	394
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	395 def _handle_error(self, exception, environ, start_response):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	396 code = 500
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	397 if isinstance(exception, HTTPException):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	398 code = exception.code
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	399
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	400 path = 'error'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	401 if isinstance(exception, NotFound):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	402 path += '404'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	403
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	404 descriptions = self._get_exception_descriptions(exception)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	405
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	406 env = Environment(loader=ErrorMessageLoader())
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	407 template = env.get_template(path)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	408 context = {'details': descriptions}
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	409 response = Response(template.render(context), mimetype='text/html')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	410 response.status_code = code
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	411 return response(environ, start_response)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	412
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	413 def _get_exception_descriptions(self, exception):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	414 desc = []
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	415 while exception is not None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	416 if isinstance(exception, HTTPException):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	417 desc.append(exception.description)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	418 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	419 desc.append(str(exception))
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	420
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	421 inner_ex = exception.__cause__
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	422 if inner_ex is None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	423 inner_ex = exception.__context__
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	424 exception = inner_ex
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	425 return desc
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	426
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	427
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	428 class RouteNotFoundError(Exception):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	429 pass
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	430
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	431
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	432 class SourceNotFoundError(Exception):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	433 pass
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	434
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	435
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	436 content_type_map = {
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	437 'html': 'text/html',
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	438 'xml': 'text/xml',
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	439 'txt': 'text/plain',
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	440 'text': 'text/plain',
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	441 'css': 'text/css',
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	442 'xhtml': 'application/xhtml+xml',
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	443 'atom': 'application/atom+xml', # or 'text/xml'?
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	444 'rss': 'application/rss+xml', # or 'text/xml'?
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	445 'json': 'application/json'}
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	446
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	447
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	448 def find_routes(routes, uri):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	449 res = []
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	450 for route in routes:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	451 metadata = route.matchUri(uri)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	452 if metadata is not None:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	453 res.append((route, metadata))
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	454 return res
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	455
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	456
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	457 class ErrorMessageLoader(FileSystemLoader):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	458 def __init__(self):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	459 base_dir = os.path.join(os.path.dirname(__file__), 'resources',
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	460 'messages')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	461 super(ErrorMessageLoader, self).__init__(base_dir)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	462
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	463 def get_source(self, env, template):
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	464 template += '.html'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	465 return super(ErrorMessageLoader, self).get_source(env, template)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	466
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	467
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	468 def load_mimetype_map():
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	469 mimetype_map = {}
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	470 sep_re = re.compile(r'\s+')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	471 path = os.path.join(os.path.dirname(__file__), 'mime.types')
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	472 with open(path, 'r') as f:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	473 for line in f:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	474 tokens = sep_re.split(line)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	475 if len(tokens) > 1:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	476 for t in tokens[1:]:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	477 mimetype_map[t] = tokens[0]
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	478 return mimetype_map
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	479

374

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

1 import io

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

2 import os

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

3 import re

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

4 import gzip

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

5 import time

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

6 import os.path

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

7 import hashlib

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

8 import logging

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

9 import datetime

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

10 from werkzeug.exceptions import (

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

11 NotFound, MethodNotAllowed, InternalServerError, HTTPException)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

12 from werkzeug.wrappers import Request, Response

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

13 from werkzeug.wsgi import ClosingIterator, wrap_file

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

14 from jinja2 import FileSystemLoader, Environment

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

15 from piecrust.app import PieCrust

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

16 from piecrust.rendering import QualifiedPage, PageRenderingContext, render_page

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

17 from piecrust.sources.base import MODE_PARSING

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

18 from piecrust.uriutil import split_sub_uri

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

19

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

20

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

21 logger = logging.getLogger(__name__)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

22

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

23

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

24 class ServeRecord(object):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

25 def __init__(self):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

26 self.entries = {}

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

27

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

28 def addEntry(self, entry):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

29 key = self._makeKey(entry.uri, entry.sub_num)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

30 self.entries[key] = entry

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

31

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

32 def getEntry(self, uri, sub_num):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

33 key = self._makeKey(uri, sub_num)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

34 return self.entries.get(key)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

35

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

36 def _makeKey(self, uri, sub_num):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

37 return "%s:%s" % (uri, sub_num)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

38

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

39

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

40 class ServeRecordPageEntry(object):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

41 def __init__(self, uri, sub_num):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

42 self.uri = uri

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

43 self.sub_num = sub_num

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

44 self.used_source_names = set()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

45

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

46

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

47 class WsgiServerWrapper(object):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

48 def __init__(self, server):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

49 self.server = server

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

50

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

51 def __call__(self, environ, start_response):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

52 return self.server._run_request(environ, start_response)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

53

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

54

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

55 class Server(object):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

56 def __init__(self, root_dir,

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

57 debug=False, sub_cache_dir=None,

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

58 static_preview=True, run_sse_check=None):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

59 self.root_dir = root_dir

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

60 self.debug = debug

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

61 self.sub_cache_dir = sub_cache_dir

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

62 self.run_sse_check = run_sse_check

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

63 self.static_preview = static_preview

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

64 self._out_dir = None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

65 self._page_record = None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

66 self._proc_loop = None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

67 self._mimetype_map = load_mimetype_map()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

68

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

69 def getWsgiApp(self):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

70 # Bake all the assets so we know what we have, and so we can serve

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

71 # them to the client. We need a temp app for this.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

72 app = PieCrust(root_dir=self.root_dir, debug=self.debug)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

73 app._useSubCacheDir(self.sub_cache_dir)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

74 self._out_dir = os.path.join(app.sub_cache_dir, 'server')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

75 self._page_record = ServeRecord()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

76

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

77 if not self.run_sse_check or self.run_sse_check():

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

78 # When using a server with code reloading, some implementations

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

79 # use process forking and we end up going here twice. We only want

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

80 # to start the pipeline loop in the inner process most of the

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

81 # time so we let the implementation tell us if this is OK.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

82 from piecrust.processing.base import ProcessorPipeline

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

83 from piecrust.serving.procloop import ProcessingLoop

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

84 pipeline = ProcessorPipeline(app, self._out_dir)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

85 self._proc_loop = ProcessingLoop(pipeline)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

86 self._proc_loop.start()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

87

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

88 # Run the WSGI app.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

89 wsgi_wrapper = WsgiServerWrapper(self)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

90 return wsgi_wrapper

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

91

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

92 def _run_request(self, environ, start_response):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

93 try:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

94 return self._try_run_request(environ, start_response)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

95 except Exception as ex:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

96 if self.debug:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

97 raise

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

98 return self._handle_error(ex, environ, start_response)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

99

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

100 def _try_run_request(self, environ, start_response):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

101 request = Request(environ)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

102

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

103 # We don't support anything else than GET requests since we're

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

104 # previewing something that will be static later.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

105 if self.static_preview and request.method != 'GET':

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

106 logger.error("Only GET requests are allowed, got %s" %

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

107 request.method)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

108 raise MethodNotAllowed()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

109

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

110 # Handle special requests right away.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

111 response = self._try_special_request(environ, request)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

112 if response is not None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

113 return response(environ, start_response)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

114

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

115 # Also handle requests to a pipeline-built asset right away.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

116 response = self._try_serve_asset(environ, request)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

117 if response is not None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

118 return response(environ, start_response)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

119

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

120 # Create the app for this request.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

121 app = PieCrust(root_dir=self.root_dir, debug=self.debug)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

122 app._useSubCacheDir(self.sub_cache_dir)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

123 app.config.set('site/root', '/')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

124 app.config.set('server/is_serving', True)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

125 if (app.config.get('site/enable_debug_info') and

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

126 '!debug' in request.args):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

127 app.config.set('site/show_debug_info', True)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

128

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

129 # We'll serve page assets directly from where they are.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

130 app.env.base_asset_url_format = '/_asset/%path%'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

131

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

132 # Let's see if it can be a page asset.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

133 response = self._try_serve_page_asset(app, environ, request)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

134 if response is not None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

135 return response(environ, start_response)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

136

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

137 # Nope. Let's see if it's an actual page.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

138 try:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

139 response = self._try_serve_page(app, environ, request)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

140 return response(environ, start_response)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

141 except (RouteNotFoundError, SourceNotFoundError) as ex:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

142 raise NotFound(str(ex)) from ex

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

143 except HTTPException:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

144 raise

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

145 except Exception as ex:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

146 if app.debug:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

147 logger.exception(ex)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

148 raise

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

149 msg = str(ex)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

150 logger.error(msg)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

151 raise InternalServerError(msg) from ex

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

152

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

153 def _try_special_request(self, environ, request):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

154 static_mount = '/__piecrust_static/'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

155 if request.path.startswith(static_mount):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

156 rel_req_path = request.path[len(static_mount):]

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

157 mount = os.path.join(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

158 os.path.dirname(__file__),

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

159 'resources', 'server')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

160 full_path = os.path.join(mount, rel_req_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

161 try:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

162 response = self._make_wrapped_file_response(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

163 environ, request, full_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

164 return response

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

165 except OSError:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

166 pass

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

167

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

168 debug_mount = '/__piecrust_debug/'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

169 if request.path.startswith(debug_mount):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

170 rel_req_path = request.path[len(debug_mount):]

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

171 if rel_req_path == 'pipeline_status':

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

172 from piecrust.server.procloop import (

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

173 PipelineStatusServerSideEventProducer)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

174 provider = PipelineStatusServerSideEventProducer(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

175 self._proc_loop.status_queue)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

176 it = ClosingIterator(provider.run(), [provider.close])

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

177 response = Response(it)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

178 response.headers['Cache-Control'] = 'no-cache'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

179 if 'text/event-stream' in request.accept_mimetypes:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

180 response.mimetype = 'text/event-stream'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

181 response.direct_passthrough = True

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

182 response.implicit_sequence_conversion = False

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

183 return response

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

184

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

185 return None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

186

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

187 def _try_serve_asset(self, environ, request):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

188 rel_req_path = request.path.lstrip('/').replace('/', os.sep)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

189 if request.path.startswith('/_cache/'):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

190 # Some stuff needs to be served directly from the cache directory,

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

191 # like LESS CSS map files.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

192 full_path = os.path.join(self.root_dir, rel_req_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

193 else:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

194 full_path = os.path.join(self._out_dir, rel_req_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

195

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

196 try:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

197 response = self._make_wrapped_file_response(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

198 environ, request, full_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

199 return response

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

200 except OSError:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

201 pass

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

202 return None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

203

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

204 def _try_serve_page_asset(self, app, environ, request):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

205 if not request.path.startswith('/_asset/'):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

206 return None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

207

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

208 full_path = os.path.join(app.root_dir, request.path[len('/_asset/'):])

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

209 if not os.path.isfile(full_path):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

210 return None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

211

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

212 return self._make_wrapped_file_response(environ, request, full_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

213

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

214 def _try_serve_page(self, app, environ, request):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

215 # Try to find what matches the requested URL.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

216 req_path, page_num = split_sub_uri(app, request.path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

217

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

218 routes = find_routes(app.routes, req_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

219 if len(routes) == 0:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

220 raise RouteNotFoundError("Can't find route for: %s" % req_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

221

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

222 rendered_page = None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

223 first_not_found = None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

224 for route, route_metadata in routes:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

225 try:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

226 logger.debug("Trying to render match from source '%s'." %

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

227 route.source_name)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

228 rendered_page = self._try_render_page(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

229 app, route, route_metadata, page_num, req_path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

230 if rendered_page is not None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

231 break

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

232 except NotFound as nfe:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

233 if first_not_found is None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

234 first_not_found = nfe

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

235 else:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

236 raise SourceNotFoundError(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

237 "Can't find path for: %s (looked in: %s)" %

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

238 (req_path, [r.source_name for r, _ in routes]))

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

239

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

240 # If we haven't found any good match, raise whatever exception we

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

241 # first got. Otherwise, raise a generic exception.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

242 if rendered_page is None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

243 first_not_found = first_not_found or NotFound(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

244 "This page couldn't be found.")

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

245 raise first_not_found

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

246

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

247 # Start doing stuff.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

248 page = rendered_page.page

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

249 rp_content = rendered_page.content

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

250

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

251 # Profiling.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

252 if app.config.get('site/show_debug_info'):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

253 now_time = time.clock()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

254 timing_info = (

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

255 '%8.1f ms' %

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

256 ((now_time - app.env.start_time) * 1000.0))

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

257 rp_content = rp_content.replace(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

258 '__PIECRUST_TIMING_INFORMATION__', timing_info)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

259

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

260 # Build the response.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

261 response = Response()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

262

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

263 etag = hashlib.md5(rp_content.encode('utf8')).hexdigest()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

264 if not app.debug and etag in request.if_none_match:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

265 response.status_code = 304

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

266 return response

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

267

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

268 response.set_etag(etag)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

269 response.content_md5 = etag

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

270

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

271 cache_control = response.cache_control

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

272 if app.debug:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

273 cache_control.no_cache = True

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

274 cache_control.must_revalidate = True

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

275 else:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

276 cache_time = (page.config.get('cache_time') or

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

277 app.config.get('site/cache_time'))

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

278 if cache_time:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

279 cache_control.public = True

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

280 cache_control.max_age = cache_time

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

281

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

282 content_type = page.config.get('content_type')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

283 if content_type and '/' not in content_type:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

284 mimetype = content_type_map.get(content_type, content_type)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

285 else:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

286 mimetype = content_type

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

287 if mimetype:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

288 response.mimetype = mimetype

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

289

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

290 if ('gzip' in request.accept_encodings and

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

291 app.config.get('site/enable_gzip')):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

292 try:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

293 with io.BytesIO() as gzip_buffer:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

294 with gzip.open(gzip_buffer, mode='wt',

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

295 encoding='utf8') as gzip_file:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

296 gzip_file.write(rp_content)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

297 rp_content = gzip_buffer.getvalue()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

298 response.content_encoding = 'gzip'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

299 except Exception:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

300 logger.exception("Error compressing response, "

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

301 "falling back to uncompressed.")

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

302 response.set_data(rp_content)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

303

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

304 return response

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

305

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

306 def _try_render_page(self, app, route, route_metadata, page_num, req_path):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

307 # Match the route to an actual factory.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

308 taxonomy_info = None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

309 source = app.getSource(route.source_name)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

310 if route.taxonomy_name is None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

311 factory = source.findPageFactory(route_metadata, MODE_PARSING)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

312 if factory is None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

313 return None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

314 else:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

315 taxonomy = app.getTaxonomy(route.taxonomy_name)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

316 route_terms = route_metadata.get(taxonomy.term_name)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

317 if route_terms is None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

318 return None

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

319

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

320 tax_page_ref = taxonomy.getPageRef(source.name)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

321 factory = tax_page_ref.getFactory()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

322 tax_terms = route.unslugifyTaxonomyTerm(route_terms)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

323 route_metadata[taxonomy.term_name] = tax_terms

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

324 taxonomy_info = (taxonomy, tax_terms)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

325

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

326 # Build the page.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

327 page = factory.buildPage()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

328 # We force the rendering of the page because it could not have

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

329 # changed, but include pages that did change.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

330 qp = QualifiedPage(page, route, route_metadata)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

331 render_ctx = PageRenderingContext(qp,

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

332 page_num=page_num,

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

333 force_render=True)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

334 if taxonomy_info is not None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

335 taxonomy, tax_terms = taxonomy_info

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

336 render_ctx.setTaxonomyFilter(taxonomy, tax_terms)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

337

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

338 # See if this page is known to use sources. If that's the case,

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

339 # just don't use cached rendered segments for that page (but still

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

340 # use them for pages that are included in it).

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

341 uri = qp.getUri()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

342 assert uri == req_path

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

343 entry = self._page_record.getEntry(uri, page_num)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

344 if (taxonomy_info is not None or entry is None or

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

345 entry.used_source_names):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

346 cache_key = '%s:%s' % (uri, page_num)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

347 app.env.rendered_segments_repository.invalidate(cache_key)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

348

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

349 # Render the page.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

350 rendered_page = render_page(render_ctx)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

351

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

352 # Check if this page is a taxonomy page that actually doesn't match

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

353 # anything.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

354 if taxonomy_info is not None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

355 paginator = rendered_page.data.get('pagination')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

356 if (paginator and paginator.is_loaded and

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

357 len(paginator.items) == 0):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

358 taxonomy = taxonomy_info[0]

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

359 message = ("This URL matched a route for taxonomy '%s' but "

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

360 "no pages have been found to have it. This page "

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

361 "won't be generated by a bake." % taxonomy.name)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

362 raise NotFound(message)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

363

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

364 # Remember stuff for next time.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

365 if entry is None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

366 entry = ServeRecordPageEntry(req_path, page_num)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

367 self._page_record.addEntry(entry)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

368 for p, pinfo in render_ctx.render_passes.items():

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

369 entry.used_source_names |= pinfo.used_source_names

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

370

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

371 # Ok all good.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

372 return rendered_page

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

373

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

374 def _make_wrapped_file_response(self, environ, request, path):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

375 logger.debug("Serving %s" % path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

376

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

377 # Check if we can return a 304 status code.

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

378 mtime = os.path.getmtime(path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

379 etag_str = '%s$$%s' % (path, mtime)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

380 etag = hashlib.md5(etag_str.encode('utf8')).hexdigest()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

381 if etag in request.if_none_match:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

382 response = Response()

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

383 response.status_code = 304

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

384 return response

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

385

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

386 wrapper = wrap_file(environ, open(path, 'rb'))

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

387 response = Response(wrapper)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

388 _, ext = os.path.splitext(path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

389 response.set_etag(etag)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

390 response.last_modified = datetime.datetime.fromtimestamp(mtime)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

391 response.mimetype = self._mimetype_map.get(

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

392 ext.lstrip('.'), 'text/plain')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

393 return response

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

394

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

395 def _handle_error(self, exception, environ, start_response):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

396 code = 500

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

397 if isinstance(exception, HTTPException):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

398 code = exception.code

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

399

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

400 path = 'error'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

401 if isinstance(exception, NotFound):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

402 path += '404'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

403

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

404 descriptions = self._get_exception_descriptions(exception)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

405

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

406 env = Environment(loader=ErrorMessageLoader())

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

407 template = env.get_template(path)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

408 context = {'details': descriptions}

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

409 response = Response(template.render(context), mimetype='text/html')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

410 response.status_code = code

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

411 return response(environ, start_response)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

412

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

413 def _get_exception_descriptions(self, exception):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

414 desc = []

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

415 while exception is not None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

416 if isinstance(exception, HTTPException):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

417 desc.append(exception.description)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

418 else:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

419 desc.append(str(exception))

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

420

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

421 inner_ex = exception.__cause__

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

422 if inner_ex is None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

423 inner_ex = exception.__context__

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

424 exception = inner_ex

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

425 return desc

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

426

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

427

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

428 class RouteNotFoundError(Exception):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

429 pass

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

430

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

431

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

432 class SourceNotFoundError(Exception):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

433 pass

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

434

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

435

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

436 content_type_map = {

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

437 'html': 'text/html',

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

438 'xml': 'text/xml',

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

439 'txt': 'text/plain',

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

440 'text': 'text/plain',

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

441 'css': 'text/css',

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

442 'xhtml': 'application/xhtml+xml',

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

443 'atom': 'application/atom+xml', # or 'text/xml'?

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

444 'rss': 'application/rss+xml', # or 'text/xml'?

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

445 'json': 'application/json'}

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

446

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

447

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

448 def find_routes(routes, uri):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

449 res = []

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

450 for route in routes:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

451 metadata = route.matchUri(uri)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

452 if metadata is not None:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

453 res.append((route, metadata))

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

454 return res

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

455

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

456

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

457 class ErrorMessageLoader(FileSystemLoader):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

458 def __init__(self):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

459 base_dir = os.path.join(os.path.dirname(__file__), 'resources',

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

460 'messages')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

461 super(ErrorMessageLoader, self).__init__(base_dir)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

462

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

463 def get_source(self, env, template):

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

464 template += '.html'

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

465 return super(ErrorMessageLoader, self).get_source(env, template)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

466

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

467

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

468 def load_mimetype_map():

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

469 mimetype_map = {}

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

470 sep_re = re.compile(r'\s+')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

471 path = os.path.join(os.path.dirname(__file__), 'mime.types')

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

472 with open(path, 'r') as f:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

473 for line in f:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

474 tokens = sep_re.split(line)

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

475 if len(tokens) > 1:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

476 for t in tokens[1:]:

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

477 mimetype_map[t] = tokens[0]

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

478 return mimetype_map

fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package.

Ludovic Chabant <ludovic@chabant.com>

parents:

diff changeset

479

Mercurial > piecrust2

annotate piecrust/serving/server.py @ 374:fa3ee8a8ee2d