jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose current time in templates

Browse Source 
Fixes #1282
This commit is contained in:
Paulus Schoutsen 2016-02-20 20:58:01 -08:00
parent 7f81122af6
commit 6847dac582
2 changed files with 23 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
homeassistant/util/template.py
View File

@ -12,6 +12,7 @@ from jinja2.sandbox import ImmutableSandboxedEnvironment
from homeassistant.const import STATE_UNKNOWN from homeassistant.const import STATE_UNKNOWN
from homeassistant.exceptions import TemplateError from homeassistant.exceptions import TemplateError
import homeassistant.util.dt as dt_util
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
_SENTINEL = object() _SENTINEL = object()
@ -45,7 +46,9 @@ def render(hass, template, variables=None, **kwargs):
return ENV.from_string(template, { return ENV.from_string(template, {
'states': AllStates(hass), 'states': AllStates(hass),
'is_state': hass.states.is_state, 'is_state': hass.states.is_state,
'is_state_attr': hass.states.is_state_attr 'is_state_attr': hass.states.is_state_attr,
'now': dt_util.now,
'utcnow': dt_util.utcnow,
}).render(kwargs).strip() }).render(kwargs).strip()
except jinja2.TemplateError as err: except jinja2.TemplateError as err:
raise TemplateError(err) raise TemplateError(err)

19
tests/util/test_template.py
View File

@ -6,8 +6,11 @@ Tests Home Assistant template util methods.
""" """
# pylint: disable=too-many-public-methods # pylint: disable=too-many-public-methods
import unittest import unittest
from unittest.mock import patch
from homeassistant.exceptions import TemplateError from homeassistant.exceptions import TemplateError
from homeassistant.util import template from homeassistant.util import template
import homeassistant.util.dt as dt_util
from tests.common import get_test_home_assistant from tests.common import get_test_home_assistant
@ -143,3 +146,19 @@ class TestUtilTemplate(unittest.TestCase):
self.assertEqual( self.assertEqual(
'unknown', 'unknown',
template.render(self.hass, '{{ states("test.object2") }}')) template.render(self.hass, '{{ states("test.object2") }}'))
@patch('homeassistant.core.dt_util.now', return_value=dt_util.now())
@patch('homeassistant.util.template.TemplateEnvironment.is_safe_callable',
return_value=True)
def test_now_function(self, mock_is_safe, mock_now):
self.assertEqual(
dt_util.now().isoformat(),
template.render(self.hass, '{{ now().isoformat() }}'))
@patch('homeassistant.core.dt_util.utcnow', return_value=dt_util.utcnow())
@patch('homeassistant.util.template.TemplateEnvironment.is_safe_callable',
return_value=True)
def test_utcnow_function(self, mock_is_safe, mock_utcnow):
self.assertEqual(
dt_util.utcnow().isoformat(),
template.render(self.hass, '{{ utcnow().isoformat() }}'))