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

author	Ludovic Chabant <ludovic@chabant.com>
date	Sat, 29 Apr 2017 21:42:22 -0700
parents
children	08e02c2a2a1a

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
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	5 import strict_rfc3339
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	6 from jinja2 import Environment
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	7 from .extensions import get_highlight_css
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	8 from piecrust.data.paginator import Paginator
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	9 from piecrust.rendering import format_text
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	10 from piecrust.uriutil import multi_replace
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	11
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 class PieCrustEnvironment(Environment):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	14 def __init__(self, app, args, *kwargs):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	15 self.app = app
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	16
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	17 # 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	18 # we want to pass to it.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	19 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	20
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	21 # 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	22 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	23 kwargs.setdefault('auto_reload', False)
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 # 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	26 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	27 'variable_start_string', 'variable_end_string',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	28 'comment_start_string', 'comment_end_string',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	29 'line_statement_prefix', 'line_comment_prefix',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	30 'trim_blocks', 'lstrip_blocks',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	31 'newline_sequence', 'keep_trailing_newline']:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	32 val = app.config.get('jinja/' + name)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	33 if val is not None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	34 kwargs.setdefault(name, val)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	35
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	36 # Twig trims blocks.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	37 if twig_compatibility_mode is True:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	38 kwargs['trim_blocks'] = True
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	39
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	40 # All good! Create the Environment.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	41 super(PieCrustEnvironment, self).__init__(args, *kwargs)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	42
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	43 # Now add globals and filters.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	44 self.globals.update({
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	45 'now': get_now_date(),
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	46 'fail': raise_exception,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	47 'highlight_css': get_highlight_css})
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	48
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	49 self.filters.update({
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	50 'keys': get_dict_keys,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	51 'values': get_dict_values,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	52 'paginate': self._paginate,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	53 'formatwith': self._formatWith,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	54 '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	55 '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	56 'nocache': add_no_cache_parameter,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	57 'wordcount': get_word_count,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	58 'stripoutertag': strip_outer_tag,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	59 'stripslash': strip_slash,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	60 'titlecase': title_case,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	61 'md5': make_md5,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	62 'atomdate': get_xml_date,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	63 'xmldate': get_xml_date,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	64 'emaildate': get_email_date,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	65 'date': get_date})
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	66
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	67 # Backwards compatibility with Twig.
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	68 if twig_compatibility_mode is True:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	69 self.filters['raw'] = self.filters['safe']
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	70 self.globals['pcfail'] = raise_exception
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	71
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	72 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	73 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	74 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	75 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	76 "on the execution stack.")
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	77 return Paginator(cpi.page, value,
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	78 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	79 items_per_page=items_per_page)
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 def _formatWith(self, value, format_name):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	82 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	83
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	84
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	85 def raise_exception(msg):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	86 raise Exception(msg)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	87
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	88
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	89 def get_dict_keys(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	90 if isinstance(value, list):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	91 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	92 return value.keys()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	93
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 get_dict_values(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	96 if isinstance(value, list):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	97 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	98 return value.values()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	99
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	100
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	101 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	102 if not param_value:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	103 param_value = time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	104 if '?' in value:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	105 value += '&'
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	106 else:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	107 value += '?'
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	108 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	109 return value
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	110
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	111
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	112 def get_word_count(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	113 return len(value.split())
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
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	116 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	117 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	118 if tag is not None:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	119 tag_pattern = re.escape(tag)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	120 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	121 m = re.match(pat, value)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	122 if m:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	123 return m.group(1)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	124 return value
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
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	127 def strip_slash(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	128 return value.rstrip('/')
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
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	131 def title_case(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	132 return value.title()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	133
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	134
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	135 def make_md5(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	136 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	137
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	138
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	139 def get_xml_date(value):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	140 """ 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	141 """
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	142 if value == 'now':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	143 value = time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	144 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	145
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	146
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	147 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	148 """ 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	149 """
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	150 if value == 'now':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	151 value = time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	152 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	153
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	154
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	155 def get_now_date():
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	156 return time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	157
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	158
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	159 def get_date(value, fmt):
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	160 if value == 'now':
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	161 value = time.time()
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	162 if '%' not in fmt:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	163 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	164 if suggest != fmt:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	165 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	166 "like: '%s'." % suggest)
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	167 else:
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	168 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	169 "for you right now, though.")
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	170 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	171 "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	172 "%s\n"
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	173 "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	174 "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	175 "#strftime-and-strptime-behavior" %
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	176 (fmt, suggest_message))
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	177 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	178
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	179
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	180 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	181 replacements = {
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	182 'd': '%d',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	183 'D': '%a',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	184 'j': '%d',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	185 'l': '%A',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	186 'w': '%w',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	187 'z': '%j',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	188 'W': '%W',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	189 'F': '%B',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	190 'm': '%m',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	191 'M': '%b',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	192 'n': '%m',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	193 'y': '%Y',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	194 'Y': '%y',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	195 'g': '%I',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	196 'G': '%H',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	197 'h': '%I',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	198 'H': '%H',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	199 'i': '%M',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	200 's': '%S',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	201 'e': '%Z',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	202 'O': '%z',
2c7e57d80bba optimize: Don't load Jinja unless we need to. Ludovic Chabant <ludovic@chabant.com> parents: diff changeset	203 '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	204 return multi_replace(fmt, replacements)

851

2c7e57d80bba optimize: Don't load Jinja unless we need to.