##// END OF EJS Templates
hgweb: pre-init mimetypes module (fixes ugly bug in python-2.6.2 mimetypes)...
hgweb: pre-init mimetypes module (fixes ugly bug in python-2.6.2 mimetypes) The ugly initialization hack in mimetypes' initialization procedure can make hg serve hit the maximum recursion depth when in threaded mode.

File last commit:

r8223:02145b70 default
r8224:1075f5c1 default
Show More
templater.py
215 lines | 7.0 KiB | text/x-python | PythonLexer
Vadim Gelfer
add doc comments to template code.
r1909 # templater.py - template expansion for output
#
# Copyright 2005, 2006 Matt Mackall <mpm@selenic.com>
#
# This software may be used and distributed according to the terms
# of the GNU General Public License, incorporated herein by reference.
Matt Mackall
Simplify i18n imports
r3891 from i18n import _
Matt Mackall
templates: move filters to their own module...
r5976 import re, sys, os
Matt Mackall
templater: use new config parser...
r8194 from mercurial import util, config
Vadim Gelfer
use safer string parser for template engine.
r1901
Brendan Cully
Allow hgweb to search for templates in more than one path....
r7107 path = ['templates', '../templates']
Vadim Gelfer
make parsestring work with strings that do not have quotes.
r1902 def parsestring(s, quoted=True):
Vadim Gelfer
add doc comments to template code.
r1909 '''parse a string using simple c-like syntax.
string must be in quotes if quoted is True.'''
Vadim Gelfer
make parsestring work with strings that do not have quotes.
r1902 if quoted:
Matt Mackall
templater: simplify parsestring
r3639 if len(s) < 2 or s[0] != s[-1]:
raise SyntaxError(_('unmatched quotes'))
return s[1:-1].decode('string_escape')
Matt Mackall
templater: use str.decode in parse_string
r3632
return s.decode('string_escape')
Vadim Gelfer
move hgweb template code out to templater
r1896
Dirkjan Ochtman
templater: separate template management and actual string processing
r8218 class engine(object):
Vadim Gelfer
add doc comments to template code.
r1909 '''template expansion engine.
template expansion works like this. a map file contains key=value
pairs. if value is quoted, it is treated as string. otherwise, it
is treated as name of template file.
templater is asked to expand a key in map. it looks up key, and
TK Soh
minor typo fix in templater's docstring
r4334 looks for strings like this: {foo}. it expands {foo} by looking up
Vadim Gelfer
add doc comments to template code.
r1909 foo in map, and substituting it. expansion is recursive: it stops
when there is no more {foo} to replace.
expansion also allows formatting and filtering.
format uses key to expand each item in list. syntax is
{key%format}.
filter uses function to transform value. syntax is
{key|filter1|filter2|...}.'''
Matt Mackall
template: fold template() into __call__, minor optimizations...
r3638 template_re = re.compile(r"(?:(?:#(?=[\w\|%]+#))|(?:{(?=[\w\|%]+})))"
r"(\w+)(?:(?:%(\w+))|((?:\|\w+)*))[#}]")
Dirkjan Ochtman
templater: separate template management and actual string processing
r8218 def __init__(self, loader, filters={}, defaults={}):
self.loader = loader
self.filters = filters
self.defaults = defaults
def process(self, t, map):
'''Perform expansion. t is name of map element to expand. map contains
added elements for use during expansion. Is a generator.'''
tmpl = self.loader(t)
iters = [self._process(tmpl, map)]
while iters:
try:
item = iters[0].next()
except StopIteration:
iters.pop(0)
continue
if isinstance(item, str):
yield item
elif item is None:
yield ''
elif hasattr(item, '__iter__'):
iters.insert(0, iter(item))
else:
yield str(item)
def _process(self, tmpl, map):
'''Render a template. Returns a generator.'''
while tmpl:
m = self.template_re.search(tmpl)
if not m:
yield tmpl
break
start, end = m.span(0)
key, format, fl = m.groups()
if start:
yield tmpl[:start]
tmpl = tmpl[end:]
if key in map:
v = map[key]
else:
v = self.defaults.get(key, "")
if callable(v):
v = v(**map)
if format:
if not hasattr(v, '__iter__'):
raise SyntaxError(_("Error expanding '%s%%%s'")
% (key, format))
lm = map.copy()
for i in v:
lm.update(i)
yield self.process(format, lm)
else:
if fl:
for f in fl.split("|")[1:]:
v = self.filters[f](v)
yield v
class templater(object):
Brendan Cully
templater: return data in increasing chunk sizes...
r7396 def __init__(self, mapfile, filters={}, defaults={}, cache={},
minchunk=1024, maxchunk=65536):
Vadim Gelfer
add doc comments to template code.
r1909 '''set up template engine.
mapfile is name of file to read map definitions from.
filters is dict of functions. each transforms a value into another.
defaults is dict of default map definitions.'''
Vadim Gelfer
improve template errors when something is wrong.
r1905 self.mapfile = mapfile or 'template'
Shun-ichi Goto
Duplicate cache when creating templater.
r1975 self.cache = cache.copy()
Vadim Gelfer
move hgweb template code out to templater
r1896 self.map = {}
Vadim Gelfer
improve template errors when something is wrong.
r1905 self.base = (mapfile and os.path.dirname(mapfile)) or ''
Vadim Gelfer
move hgweb template code out to templater
r1896 self.filters = filters
Vadim Gelfer
fix template bug that made hgweb break....
r1964 self.defaults = defaults
Brendan Cully
templater: return data in increasing chunk sizes...
r7396 self.minchunk, self.maxchunk = minchunk, maxchunk
Vadim Gelfer
move hgweb template code out to templater
r1896
Vadim Gelfer
improve template errors when something is wrong.
r1905 if not mapfile:
return
Dirkjan Ochtman
give better error message on non-existent mapfile (issue813)
r6337 if not os.path.exists(mapfile):
raise util.Abort(_('style not found: %s') % mapfile)
Matt Mackall
templater: use new config parser...
r8194 conf = config.config()
conf.read(mapfile)
for key, val in conf[''].items():
if val[0] in "'\"":
try:
self.cache[key] = parsestring(val)
except SyntaxError, inst:
raise SyntaxError('%s: %s' %
Matt Mackall
config: getsource -> source
r8198 (conf.source('', key), inst.args[0]))
Vadim Gelfer
move hgweb template code out to templater
r1896 else:
Matt Mackall
templater: use new config parser...
r8194 self.map[key] = os.path.join(self.base, val)
Vadim Gelfer
move hgweb template code out to templater
r1896
Vadim Gelfer
many small changes to templater....
r1899 def __contains__(self, key):
Matt Mackall
templater: simplify cache and remove filter argument in __call__
r3637 return key in self.cache or key in self.map
Vadim Gelfer
many small changes to templater....
r1899
Dirkjan Ochtman
templater: separate template management and actual string processing
r8218 def load(self, t):
Dirkjan Ochtman
templater: make a template a string-only iterator
r6783 '''Get the template for the given template name. Use a local cache.'''
Christian Ebert
Prefer i in d over d.has_key(i)
r5915 if not t in self.cache:
Vadim Gelfer
improve template errors when something is wrong.
r1905 try:
Matt Mackall
templater: simplify cache and remove filter argument in __call__
r3637 self.cache[t] = file(self.map[t]).read()
Vadim Gelfer
improve template errors when something is wrong.
r1905 except IOError, inst:
raise IOError(inst.args[0], _('template file %s: %s') %
(self.map[t], inst.args[1]))
Dirkjan Ochtman
templater: make a template a string-only iterator
r6783 return self.cache[t]
Vadim Gelfer
move hgweb template code out to templater
r1896
Dirkjan Ochtman
templater: make a template a string-only iterator
r6783 def __call__(self, t, **map):
Dirkjan Ochtman
templater: separate template management and actual string processing
r8218 proc = engine(self.load, self.filters, self.defaults)
stream = proc.process(t, map)
Brendan Cully
templater: return data in increasing chunk sizes...
r7396 if self.minchunk:
stream = util.increasingchunks(stream, min=self.minchunk,
max=self.maxchunk)
return stream
Dirkjan Ochtman
kill some trailing spaces
r7434
Vadim Gelfer
many small changes to templater....
r1899 def templatepath(name=None):
Vadim Gelfer
add doc comments to template code.
r1909 '''return location of template file or directory (if no name).
returns None if not found.'''
Brendan Cully
Allow hgweb to search for templates in more than one path....
r7107 normpaths = []
Vadim Gelfer
move changeset_templater into templater module.
r2189
Lee Cantey
Windows py2exe version didn't handle names given to templatepath() correctly
r2001 # executable version (py2exe) doesn't support __file__
if hasattr(sys, 'frozen'):
module = sys.executable
else:
module = __file__
Brendan Cully
Allow hgweb to search for templates in more than one path....
r7107 for f in path:
if f.startswith('/'):
p = f
else:
fl = f.split('/')
p = os.path.join(os.path.dirname(module), *fl)
if name:
p = os.path.join(p, name)
if name and os.path.exists(p):
Vadim Gelfer
move hgweb.templatepath into templater
r1897 return os.path.normpath(p)
Brendan Cully
Allow hgweb to search for templates in more than one path....
r7107 elif os.path.isdir(p):
normpaths.append(os.path.normpath(p))
return normpaths
Vadim Gelfer
move changeset_templater into templater module.
r2189
Dirkjan Ochtman
templater: move stylemap function from hgweb to templater
r7966 def stylemap(style, paths=None):
"""Return path to mapfile for a given style.
Searches mapfile in the following locations:
1. templatepath/style/map
2. templatepath/map-style
3. templatepath/map
"""
if paths is None:
paths = templatepath()
elif isinstance(paths, str):
Dirkjan Ochtman
templater: fix little problem from stylemap() changes
r8223 paths = [paths]
Dirkjan Ochtman
templater: move stylemap function from hgweb to templater
r7966
locations = style and [os.path.join(style, "map"), "map-" + style] or []
locations.append("map")
for path in paths:
for location in locations:
mapfile = os.path.join(path, location)
if os.path.isfile(mapfile):
return mapfile
raise RuntimeError("No hgweb templates found in %r" % paths)
Matt Mackall
templates: move filters to their own module...
r5976 def stringify(thing):
'''turn nested template iterator into string.'''
Dirkjan Ochtman
templatefilters: strings are iterators, but shouldn't be joined anyway
r8015 if hasattr(thing, '__iter__') and not isinstance(thing, str):
Matt Mackall
templates: move filters to their own module...
r5976 return "".join([stringify(t) for t in thing if t is not None])
return str(thing)