Backup_Repos/beets
1
0
Fork 0
mirror of https://github.com/beetbox/beets.git synced 2026-01-05 23:43:31 +01:00
Code Issues Projects Releases Wiki Activity

fix Unicode identifiers; add performance test

Browse source
This commit is contained in:
Adrian Sampson 2012-05-03 15:16:28 -07:00
parent a8338b6676
commit f6e06813ca
1 changed files with 23 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
beets/util/functemplate.py
View file

@ -152,8 +152,8 @@ class Symbol(object):
def translate(self):
"""Compile the variable lookup."""
expr = ex_rvalue(VARIABLE_PREFIX + self.ident)
return [expr], set([self.ident]), set()
expr = ex_rvalue(VARIABLE_PREFIX + self.ident.encode('utf8'))
return [expr], set([self.ident.encode('utf8')]), set()
class Call(object):
"""A function call in a template."""
@ -185,7 +185,7 @@ class Call(object):
def translate(self):
"""Compile the function call."""
varnames = set()
funcnames = set([self.ident])
funcnames = set([self.ident.encode('utf8')])
arg_exprs = []
for arg in self.args:
@ -207,7 +207,7 @@ class Call(object):
))
subexpr_call = ex_call(
FUNCTION_PREFIX + self.ident,
FUNCTION_PREFIX + self.ident.encode('utf8'),
arg_exprs
)
return [subexpr_call], varnames, funcnames
@ -523,3 +523,22 @@ class Template(object):
return u''.join(parts)
return wrapper_func
# Performance tests.
if __name__ == '__main__':
import timeit
_tmpl = Template(u'foo $bar %baz{foozle $bar barzle} $bar')
_vars = {'bar': 'qux'}
_funcs = {'baz': unicode.upper}
_compiled = _tmpl.translate()
interp_time = timeit.timeit('_tmpl.substitute(_vars, _funcs)',
'from __main__ import _tmpl, _vars, _funcs',
number=10000)
print interp_time
comp_time = timeit.timeit('_compiled(_vars, _funcs)',
'from __main__ import _compiled, _vars, _funcs',
number=10000)
print comp_time
print 'Speedup:', interp_time / comp_time