jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-22 04:37:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow iteration in python_script (#8134)

Browse Source 
* Allow iteration in python_script

* Add tests
This commit is contained in:
Paulus Schoutsen 2017-06-21 04:32:50 -07:00 committed by Pascal Vizeli
parent 4d2b79156d
commit 6398e92836
2 changed files with 16 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
homeassistant/components/python_script.py
View File

@ -92,6 +92,7 @@ def execute(hass, filename, source, data=None):
'_print_': StubPrinter,
'_getattr_': protected_getattr,
'_write_': full_write_guard,
'_getiter_': iter,
}
logger = logging.getLogger('{}.{}'.format(__name__, filename))
local = {

15
tests/components/test_python_script.py
View File

@ -149,3 +149,18 @@ hass.stop()
yield from hass.async_block_till_done()
assert "Not allowed to access HomeAssistant.stop" in caplog.text
@asyncio.coroutine
def test_iterating(hass):
"""Test compile error logs error."""
source = """
for i in [1, 2]:
hass.states.set('hello.{}'.format(i), 'world')
"""
hass.async_add_job(execute, hass, 'test.py', source, {})
yield from hass.async_block_till_done()
assert hass.states.is_state('hello.1', 'world')
assert hass.states.is_state('hello.2', 'world')