evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Integrate utility functions into restricted Python environment (#8427)

Browse source
This commit is contained in:
Open Home Automation 2017-07-11 03:42:42 +02:00 • committed by Paulus Schoutsen
parent 821d01f82c
commit fe4abc8454
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
homeassistant/components/python_script.py
View file

@ -61,6 +61,7 @@ def execute(hass, filename, source, data=None):
"""Execute Python source."""
from RestrictedPython import compile_restricted_exec
from RestrictedPython.Guards import safe_builtins, full_write_guard
from RestrictedPython.Utilities import utility_builtins
compiled = compile_restricted_exec(source, filename=filename)
@ -87,8 +88,10 @@ def execute(hass, filename, source, data=None):
return getattr(obj, name, default)
builtins = safe_builtins.copy()
builtins.update(utility_builtins)
restricted_globals = {
'__builtins__': safe_builtins,
'__builtins__': builtins,
'_print_': StubPrinter,
'_getattr_': protected_getattr,
'_write_': full_write_guard,