Improve rflink typing (#86538)

This commit is contained in:
Marc Mueller 2023-01-24 16:47:34 +01:00 committed by GitHub
parent 43f3b0f933
commit d684aa4225
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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: