piecrust2: piecrust/serving/server.py annotate

annotate piecrust/serving/server.py @ 1145:e94737572542

serve: Fix an issue where false positive matches were rendered as the requested page. Now we try to render the page, but also try to detect for the most common "empty" pages.

author	Ludovic Chabant <ludovic@chabant.com>
date	Tue, 05 Jun 2018 22:08:51 -0700
parents	45ad976712ec
children

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 gzip
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	4 import time
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	5 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	6 import hashlib
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 import logging
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8 from werkzeug.exceptions import (
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	9 NotFound, MethodNotAllowed, InternalServerError, HTTPException)
374 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.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	11 from jinja2 import FileSystemLoader, Environment
409 2bb5327c4c1f debug: Fix serving of resources now that the module moved to a sub-folder. Ludovic Chabant <ludovic@chabant.com> parents: 391 diff changeset	12 from piecrust import CACHE_DIR, RESOURCES_DIR
1145 e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	13 from piecrust.page import PageNotFoundError
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	14 from piecrust.rendering import RenderingContext, render_page
555 daf8df5ade7d serve: Refactor the server to make pieces usable by the debugging middleware. Ludovic Chabant <ludovic@chabant.com> parents: 554 diff changeset	15 from piecrust.routing import RouteNotFoundError
daf8df5ade7d serve: Refactor the server to make pieces usable by the debugging middleware. Ludovic Chabant <ludovic@chabant.com> parents: 554 diff changeset	16 from piecrust.serving.util import (
1145 e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	17 content_type_map, make_wrapped_file_response, get_requested_pages,
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	18 get_app_for_server)
555 daf8df5ade7d serve: Refactor the server to make pieces usable by the debugging middleware. Ludovic Chabant <ludovic@chabant.com> parents: 554 diff changeset	19 from piecrust.sources.base import SourceNotFoundError
374 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
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22 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	23
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	24
917 33a89139c284 serve: Add `--admin` option to run the administration panel. Ludovic Chabant <ludovic@chabant.com> parents: 862 diff changeset	25 class PieCrustServer(object):
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	26 """ A WSGI application that serves a PieCrust website.
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	27 """
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	28 def __init__(self, appfactory, **kwargs):
917 33a89139c284 serve: Add `--admin` option to run the administration panel. Ludovic Chabant <ludovic@chabant.com> parents: 862 diff changeset	29 self.server = _ServerImpl(appfactory, **kwargs)
553 cc6f3dbe3048 serve: Extract some of the server's functionality into WSGI middlewares. Ludovic Chabant <ludovic@chabant.com> parents: 552 diff changeset	30
cc6f3dbe3048 serve: Extract some of the server's functionality into WSGI middlewares. Ludovic Chabant <ludovic@chabant.com> parents: 552 diff changeset	31 def __call__(self, environ, start_response):
cc6f3dbe3048 serve: Extract some of the server's functionality into WSGI middlewares. Ludovic Chabant <ludovic@chabant.com> parents: 552 diff changeset	32 return self.server._run_request(environ, start_response)
cc6f3dbe3048 serve: Extract some of the server's functionality into WSGI middlewares. Ludovic Chabant <ludovic@chabant.com> parents: 552 diff changeset	33
cc6f3dbe3048 serve: Extract some of the server's functionality into WSGI middlewares. Ludovic Chabant <ludovic@chabant.com> parents: 552 diff changeset	34
517 c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	35 class MultipleNotFound(HTTPException):
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	36 """ Represents a 404 (not found) error that tried to serve one or
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	37 more pages. It will report which pages it tried to serve
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	38 before failing.
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	39 """
517 c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	40 code = 404
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	41
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	42 def __init__(self, description, nfes):
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	43 super(MultipleNotFound, self).__init__(description)
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	44 self._nfes = nfes
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	45
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	46 def get_description(self, environ=None):
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	47 from werkzeug.utils import escape
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	48 desc = '<p>' + self.description + '</p>'
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	49 desc += '<p>'
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	50 for nfe in self._nfes:
555 daf8df5ade7d serve: Refactor the server to make pieces usable by the debugging middleware. Ludovic Chabant <ludovic@chabant.com> parents: 554 diff changeset	51 desc += '<li>' + escape(str(nfe)) + '</li>'
517 c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	52 desc += '</p>'
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	53 return desc
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	54
c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	55
917 33a89139c284 serve: Add `--admin` option to run the administration panel. Ludovic Chabant <ludovic@chabant.com> parents: 862 diff changeset	56 class _ServerImpl(object):
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	57 """ The PieCrust server.
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	58 """
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	59 def __init__(self, appfactory,
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	60 enable_debug_info=True,
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	61 root_url='/',
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	62 static_preview=True):
81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	63 self.appfactory = appfactory
375 aade4ea57e7f serve: Add ability to suppress the debug info window programmatically. Ludovic Chabant <ludovic@chabant.com> parents: 374 diff changeset	64 self.enable_debug_info = enable_debug_info
575 657384f08ca3 serve: Make it possible to preview pages with a custom root URL. Ludovic Chabant <ludovic@chabant.com> parents: 555 diff changeset	65 self.root_url = root_url
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	66 self.static_preview = static_preview
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	67 self._out_dir = os.path.join(
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	68 appfactory.root_dir,
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	69 CACHE_DIR,
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	70 (appfactory.cache_key or 'default'),
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	71 'server')
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	72
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	73 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	74 try:
552 9612cfc6455a serve: Rewrite of the Server-Sent Event code for build notifications. Ludovic Chabant <ludovic@chabant.com> parents: 551 diff changeset	75 response = self._try_run_request(environ)
553 cc6f3dbe3048 serve: Extract some of the server's functionality into WSGI middlewares. Ludovic Chabant <ludovic@chabant.com> parents: 552 diff changeset	76 return response(environ, start_response)
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	77 except Exception as ex:
680 c2ea75e37540 serve: Fix some crashes introduced by recent refactor. Ludovic Chabant <ludovic@chabant.com> parents: 666 diff changeset	78 if self.appfactory.debug:
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	79 raise
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	80 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	81
552 9612cfc6455a serve: Rewrite of the Server-Sent Event code for build notifications. Ludovic Chabant <ludovic@chabant.com> parents: 551 diff changeset	82 def _try_run_request(self, environ):
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	83 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	84
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	85 # 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	86 # 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	87 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	88 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	89 request.method)
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	90 raise MethodNotAllowed()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	91
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	92 # Handle requests to a pipeline-built asset right away.
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	93 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	94 if response is not None:
552 9612cfc6455a serve: Rewrite of the Server-Sent Event code for build notifications. Ludovic Chabant <ludovic@chabant.com> parents: 551 diff changeset	95 return response
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	96
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	97 # Same for page assets.
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	98 response = self._try_serve_page_asset(
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	99 self.appfactory.root_dir, environ, request)
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	100 if response is not None:
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	101 return response
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	102
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	103 # Create the app for this request.
666 81d9c3a3a0b5 internal: Get rid of the whole "sub cache" business. Ludovic Chabant <ludovic@chabant.com> parents: 663 diff changeset	104 app = get_app_for_server(self.appfactory,
575 657384f08ca3 serve: Make it possible to preview pages with a custom root URL. Ludovic Chabant <ludovic@chabant.com> parents: 555 diff changeset	105 root_url=self.root_url)
979 45ad976712ec tests: Big push to get the tests to pass again. Ludovic Chabant <ludovic@chabant.com> parents: 922 diff changeset	106 if (app.config.get('server/enable_debug_info') and
375 aade4ea57e7f serve: Add ability to suppress the debug info window programmatically. Ludovic Chabant <ludovic@chabant.com> parents: 374 diff changeset	107 self.enable_debug_info and
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	108 '!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	109 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	110
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	111 # Let's try to serve a page.
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	112 try:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	113 response = self._try_serve_page(app, environ, request)
552 9612cfc6455a serve: Rewrite of the Server-Sent Event code for build notifications. Ludovic Chabant <ludovic@chabant.com> parents: 551 diff changeset	114 return response
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	115 except (RouteNotFoundError, SourceNotFoundError) as ex:
517 c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	116 raise NotFound() from ex
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	117 except HTTPException:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	118 raise
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	119 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	120 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	121 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	122 raise
523 b22e69ff54f4 serve: Don't show the same error message twice. Ludovic Chabant <ludovic@chabant.com> parents: 520 diff changeset	123 logger.error(str(ex))
b22e69ff54f4 serve: Don't show the same error message twice. Ludovic Chabant <ludovic@chabant.com> parents: 520 diff changeset	124 msg = "There was an error trying to serve: %s" % request.path
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	125 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	126
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	127 def _try_serve_asset(self, environ, request):
575 657384f08ca3 serve: Make it possible to preview pages with a custom root URL. Ludovic Chabant <ludovic@chabant.com> parents: 555 diff changeset	128 offset = len(self.root_url)
657384f08ca3 serve: Make it possible to preview pages with a custom root URL. Ludovic Chabant <ludovic@chabant.com> parents: 555 diff changeset	129 rel_req_path = request.path[offset:].replace('/', os.sep)
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	130 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	131 # 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	132 # like LESS CSS map files.
913 e053a27e5aae server: Fix crash when serving temp files like CSS maps. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	133 full_path = os.path.join(self.appfactory.root_dir, rel_req_path)
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	134 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	135 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	136
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	137 try:
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	138 return make_wrapped_file_response(environ, request, full_path)
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	139 except OSError:
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	140 return None
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	141
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	142 def _try_serve_page_asset(self, app_root_dir, environ, request):
575 657384f08ca3 serve: Make it possible to preview pages with a custom root URL. Ludovic Chabant <ludovic@chabant.com> parents: 555 diff changeset	143 if not request.path.startswith(self.root_url + '_asset/'):
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	144 return None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	145
575 657384f08ca3 serve: Make it possible to preview pages with a custom root URL. Ludovic Chabant <ludovic@chabant.com> parents: 555 diff changeset	146 offset = len(self.root_url + '_asset/')
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	147 full_path = os.path.join(app_root_dir, request.path[offset:])
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	148
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	149 try:
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	150 return make_wrapped_file_response(environ, request, full_path)
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	151 except OSError:
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	152 return None
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	153
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	154 def _try_serve_page(self, app, environ, request):
555 daf8df5ade7d serve: Refactor the server to make pieces usable by the debugging middleware. Ludovic Chabant <ludovic@chabant.com> parents: 554 diff changeset	155 # Find a matching page.
1145 e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	156 req_pages, not_founds = get_requested_pages(app, request.path)
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	157
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	158 rendered_page = None
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	159
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	160 for req_page in req_pages:
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	161 # We have a page, let's try to render it.
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	162 render_ctx = RenderingContext(req_page.page,
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	163 sub_num=req_page.sub_num,
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	164 force_render=True)
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	165 req_page.page.source.prepareRenderContext(render_ctx)
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	166
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	167 # Render the page.
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	168 this_rendered_page = render_page(render_ctx)
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	169
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	170 # We might have rendered a page that technically exists, but
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	171 # has no interesting content, like a tag page for a tag that
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	172 # isn't used by any page. To eliminate these false positives,
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	173 # we check if there was pagination used, and if so, if it had
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	174 # anything in it.
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	175 # TODO: we might need a more generic system for other cases.
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	176 render_info = this_rendered_page.render_info
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	177 if (render_info['used_pagination'] and
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	178 not render_info['pagination_has_items']):
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	179 not_founds.append(PageNotFoundError(
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	180 ("Rendered '%s' (page %d) in source '%s' "
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	181 "but got empty content:\n\n%s\n\n") %
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	182 (req_page.req_path, req_page.sub_num,
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	183 req_page.page.source.name, this_rendered_page.content)))
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	184 continue
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	185
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	186 rendered_page = this_rendered_page
e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	187 break
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	188
555 daf8df5ade7d serve: Refactor the server to make pieces usable by the debugging middleware. Ludovic Chabant <ludovic@chabant.com> parents: 554 diff changeset	189 # If we haven't found any good match, report all the places we didn't
daf8df5ade7d serve: Refactor the server to make pieces usable by the debugging middleware. Ludovic Chabant <ludovic@chabant.com> parents: 554 diff changeset	190 # find it at.
1145 e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	191 if rendered_page is None:
555 daf8df5ade7d serve: Refactor the server to make pieces usable by the debugging middleware. Ludovic Chabant <ludovic@chabant.com> parents: 554 diff changeset	192 msg = "Can't find path for '%s':" % request.path
1145 e94737572542 serve: Fix an issue where false positive matches were rendered as the requested page. Ludovic Chabant <ludovic@chabant.com> parents: 979 diff changeset	193 raise MultipleNotFound(msg, not_founds)
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	194
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	195 # 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	196 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	197 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	198
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	199 # Profiling.
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	200 if app.config.get('site/show_debug_info'):
554 155c7e20414f serve: Fix timing information in the debug window. Ludovic Chabant <ludovic@chabant.com> parents: 553 diff changeset	201 now_time = time.perf_counter()
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	202 timing_info = (
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	203 '%8.1f ms' %
4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	204 ((now_time - app.env.start_time) * 1000.0))
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	205 rp_content = rp_content.replace(
852 4850f8c21b6e core: Start of the big refactor for PieCrust 3.0. Ludovic Chabant <ludovic@chabant.com> parents: 711 diff changeset	206 '__PIECRUST_TIMING_INFORMATION__', timing_info)
374 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 # 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	209 response = Response()
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	210
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	211 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	212 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	213 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	214 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	215
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	216 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	217 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	218
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	219 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	220 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	221 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	222 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	223 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	224 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	225 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	226 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	227 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	228 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	229
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	230 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	231 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	232 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	233 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	234 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	235 if mimetype:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	236 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	237
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	238 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	239 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	240 try:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	241 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	242 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	243 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	244 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	245 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	246 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	247 except Exception:
523 b22e69ff54f4 serve: Don't show the same error message twice. Ludovic Chabant <ludovic@chabant.com> parents: 520 diff changeset	248 logger.error("Error compressing response, "
b22e69ff54f4 serve: Don't show the same error message twice. Ludovic Chabant <ludovic@chabant.com> parents: 520 diff changeset	249 "falling back to uncompressed.")
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	250 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	251
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	252 return response
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	253
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	254 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	255 code = 500
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	256 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	257 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	258
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	259 path = 'error'
517 c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	260 if isinstance(exception, (NotFound, MultipleNotFound)):
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	261 path += '404'
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 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	264
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	265 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	266 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	267 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	268 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	269 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	270 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	271
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	272 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	273 desc = []
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	274 while exception is not None:
517 c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	275 if isinstance(exception, MultipleNotFound):
663 3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 575 diff changeset	276 desc += [str(e) for e in exception._nfes]
517 c7e8b4a5afe3 serve: Improve error reporting when pages are not found. Ludovic Chabant <ludovic@chabant.com> parents: 515 diff changeset	277 elif isinstance(exception, HTTPException):
663 3ceeca7bb71c themes: Add support for a `--theme` argument to `chef`. Ludovic Chabant <ludovic@chabant.com> parents: 575 diff changeset	278 desc.append(exception.get_description())
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	279 else:
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	280 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	281
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	282 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	283 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	284 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	285 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	286 return desc
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	287
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	288
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	289 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	290 def __init__(self):
409 2bb5327c4c1f debug: Fix serving of resources now that the module moved to a sub-folder. Ludovic Chabant <ludovic@chabant.com> parents: 391 diff changeset	291 base_dir = os.path.join(RESOURCES_DIR, 'messages')
374 fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	292 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	293
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	294 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	295 template += '.html'
fa3ee8a8ee2d serve: Split the server code in a couple modules inside a `serving` package. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	296 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	297

Mercurial > piecrust2

annotate piecrust/serving/server.py @ 1145:e94737572542