Add datetime option to input_datetime.set_datetime service (#24975)
This commit is contained in:
parent
b11171aaeb
commit
1431fd6fbd
3 changed files with 103 additions and 27 deletions
|
@ -4,7 +4,8 @@ import datetime
|
|||
|
||||
import voluptuous as vol
|
||||
|
||||
from homeassistant.const import ATTR_ENTITY_ID, CONF_ICON, CONF_NAME
|
||||
from homeassistant.const import (
|
||||
ATTR_DATE, ATTR_ENTITY_ID, ATTR_TIME, CONF_ICON, CONF_NAME)
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
from homeassistant.helpers.entity_component import EntityComponent
|
||||
from homeassistant.helpers.restore_state import RestoreEntity
|
||||
|
@ -22,8 +23,7 @@ CONF_INITIAL = 'initial'
|
|||
|
||||
DEFAULT_VALUE = '1970-01-01 00:00:00'
|
||||
|
||||
ATTR_DATE = 'date'
|
||||
ATTR_TIME = 'time'
|
||||
ATTR_DATETIME = 'datetime'
|
||||
|
||||
SERVICE_SET_DATETIME = 'set_datetime'
|
||||
|
||||
|
@ -31,6 +31,7 @@ SERVICE_SET_DATETIME_SCHEMA = vol.Schema({
|
|||
vol.Optional(ATTR_ENTITY_ID): cv.entity_ids,
|
||||
vol.Optional(ATTR_DATE): cv.date,
|
||||
vol.Optional(ATTR_TIME): cv.time,
|
||||
vol.Optional(ATTR_DATETIME): cv.datetime,
|
||||
})
|
||||
|
||||
|
||||
|
@ -77,12 +78,19 @@ async def async_setup(hass, config):
|
|||
"""Handle a call to the input datetime 'set datetime' service."""
|
||||
time = call.data.get(ATTR_TIME)
|
||||
date = call.data.get(ATTR_DATE)
|
||||
if (entity.has_date and not date) or (entity.has_time and not time):
|
||||
dttm = call.data.get(ATTR_DATETIME)
|
||||
# pylint: disable=too-many-boolean-expressions
|
||||
if (dttm and (date or time)
|
||||
or entity.has_date and not (date or dttm)
|
||||
or entity.has_time and not (time or dttm)):
|
||||
_LOGGER.error("Invalid service data for %s "
|
||||
"input_datetime.set_datetime: %s",
|
||||
entity.entity_id, str(call.data))
|
||||
return
|
||||
|
||||
if dttm:
|
||||
date = dttm.date()
|
||||
time = dttm.time()
|
||||
entity.async_set_datetime(date, time)
|
||||
|
||||
component.async_register_entity_service(
|
||||
|
|
|
@ -3,7 +3,9 @@ set_datetime:
|
|||
fields:
|
||||
entity_id: {description: Entity id of the input datetime to set the new value.,
|
||||
example: input_datetime.test_date_time}
|
||||
date: {description: The target date the entity should be set to.,
|
||||
date: {description: The target date the entity should be set to. Do not use with datetime.,
|
||||
example: '"date": "2019-04-22"'}
|
||||
time: {description: The target time the entity should be set to.,
|
||||
time: {description: The target time the entity should be set to. Do not use with datetime.,
|
||||
example: '"time": "05:30:00"'}
|
||||
datetime: {description: The target date & time the entity should be set to. Do not use with date or time.,
|
||||
example: '"datetime": "2019-04-22 05:30:00"'}
|
||||
|
|
|
@ -9,17 +9,26 @@ import voluptuous as vol
|
|||
from homeassistant.core import CoreState, State, Context
|
||||
from homeassistant.setup import async_setup_component
|
||||
from homeassistant.components.input_datetime import (
|
||||
DOMAIN, ATTR_ENTITY_ID, ATTR_DATE, ATTR_TIME, SERVICE_SET_DATETIME)
|
||||
DOMAIN, ATTR_ENTITY_ID, ATTR_DATE, ATTR_DATETIME, ATTR_TIME,
|
||||
SERVICE_SET_DATETIME)
|
||||
|
||||
from tests.common import mock_restore_cache
|
||||
|
||||
|
||||
async def async_set_date_and_time(hass, entity_id, dt_value):
|
||||
"""Set date and / or time of input_datetime."""
|
||||
await hass.services.async_call(DOMAIN, SERVICE_SET_DATETIME, {
|
||||
ATTR_ENTITY_ID: entity_id,
|
||||
ATTR_DATE: dt_value.date(),
|
||||
ATTR_TIME: dt_value.time()
|
||||
}, blocking=True)
|
||||
|
||||
|
||||
async def async_set_datetime(hass, entity_id, dt_value):
|
||||
"""Set date and / or time of input_datetime."""
|
||||
await hass.services.async_call(DOMAIN, SERVICE_SET_DATETIME, {
|
||||
ATTR_ENTITY_ID: entity_id,
|
||||
ATTR_DATE: dt_value.date(),
|
||||
ATTR_TIME: dt_value.time()
|
||||
ATTR_DATETIME: dt_value
|
||||
}, blocking=True)
|
||||
|
||||
|
||||
|
@ -38,10 +47,9 @@ async def test_invalid_configs(hass):
|
|||
assert not await async_setup_component(hass, DOMAIN, {DOMAIN: cfg})
|
||||
|
||||
|
||||
@asyncio.coroutine
|
||||
def test_set_datetime(hass):
|
||||
"""Test set_datetime method."""
|
||||
yield from async_setup_component(hass, DOMAIN, {
|
||||
async def test_set_datetime(hass):
|
||||
"""Test set_datetime method using date & time."""
|
||||
await async_setup_component(hass, DOMAIN, {
|
||||
DOMAIN: {
|
||||
'test_datetime': {
|
||||
'has_time': True,
|
||||
|
@ -51,9 +59,9 @@ def test_set_datetime(hass):
|
|||
|
||||
entity_id = 'input_datetime.test_datetime'
|
||||
|
||||
dt_obj = datetime.datetime(2017, 9, 7, 19, 46)
|
||||
dt_obj = datetime.datetime(2017, 9, 7, 19, 46, 30)
|
||||
|
||||
yield from async_set_datetime(hass, entity_id, dt_obj)
|
||||
await async_set_date_and_time(hass, entity_id, dt_obj)
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state.state == str(dt_obj)
|
||||
|
@ -65,13 +73,43 @@ def test_set_datetime(hass):
|
|||
assert state.attributes['day'] == 7
|
||||
assert state.attributes['hour'] == 19
|
||||
assert state.attributes['minute'] == 46
|
||||
assert state.attributes['second'] == 30
|
||||
assert state.attributes['timestamp'] == dt_obj.timestamp()
|
||||
|
||||
|
||||
@asyncio.coroutine
|
||||
def test_set_datetime_time(hass):
|
||||
async def test_set_datetime_2(hass):
|
||||
"""Test set_datetime method using datetime."""
|
||||
await async_setup_component(hass, DOMAIN, {
|
||||
DOMAIN: {
|
||||
'test_datetime': {
|
||||
'has_time': True,
|
||||
'has_date': True
|
||||
},
|
||||
}})
|
||||
|
||||
entity_id = 'input_datetime.test_datetime'
|
||||
|
||||
dt_obj = datetime.datetime(2017, 9, 7, 19, 46, 30)
|
||||
|
||||
await async_set_datetime(hass, entity_id, dt_obj)
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state.state == str(dt_obj)
|
||||
assert state.attributes['has_time']
|
||||
assert state.attributes['has_date']
|
||||
|
||||
assert state.attributes['year'] == 2017
|
||||
assert state.attributes['month'] == 9
|
||||
assert state.attributes['day'] == 7
|
||||
assert state.attributes['hour'] == 19
|
||||
assert state.attributes['minute'] == 46
|
||||
assert state.attributes['second'] == 30
|
||||
assert state.attributes['timestamp'] == dt_obj.timestamp()
|
||||
|
||||
|
||||
async def test_set_datetime_time(hass):
|
||||
"""Test set_datetime method with only time."""
|
||||
yield from async_setup_component(hass, DOMAIN, {
|
||||
await async_setup_component(hass, DOMAIN, {
|
||||
DOMAIN: {
|
||||
'test_time': {
|
||||
'has_time': True,
|
||||
|
@ -81,24 +119,23 @@ def test_set_datetime_time(hass):
|
|||
|
||||
entity_id = 'input_datetime.test_time'
|
||||
|
||||
dt_obj = datetime.datetime(2017, 9, 7, 19, 46)
|
||||
dt_obj = datetime.datetime(2017, 9, 7, 19, 46, 30)
|
||||
time_portion = dt_obj.time()
|
||||
|
||||
yield from async_set_datetime(hass, entity_id, dt_obj)
|
||||
await async_set_date_and_time(hass, entity_id, dt_obj)
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state.state == str(time_portion)
|
||||
assert state.attributes['has_time']
|
||||
assert not state.attributes['has_date']
|
||||
|
||||
assert state.attributes['timestamp'] == (19 * 3600) + (46 * 60)
|
||||
assert state.attributes['timestamp'] == (19 * 3600) + (46 * 60) + 30
|
||||
|
||||
|
||||
@asyncio.coroutine
|
||||
def test_set_invalid(hass):
|
||||
async def test_set_invalid(hass):
|
||||
"""Test set_datetime method with only time."""
|
||||
initial = '2017-01-01'
|
||||
yield from async_setup_component(hass, DOMAIN, {
|
||||
await async_setup_component(hass, DOMAIN, {
|
||||
DOMAIN: {
|
||||
'test_date': {
|
||||
'has_time': False,
|
||||
|
@ -113,11 +150,40 @@ def test_set_invalid(hass):
|
|||
time_portion = dt_obj.time()
|
||||
|
||||
with pytest.raises(vol.Invalid):
|
||||
yield from hass.services.async_call('input_datetime', 'set_datetime', {
|
||||
await hass.services.async_call('input_datetime', 'set_datetime', {
|
||||
'entity_id': 'test_date',
|
||||
'time': time_portion
|
||||
})
|
||||
yield from hass.async_block_till_done()
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state.state == initial
|
||||
|
||||
|
||||
async def test_set_invalid_2(hass):
|
||||
"""Test set_datetime method with date and datetime."""
|
||||
initial = '2017-01-01'
|
||||
await async_setup_component(hass, DOMAIN, {
|
||||
DOMAIN: {
|
||||
'test_date': {
|
||||
'has_time': False,
|
||||
'has_date': True,
|
||||
'initial': initial
|
||||
}
|
||||
}})
|
||||
|
||||
entity_id = 'input_datetime.test_date'
|
||||
|
||||
dt_obj = datetime.datetime(2017, 9, 7, 19, 46)
|
||||
time_portion = dt_obj.time()
|
||||
|
||||
with pytest.raises(vol.Invalid):
|
||||
await hass.services.async_call('input_datetime', 'set_datetime', {
|
||||
'entity_id': 'test_date',
|
||||
'time': time_portion,
|
||||
'datetime': dt_obj
|
||||
})
|
||||
await hass.async_block_till_done()
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state.state == initial
|
||||
|
@ -139,7 +205,7 @@ def test_set_datetime_date(hass):
|
|||
dt_obj = datetime.datetime(2017, 9, 7, 19, 46)
|
||||
date_portion = dt_obj.date()
|
||||
|
||||
yield from async_set_datetime(hass, entity_id, dt_obj)
|
||||
yield from async_set_date_and_time(hass, entity_id, dt_obj)
|
||||
|
||||
state = hass.states.get(entity_id)
|
||||
assert state.state == str(date_portion)
|
||||
|
|
Loading…
Add table
Reference in a new issue