1
mirror of https://github.com/home-assistant/core synced 2024-07-18 12:02:20 +02:00

Add lock typing in freedompro (#73544)

This commit is contained in:
epenet 2022-06-16 10:40:55 +02:00 committed by GitHub
parent 521d52a8b9
commit 7731cfd978
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,5 +1,6 @@
"""Support for Freedompro lock."""
import json
from typing import Any
from pyfreedompro import put_state
@ -75,10 +76,10 @@ class Device(CoordinatorEntity, LockEntity):
await super().async_added_to_hass()
self._handle_coordinator_update()
async def async_lock(self, **kwargs):
async def async_lock(self, **kwargs: Any) -> None:
"""Async function to lock the lock."""
payload = {"lock": 1}
payload = json.dumps(payload)
payload_dict = {"lock": 1}
payload = json.dumps(payload_dict)
await put_state(
self._session,
self._api_key,
@ -87,10 +88,10 @@ class Device(CoordinatorEntity, LockEntity):
)
await self.coordinator.async_request_refresh()
async def async_unlock(self, **kwargs):
async def async_unlock(self, **kwargs: Any) -> None:
"""Async function to unlock the lock."""
payload = {"lock": 0}
payload = json.dumps(payload)
payload_dict = {"lock": 0}
payload = json.dumps(payload_dict)
await put_state(
self._session,
self._api_key,