piecrust2: piecrust/templating/jinja/environment.py annotate

annotate piecrust/templating/jinja/environment.py @ 854:08e02c2a2a1a

core: Keep refactoring, this time to prepare for generator sources. - Make a few APIs simpler. - Content pipelines create their own jobs, so that generator sources can keep aborting in `getContents`, but rely on their pipeline to generate pages for baking.

author	Ludovic Chabant <ludovic@chabant.com>
date	Sun, 04 Jun 2017 23:34:28 -0700
parents	2c7e57d80bba
children	41b0c94f9833

rev	line source
851 2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	1 import re
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	2 import time
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	3 import email.utils
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	4 import hashlib
854 08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	5 import logging
851 2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	6 import strict_rfc3339
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 from jinja2 import Environment
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8 from .extensions import get_highlight_css
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	9 from piecrust.data.paginator import Paginator
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	10 from piecrust.rendering import format_text
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	11 from piecrust.uriutil import multi_replace
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	12
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	13
854 08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	14 logger = logging.getLogger(__name__)
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	15
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	16
851 2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	17 class PieCrustEnvironment(Environment):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	18 def __init__(self, app, args, *kwargs):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	19 self.app = app
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	20
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 # Before we create the base Environement, let's figure out the options
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	22 # we want to pass to it.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	23 twig_compatibility_mode = app.config.get('jinja/twig_compatibility')
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	24
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	25 # Disable auto-reload when we're baking.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	26 if app.config.get('baker/is_baking'):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	27 kwargs.setdefault('auto_reload', False)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	28
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	29 # Let the user override most Jinja options via the site config.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	30 for name in ['block_start_string', 'block_end_string',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	31 'variable_start_string', 'variable_end_string',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	32 'comment_start_string', 'comment_end_string',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	33 'line_statement_prefix', 'line_comment_prefix',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	34 'trim_blocks', 'lstrip_blocks',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	35 'newline_sequence', 'keep_trailing_newline']:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	36 val = app.config.get('jinja/' + name)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	37 if val is not None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	38 kwargs.setdefault(name, val)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	39
854 08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	40 # Undefined behaviour.
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	41 undef = app.config.get('jinja/undefined')
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	42 if undef == 'logging':
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	43 from jinja2 import make_logging_undefined
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	44 kwargs.setdefault('undefined',
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	45 make_logging_undefined(logger))
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	46 elif undef == 'strict':
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	47 from jinja2 import StrictUndefined
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	48 kwargs.setdefault('undefined', StrictUndefined)
08e02c2a2a1a core: Keep refactoring, this time to prepare for generator sources. Ludovic Chabant <ludovic@chabant.com> parents: 851 diff changeset	49
851 2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	50 # Twig trims blocks.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	51 if twig_compatibility_mode is True:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	52 kwargs['trim_blocks'] = True
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	53
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	54 # All good! Create the Environment.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	55 super(PieCrustEnvironment, self).__init__(args, *kwargs)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	56
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	57 # Now add globals and filters.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	58 self.globals.update({
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	59 'now': get_now_date(),
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	60 'fail': raise_exception,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	61 'highlight_css': get_highlight_css})
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	62
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	63 self.filters.update({
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	64 'keys': get_dict_keys,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	65 'values': get_dict_values,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	66 'paginate': self._paginate,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	67 'formatwith': self._formatWith,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	68 'markdown': lambda v: self._formatWith(v, 'markdown'),
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	69 'textile': lambda v: self._formatWith(v, 'textile'),
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	70 'nocache': add_no_cache_parameter,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	71 'wordcount': get_word_count,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	72 'stripoutertag': strip_outer_tag,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	73 'stripslash': strip_slash,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	74 'titlecase': title_case,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	75 'md5': make_md5,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	76 'atomdate': get_xml_date,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	77 'xmldate': get_xml_date,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	78 'emaildate': get_email_date,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	79 'date': get_date})
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	80
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	81 # Backwards compatibility with Twig.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	82 if twig_compatibility_mode is True:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	83 self.filters['raw'] = self.filters['safe']
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	84 self.globals['pcfail'] = raise_exception
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	85
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	86 def _paginate(self, value, items_per_page=5):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	87 cpi = self.app.env.exec_info_stack.current_page_info
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	88 if cpi is None or cpi.page is None or cpi.render_ctx is None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	89 raise Exception("Can't paginate when no page has been pushed "
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	90 "on the execution stack.")
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	91 return Paginator(cpi.page, value,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	92 page_num=cpi.render_ctx.page_num,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	93 items_per_page=items_per_page)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	94
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	95 def _formatWith(self, value, format_name):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	96 return format_text(self.app, format_name, value)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	97
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	98
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	99 def raise_exception(msg):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	100 raise Exception(msg)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	101
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	102
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	103 def get_dict_keys(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	104 if isinstance(value, list):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	105 return [i[0] for i in value]
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	106 return value.keys()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	107
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	108
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	109 def get_dict_values(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	110 if isinstance(value, list):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	111 return [i[1] for i in value]
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	112 return value.values()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	113
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	114
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	115 def add_no_cache_parameter(value, param_name='t', param_value=None):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	116 if not param_value:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	117 param_value = time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	118 if '?' in value:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	119 value += '&'
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	120 else:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	121 value += '?'
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	122 value += '%s=%s' % (param_name, param_value)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	123 return value
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	124
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	125
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	126 def get_word_count(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	127 return len(value.split())
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	128
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	129
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	130 def strip_outer_tag(value, tag=None):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	131 tag_pattern = '[a-z]+[a-z0-9]*'
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	132 if tag is not None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	133 tag_pattern = re.escape(tag)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	134 pat = r'^\<' + tag_pattern + r'\>(.*)\</' + tag_pattern + '>$'
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	135 m = re.match(pat, value)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	136 if m:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	137 return m.group(1)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	138 return value
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	139
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	140
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	141 def strip_slash(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	142 return value.rstrip('/')
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	143
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	144
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	145 def title_case(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	146 return value.title()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	147
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	148
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	149 def make_md5(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	150 return hashlib.md5(value.lower().encode('utf8')).hexdigest()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	151
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	152
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	153 def get_xml_date(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	154 """ Formats timestamps like 1985-04-12T23:20:50.52Z
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	155 """
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	156 if value == 'now':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	157 value = time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	158 return strict_rfc3339.timestamp_to_rfc3339_localoffset(int(value))
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	159
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	160
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	161 def get_email_date(value, localtime=False):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	162 """ Formats timestamps like Fri, 09 Nov 2001 01:08:47 -0000
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	163 """
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	164 if value == 'now':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	165 value = time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	166 return email.utils.formatdate(value, localtime=localtime)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	167
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	168
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	169 def get_now_date():
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	170 return time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	171
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	172
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	173 def get_date(value, fmt):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	174 if value == 'now':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	175 value = time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	176 if '%' not in fmt:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	177 suggest = php_format_to_strftime_format(fmt)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	178 if suggest != fmt:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	179 suggest_message = ("You probably want a format that looks "
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	180 "like: '%s'." % suggest)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	181 else:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	182 suggest_message = ("We can't suggest a proper date format "
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	183 "for you right now, though.")
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	184 raise Exception("Got incorrect date format: '%s\n"
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	185 "PieCrust 1 date formats won't work in PieCrust 2. "
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	186 "%s\n"
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	187 "Please check the `strftime` formatting page here: "
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	188 "https://docs.python.org/3/library/datetime.html"
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	189 "#strftime-and-strptime-behavior" %
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	190 (fmt, suggest_message))
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	191 return time.strftime(fmt, time.localtime(value))
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	192
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	193
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	194 def php_format_to_strftime_format(fmt):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	195 replacements = {
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	196 'd': '%d',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	197 'D': '%a',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	198 'j': '%d',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	199 'l': '%A',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	200 'w': '%w',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	201 'z': '%j',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	202 'W': '%W',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	203 'F': '%B',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	204 'm': '%m',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	205 'M': '%b',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	206 'n': '%m',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	207 'y': '%Y',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	208 'Y': '%y',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	209 'g': '%I',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	210 'G': '%H',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	211 'h': '%I',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	212 'H': '%H',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	213 'i': '%M',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	214 's': '%S',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	215 'e': '%Z',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	216 'O': '%z',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	217 'c': '%Y-%m-%dT%H:%M:%SZ'}
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	218 return multi_replace(fmt, replacements)

Mercurial > piecrust2

annotate piecrust/templating/jinja/environment.py @ 854:08e02c2a2a1a