KittyPatch/kitty
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Compile template and restrict globals for eval

Browse Source
This commit is contained in:
chriszarate 2020-05-13 23:30:12 -04:00
parent d6ddfa9d18
commit 8bb8615a5a
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
kitty/tab_bar.py
View File

@ -54,14 +54,18 @@ def draw_title(draw_data: DrawData, screen: Screen, tab: TabBarData, index: int)
screen.cursor.fg = draw_data.bell_fg
screen.draw('🔔 ')
screen.cursor.fg = fg
layout_name = tab.layout_name
num_windows = tab.num_windows
template = draw_data.title_template
title = tab.title
if tab.is_active and draw_data.active_title_template is not None:
template = draw_data.active_title_template
try:
title = eval(f"f'{template}'")
eval_locals = {
'index': index,
'layout_name': tab.layout_name,
'num_windows': tab.num_windows,
'title': tab.title
}
compiled_template = compile('f"""' + template + '"""', '<tab_template>', 'eval')
title = eval(compiled_template, {'__builtins__': {}}, eval_locals)
except Exception as e:
if template not in template_failures:
template_failures.add(template)