Small performance & clarity improvement

This commit is contained in:
jaseg 2013-08-17 20:10:20 +02:00
parent bcf2ffdb8f
commit ea24fc5026

View file

@ -73,5 +73,6 @@ def render_pony(name, text, balloonstyle, width=80, center=False, centertext=Fal
ponywidth = max([ len(re.sub(r'\x1B\[[0-9;]+m|\$.*\$', '', line)) for line in pony ])
indent = ' '*int((width-ponywidth)/2)
wre = re.compile('((\x1B\[[0-9;]+m)*.){0,%s}' % width)
return ''.join([ indent+wre.search(line).group()+'\n' for line in pony ])
reset = '\n'
return indent+(reset+indent).join([ wre.search(line).group() for line in pony ])+reset