From d684aa4225205efc50be680e02e2f9c8a96c57ad Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Tue, 24 Jan 2023 16:47:34 +0100 Subject: [PATCH] Improve `rflink` typing (#86538) --- homeassistant/components/rflink/__init__.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/rflink/__init__.py b/homeassistant/components/rflink/__init__.py index e703f0d09de..f5a0f0808da 100644 --- a/homeassistant/components/rflink/__init__.py +++ b/homeassistant/components/rflink/__init__.py @@ -6,7 +6,7 @@ from collections import defaultdict import logging import async_timeout -from rflink.protocol import create_rflink_connection +from rflink.protocol import ProtocolBase, create_rflink_connection from serial import SerialException import voluptuous as vol @@ -478,12 +478,16 @@ class RflinkCommand(RflinkDevice): # Keep repetition tasks to cancel if state is changed before repetitions # are sent - _repetition_task = None + _repetition_task: asyncio.Task[None] | None = None - _protocol = None + _protocol: ProtocolBase | None = None + + _wait_ack: bool | None = None @classmethod - def set_rflink_protocol(cls, protocol, wait_ack=None): + def set_rflink_protocol( + cls, protocol: ProtocolBase | None, wait_ack: bool | None = None + ) -> None: """Set the Rflink asyncio protocol as a class variable.""" cls._protocol = protocol if wait_ack is not None: