Ensure service calls are typed in remember-the-milk (#63018)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2022-01-02 20:43:38 +01:00 committed by GitHub
parent 724f5dbf1a
commit 9248865443
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -7,7 +7,7 @@ from rtmapi import Rtm, RtmRequestFailedException
import voluptuous as vol
from homeassistant.const import CONF_API_KEY, CONF_ID, CONF_NAME, CONF_TOKEN, STATE_OK
from homeassistant.core import HomeAssistant
from homeassistant.core import HomeAssistant, ServiceCall
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import Entity
from homeassistant.helpers.entity_component import EntityComponent
@ -270,7 +270,7 @@ class RememberTheMilk(Entity):
self._token_valid = True
return self._token_valid
def create_task(self, call):
def create_task(self, call: ServiceCall) -> None:
"""Create a new task on Remember The Milk.
You can use the smart syntax to define the attributes of a new task,
@ -278,7 +278,7 @@ class RememberTheMilk(Entity):
due date to today.
"""
try:
task_name = call.data.get(CONF_NAME)
task_name = call.data[CONF_NAME]
hass_id = call.data.get(CONF_ID)
rtm_id = None
if hass_id is not None:
@ -320,12 +320,10 @@ class RememberTheMilk(Entity):
self._name,
rtm_exception,
)
return False
return True
def complete_task(self, call):
def complete_task(self, call: ServiceCall) -> None:
"""Complete a task that was previously created by this component."""
hass_id = call.data.get(CONF_ID)
hass_id = call.data[CONF_ID]
rtm_id = self._rtm_config.get_rtm_id(self._name, hass_id)
if rtm_id is None:
_LOGGER.error(
@ -334,7 +332,7 @@ class RememberTheMilk(Entity):
hass_id,
self._name,
)
return False
return
try:
result = self._rtm_api.rtm.timelines.create()
timeline = result.timeline.value
@ -354,7 +352,6 @@ class RememberTheMilk(Entity):
self._name,
rtm_exception,
)
return True
@property
def name(self):