androidtvremote2 0.2.2__tar.gz → 0.3.0__tar.gz

{androidtvremote2-0.2.2/src/androidtvremote2.egg-info → androidtvremote2-0.3.0}/PKG-INFO

@@ -1,9 +1,9 @@
 Metadata-Version: 2.4
 Name: androidtvremote2
-Version: 0.2.2
+Version: 0.3.0
 Summary: A Python library for interacting with Android TV using the Android TV Remote protocol v2
 Author-email: tronikos <tronikos@gmail.com>
-License: Apache-2.0
+License-Expression: Apache-2.0
 Project-URL: Homepage, https://github.com/tronikos/androidtvremote2
 Project-URL: Bug Tracker, https://github.com/tronikos/androidtvremote2/issues
 Requires-Python: >=3.10
@@ -12,6 +12,12 @@ License-File: LICENSE
 Requires-Dist: aiofiles>=0.8
 Requires-Dist: cryptography>=3
 Requires-Dist: protobuf>=4.21
+Provides-Extra: test
+Requires-Dist: pytest; extra == "test"
+Provides-Extra: demo
+Requires-Dist: pyaudio; extra == "demo"
+Requires-Dist: pynput; extra == "demo"
+Requires-Dist: zeroconf; extra == "demo"
 Dynamic: license-file
 
 # androidtvremote2
@@ -19,9 +25,11 @@ Dynamic: license-file
 A Python library for interacting with Android TV using the Android TV Remote protocol v2. This is the same protocol the Google TV mobile app is using. It doesn't require ADB or enabling developer tools on the Android TV device. It only requires the [Android TV Remote Service](https://play.google.com/store/apps/details?id=com.google.android.tv.remote.service) that comes pre-installed on most Android TV devices.
 
 For a list of the most common commands you can send to the Android TV see: [TvKeys](https://github.com/tronikos/androidtvremote2/blob/main/TvKeys.txt).
-For a full list see [here](https://github.com/tronikos/androidtvremote2/blob/b4c49ac03043b1b9c40c2f2960e466d5a3b8bd67/src/androidtvremote2/remotemessage.proto#L90).
+For a full list see [remotemessage.proto](https://github.com/tronikos/androidtvremote2/blob/b4c49ac03043b1b9c40c2f2960e466d5a3b8bd67/src/androidtvremote2/remotemessage.proto#L90).
 In addition to commands you can send URLs to open apps registered to handle them. See [this guide](https://community.home-assistant.io/t/android-tv-remote-app-links-deep-linking-guide/567921) for how to find deep links for apps.
 
+Voice commands can also be sent as PCM 16-bit mono 8 kHz audio data.
+
 ## Credits
 
 - Official [implementation](https://android.googlesource.com/platform/external/google-tv-pairing-protocol/+/refs/heads/master) of the pairing protocol in Java
@@ -45,7 +53,7 @@ source .venv/bin/activate
 
 # Install dependencies
 python -m pip install --upgrade pip
-python -m pip install .
+python -m pip install -e .
 
 # Generate *_pb2.py from *.proto
 python -m pip install grpcio-tools mypy-protobuf
@@ -53,23 +61,22 @@ python -m grpc_tools.protoc src/androidtvremote2/*.proto --python_out=src/androi
 
 # Run pre-commit
 python -m pip install pre-commit
-pre-commit autoupdate
 pre-commit install
 pre-commit run --all-files
 
-# Alternative: run formatter, lint, and type checking
-python -m pip install isort black flake8 ruff mypy
-isort . ; black . ; flake8 . ; ruff check . --fix ; mypy --install-types .
-
 # Run tests
-python -m pip install pytest
+python -m pip install -e ".[test]"
 pytest
 
 # Run demo
-python -m pip install pynput zeroconf
+python -m pip install -e ".[demo]"
 python src/demo.py
 
 # Build package
 python -m pip install build
 python -m build
 ```
+
+The voice demo requires the [PyAudio](https://pypi.org/project/PyAudio/) library.
+Depending on the target platform, [PortAudio](https://www.portaudio.com/) might have to be installed manually,
+see [PyAudio installation](https://people.csail.mit.edu/hubert/pyaudio/) for more information.

{androidtvremote2-0.2.2 → androidtvremote2-0.3.0}/README.md

@@ -3,9 +3,11 @@
 A Python library for interacting with Android TV using the Android TV Remote protocol v2. This is the same protocol the Google TV mobile app is using. It doesn't require ADB or enabling developer tools on the Android TV device. It only requires the [Android TV Remote Service](https://play.google.com/store/apps/details?id=com.google.android.tv.remote.service) that comes pre-installed on most Android TV devices.
 
 For a list of the most common commands you can send to the Android TV see: [TvKeys](https://github.com/tronikos/androidtvremote2/blob/main/TvKeys.txt).
-For a full list see [here](https://github.com/tronikos/androidtvremote2/blob/b4c49ac03043b1b9c40c2f2960e466d5a3b8bd67/src/androidtvremote2/remotemessage.proto#L90).
+For a full list see [remotemessage.proto](https://github.com/tronikos/androidtvremote2/blob/b4c49ac03043b1b9c40c2f2960e466d5a3b8bd67/src/androidtvremote2/remotemessage.proto#L90).
 In addition to commands you can send URLs to open apps registered to handle them. See [this guide](https://community.home-assistant.io/t/android-tv-remote-app-links-deep-linking-guide/567921) for how to find deep links for apps.
 
+Voice commands can also be sent as PCM 16-bit mono 8 kHz audio data.
+
 ## Credits
 
 - Official [implementation](https://android.googlesource.com/platform/external/google-tv-pairing-protocol/+/refs/heads/master) of the pairing protocol in Java
@@ -29,7 +31,7 @@ source .venv/bin/activate
 
 # Install dependencies
 python -m pip install --upgrade pip
-python -m pip install .
+python -m pip install -e .
 
 # Generate *_pb2.py from *.proto
 python -m pip install grpcio-tools mypy-protobuf
@@ -37,23 +39,22 @@ python -m grpc_tools.protoc src/androidtvremote2/*.proto --python_out=src/androi
 
 # Run pre-commit
 python -m pip install pre-commit
-pre-commit autoupdate
 pre-commit install
 pre-commit run --all-files
 
-# Alternative: run formatter, lint, and type checking
-python -m pip install isort black flake8 ruff mypy
-isort . ; black . ; flake8 . ; ruff check . --fix ; mypy --install-types .
-
 # Run tests
-python -m pip install pytest
+python -m pip install -e ".[test]"
 pytest
 
 # Run demo
-python -m pip install pynput zeroconf
+python -m pip install -e ".[demo]"
 python src/demo.py
 
 # Build package
 python -m pip install build
 python -m build
 ```
+
+The voice demo requires the [PyAudio](https://pypi.org/project/PyAudio/) library.
+Depending on the target platform, [PortAudio](https://www.portaudio.com/) might have to be installed manually,
+see [PyAudio installation](https://people.csail.mit.edu/hubert/pyaudio/) for more information.

androidtvremote2-0.3.0/pyproject.toml

@@ -0,0 +1,98 @@
+[project]
+name = "androidtvremote2"
+version = "0.3.0"
+license = "Apache-2.0"
+authors = [
+    { name="tronikos", email="tronikos@gmail.com" },
+]
+description = "A Python library for interacting with Android TV using the Android TV Remote protocol v2"
+readme = "README.md"
+requires-python = ">=3.10"
+dependencies = [
+    "aiofiles>=0.8",
+    "cryptography>=3",
+    "protobuf>=4.21",
+]
+
+[project.optional-dependencies]
+test = [
+    "pytest",
+]
+demo = [
+    "pyaudio",
+    "pynput",
+    "zeroconf",
+]
+
+
+[project.urls]
+"Homepage" = "https://github.com/tronikos/androidtvremote2"
+"Bug Tracker" = "https://github.com/tronikos/androidtvremote2/issues"
+
+[build-system]
+requires = ["setuptools"]
+build-backend = "setuptools.build_meta"
+
+[tool.ruff]
+target-version = "py310"
+line-length = 127
+exclude = [
+    ".git", ".mypy_cache", ".ruff_cache", ".venv", "__pypackages__",
+    "build", "dist", "venv", "*_pb2.py", "*_pb2_grpc.py", "*.pyi"
+]
+
+[tool.ruff.lint]
+select = [
+    "A",  # flake8-builtins
+    "ASYNC",  # flake8-async
+    "B",  # flake8-bugbear
+    "C",  # flake8-mccabe
+    "C4",  # flake8-comprehensions
+    "D",  # pydocstyle
+    "E",  # pycodestyle (error)
+    "F",  # pyflakes
+    "G",  # flake8-logging-format
+    "I",  # isort
+    "PGH",  # pygrep-hooks
+    "PL",  # Pylint
+    "RET",  # flake8-return
+    "RUF",  # Ruff-specific rules
+    "S",  # flake8-bandit (security)
+    "SIM",  # flake8-simplify
+    "T20",  # flake8-print
+    "TCH",  # flake8-type-checking
+    "TRY",  # tryceratops
+    "UP",  # pyupgrade
+    "W",  # pycodestyle (warning)
+]
+ignore = [
+    "S101",  # Ignore assert
+    "D203",  # 1 blank line required before class docstring
+    "D213",  # Multi-line docstring summary should start at the second line
+    "PLR0913",  # Too many arguments in function definition
+    "PLR1714",  # Consider merging multiple comparisons
+    "PLR2004",  # Magic value used in comparison
+    "TRY003",  # Avoid specifying long messages outside the exception class
+]
+fixable = ["ALL"]
+
+[tool.ruff.lint.mccabe]
+max-complexity = 25
+
+[tool.mypy]
+python_version = "3.10"
+exclude = [".venv", "venv", "build"]
+pretty = true
+strict = true
+show_error_codes = true
+warn_incomplete_stub = true
+enable_error_code = ["ignore-without-code", "redundant-self", "truthy-iterable"]
+
+[[tool.mypy.overrides]]
+module = ["pyaudio"]
+ignore_missing_imports = true
+
+[tool.codespell]
+skip = "*.json,*.csv,*.lock,./.git/*,./.venv/*"
+check-filenames = true
+check-hidden = true

androidtvremote2-0.3.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

{androidtvremote2-0.2.2 → androidtvremote2-0.3.0}/src/androidtvremote2/__init__.py

@@ -2,10 +2,15 @@
 
 from .androidtv_remote import AndroidTVRemote
 from .exceptions import CannotConnect, ConnectionClosed, InvalidAuth
+from .model import DeviceInfo, VolumeInfo
+from .voice_stream import VoiceStream
 
 __all__ = [
     "AndroidTVRemote",
     "CannotConnect",
     "ConnectionClosed",
+    "DeviceInfo",
     "InvalidAuth",
+    "VoiceStream",
+    "VolumeInfo",
 ]

{androidtvremote2-0.2.2 → androidtvremote2-0.3.0}/src/androidtvremote2/androidtv_remote.py

@@ -3,9 +3,9 @@
 from __future__ import annotations
 
 import asyncio
-from collections.abc import Callable
 import os
 import ssl
+from typing import TYPE_CHECKING
 from urllib.parse import urlparse
 
 import aiofiles
@@ -15,8 +15,14 @@ from .certificate_generator import generate_selfsigned_cert
 from .const import LOGGER
 from .exceptions import CannotConnect, ConnectionClosed, InvalidAuth
 from .pairing import PairingProtocol
-from .remote import RemoteProtocol
+from .remote import VOICE_SESSION_TIMEOUT, RemoteProtocol
 from .remotemessage_pb2 import RemoteDirection
+from .voice_stream import VoiceStream
+
+if TYPE_CHECKING:
+    from collections.abc import Callable
+
+    from .model import DeviceInfo, VolumeInfo
 
 
 def _load_cert_chain(certfile: str, keyfile: str) -> ssl.SSLContext:
@@ -40,6 +46,7 @@ class AndroidTVRemote:
         pair_port: int = 6467,
         loop: asyncio.AbstractEventLoop | None = None,
         enable_ime: bool = True,
+        enable_voice: bool = False,
     ) -> None:
         """Initialize.
 
@@ -52,6 +59,7 @@ class AndroidTVRemote:
         :param loop: event loop. Used for connections and futures.
         :param enable_ime: Needed for getting current_app.
                Disable for devices that show 'Use keyboard on mobile device screen'.
+        :param enable_voice: Enable sending voice commands to the device.
         """
         self._client_name = client_name
         self._certfile = certfile
@@ -61,15 +69,16 @@ class AndroidTVRemote:
         self._pair_port = pair_port
         self._loop = loop or asyncio.get_running_loop()
         self._enable_ime = enable_ime
+        self._enable_voice = enable_voice
         self._transport: asyncio.Transport | None = None
         self._remote_message_protocol: RemoteProtocol | None = None
         self._pairing_message_protocol: PairingProtocol | None = None
-        self._reconnect_task: asyncio.Task | None = None
+        self._reconnect_task: asyncio.Task[None] | None = None
         self._ssl_context: ssl.SSLContext | None = None
-        self._is_on_updated_callbacks: list[Callable] = []
-        self._current_app_updated_callbacks: list[Callable] = []
-        self._volume_info_updated_callbacks: list[Callable] = []
-        self._is_available_updated_callbacks: list[Callable] = []
+        self._is_on_updated_callbacks: list[Callable[[bool], None]] = []
+        self._current_app_updated_callbacks: list[Callable[[str], None]] = []
+        self._volume_info_updated_callbacks: list[Callable[[VolumeInfo], None]] = []
+        self._is_available_updated_callbacks: list[Callable[[bool], None]] = []
 
         def is_on_updated(is_on: bool) -> None:
             for callback in self._is_on_updated_callbacks:
@@ -79,7 +88,7 @@ class AndroidTVRemote:
             for callback in self._current_app_updated_callbacks:
                 callback(current_app)
 
-        def volume_info_updated(volume_info: dict[str, str | bool]) -> None:
+        def volume_info_updated(volume_info: VolumeInfo) -> None:
             for callback in self._volume_info_updated_callbacks:
                 callback(volume_info)
 
@@ -107,57 +116,68 @@ class AndroidTVRemote:
         return self._remote_message_protocol.current_app
 
     @property
-    def device_info(self) -> dict[str, str] | None:
+    def device_info(self) -> DeviceInfo | None:
         """Device info (manufacturer, model, sw_version)."""
         if not self._remote_message_protocol:
             return None
         return self._remote_message_protocol.device_info
 
     @property
-    def volume_info(self) -> dict[str, str | bool | int] | None:
+    def volume_info(self) -> VolumeInfo | None:
         """Volume info (level, max, muted)."""
         if not self._remote_message_protocol:
             return None
         return self._remote_message_protocol.volume_info
 
-    def add_is_on_updated_callback(self, callback: Callable) -> None:
+    @property
+    def is_voice_enabled(self) -> bool | None:
+        """Whether voice commands are enabled on the Android TV.
+
+        Depends on the requested feature at AndroidTVRemote initialization and the supported
+        features of the device.
+        """
+        if not self._remote_message_protocol:
+            return None
+        return self._remote_message_protocol.is_voice_enabled
+
+    def add_is_on_updated_callback(self, callback: Callable[[bool], None]) -> None:
         """Add a callback for when is_on is updated."""
         self._is_on_updated_callbacks.append(callback)
 
-    def remove_is_on_updated_callback(self, callback: Callable) -> None:
+    def remove_is_on_updated_callback(self, callback: Callable[[bool], None]) -> None:
         """Remove a callback previously added via add_is_on_updated_callback.
 
         :raises ValueError: if callback not previously added.
         """
         self._is_on_updated_callbacks.remove(callback)
 
-    def add_current_app_updated_callback(self, callback: Callable) -> None:
+    def add_current_app_updated_callback(self, callback: Callable[[str], None]) -> None:
         """Add a callback for when current_app is updated."""
         self._current_app_updated_callbacks.append(callback)
 
-    def remove_current_app_updated_callback(self, callback: Callable) -> None:
+    def remove_current_app_updated_callback(self, callback: Callable[[str], None]) -> None:
         """Remove a callback previously added via add_current_app_updated_callback.
 
         :raises ValueError: if callback not previously added.
         """
         self._current_app_updated_callbacks.remove(callback)
 
-    def add_volume_info_updated_callback(self, callback: Callable) -> None:
+    def add_volume_info_updated_callback(self, callback: Callable[[VolumeInfo], None]) -> None:
         """Add a callback for when volume_info is updated."""
         self._volume_info_updated_callbacks.append(callback)
 
-    def remove_volume_info_updated_callback(self, callback: Callable) -> None:
+    def remove_volume_info_updated_callback(self, callback: Callable[[VolumeInfo], None]) -> None:
         """Remove a callback previously added via add_volume_info_updated_callback.
 
         :raises ValueError: if callback not previously added.
         """
         self._volume_info_updated_callbacks.remove(callback)
 
-    def add_is_available_updated_callback(self, callback: Callable) -> None:
+    def add_is_available_updated_callback(self, callback: Callable[[bool], None]) -> None:
         """Add a callback for when the Android TV is ready to receive commands or is unavailable."""
         self._is_available_updated_callbacks.append(callback)
 
-    def remove_is_available_updated_callback(self, callback: Callable) -> None:
+    def remove_is_available_updated_callback(self, callback: Callable[[bool], None]) -> None:
         """Remove a callback previously added via add_is_available_updated_callback.
 
         :raises ValueError: if callback not previously added.
@@ -182,9 +202,7 @@ class AndroidTVRemote:
         if self._ssl_context:
             return self._ssl_context
         try:
-            ssl_context = await self._loop.run_in_executor(
-                None, _load_cert_chain, self._certfile, self._keyfile
-            )
+            ssl_context = await self._loop.run_in_executor(None, _load_cert_chain, self._certfile, self._keyfile)
         except FileNotFoundError as exc:
             LOGGER.debug("Missing certificate. Error: %s", exc)
             raise InvalidAuth from exc
@@ -214,24 +232,19 @@ class AndroidTVRemote:
                     self._on_volume_info_updated,
                     self._loop,
                     self._enable_ime,
+                    self._enable_voice,
                 ),
                 self.host,
                 self._api_port,
                 ssl=ssl_context,
             )
         except OSError as exc:
-            LOGGER.debug(
-                "Couldn't connect to %s:%s. Error: %s", self.host, self._api_port, exc
-            )
+            LOGGER.debug("Couldn't connect to %s:%s. Error: %s", self.host, self._api_port, exc)
             if isinstance(exc, ssl.SSLError):
                 raise InvalidAuth("Need to pair") from exc
-            raise CannotConnect(
-                f"Couldn't connect to {self.host}:{self._api_port}"
-            ) from exc
+            raise CannotConnect(f"Couldn't connect to {self.host}:{self._api_port}") from exc
 
-        await asyncio.wait(
-            (on_con_lost, on_remote_started), return_when=asyncio.FIRST_COMPLETED
-        )
+        await asyncio.wait((on_con_lost, on_remote_started), return_when=asyncio.FIRST_COMPLETED)
         if on_con_lost.done():
             con_lost_exc = on_con_lost.result()
             LOGGER.debug(
@@ -244,17 +257,13 @@ class AndroidTVRemote:
                 raise InvalidAuth("Need to pair again") from con_lost_exc
             raise ConnectionClosed("Connection closed") from con_lost_exc
 
-    async def _async_reconnect(
-        self, invalid_auth_callback: Callable | None = None
-    ) -> None:
+    async def _async_reconnect(self, invalid_auth_callback: Callable[[], None] | None = None) -> None:
         while self._remote_message_protocol:
             exc = await self._remote_message_protocol.on_con_lost
             self._on_is_available_updated(False)
             LOGGER.debug("Disconnected from %s. Error: %s", self.host, exc)
             delay_seconds = 0.1
-            LOGGER.debug(
-                "Trying to reconnect to %s in %s seconds", self.host, delay_seconds
-            )
+            LOGGER.debug("Trying to reconnect to %s in %s seconds", self.host, delay_seconds)
             while self._remote_message_protocol:
                 await asyncio.sleep(delay_seconds)
                 try:
@@ -279,11 +288,9 @@ class AndroidTVRemote:
                         invalid_auth_callback()
                     return
 
-    def keep_reconnecting(self, invalid_auth_callback: Callable | None = None) -> None:
+    def keep_reconnecting(self, invalid_auth_callback: Callable[[], None] | None = None) -> None:
         """Create a task to keep reconnecting whenever connection is lost."""
-        self._reconnect_task = self._loop.create_task(
-            self._async_reconnect(invalid_auth_callback)
-        )
+        self._reconnect_task = self._loop.create_task(self._async_reconnect(invalid_auth_callback))
 
     def disconnect(self) -> None:
         """Disconnect any open connections."""
@@ -305,17 +312,11 @@ class AndroidTVRemote:
         """
         ssl_context = await self._create_ssl_context()
         try:
-            _, writer = await asyncio.open_connection(
-                self.host, self._pair_port, ssl=ssl_context
-            )
+            _, writer = await asyncio.open_connection(self.host, self._pair_port, ssl=ssl_context)
         except OSError as exc:
-            LOGGER.debug(
-                "Couldn't connect to %s:%s. %s", self.host, self._pair_port, exc
-            )
+            LOGGER.debug("Couldn't connect to %s:%s. %s", self.host, self._pair_port, exc)
             raise CannotConnect from exc
-        server_cert_bytes = writer.transport.get_extra_info("ssl_object").getpeercert(
-            True
-        )
+        server_cert_bytes = writer.transport.get_extra_info("ssl_object").getpeercert(True)
         writer.close()
         server_cert = x509.load_der_x509_certificate(server_cert_bytes)
         # NVIDIA SHIELD example:
@@ -357,9 +358,7 @@ class AndroidTVRemote:
                 ssl=ssl_context,
             )
         except OSError as exc:
-            LOGGER.debug(
-                "Couldn't connect to %s:%s. %s", self.host, self._pair_port, exc
-            )
+            LOGGER.debug("Couldn't connect to %s:%s. %s", self.host, self._pair_port, exc)
             raise CannotConnect from exc
         await self._pairing_message_protocol.async_start_pairing()
 
@@ -376,9 +375,7 @@ class AndroidTVRemote:
         await self._pairing_message_protocol.async_finish_pairing(pairing_code)
         self.disconnect()
 
-    def send_key_command(
-        self, key_code: int | str, direction: int | str = RemoteDirection.SHORT
-    ) -> None:
+    def send_key_command(self, key_code: int | str, direction: int | str = RemoteDirection.SHORT) -> None:
         """Send a key press to Android TV.
 
         This does not block; it buffers the data and arranges for it to be sent out asynchronously.
@@ -418,6 +415,22 @@ class AndroidTVRemote:
             LOGGER.debug("Called send_launch_app_command after disconnect")
             raise ConnectionClosed("Called send_launch_app_command after disconnect")
         prefix = "" if urlparse(app_link_or_app_id).scheme else "market://launch?id="
-        self._remote_message_protocol.send_launch_app_command(
-            f"{prefix}{app_link_or_app_id}"
-        )
+        self._remote_message_protocol.send_launch_app_command(f"{prefix}{app_link_or_app_id}")
+
+    async def start_voice(self, timeout: float = VOICE_SESSION_TIMEOUT) -> VoiceStream:
+        """Start a streaming voice session.
+
+        A ``VoiceStream`` session wrapper is returned if the voice session can be established
+        within the given timeout. The session needs to be closed with ``end()`` (or through the
+        asynchronous context manager) before a new session is started.
+
+        :param timeout: optional timeout for session readiness. Defaults to 2 seconds.
+        :raises ConnectionClosed: if client is disconnected.
+        :raises asyncio.TimeoutError: if the device does not begin voice in time, or a voice
+                                      session is already in progress.
+        """
+        if not self._remote_message_protocol:
+            LOGGER.debug("Called start_voice after disconnect")
+            raise ConnectionClosed("Called start_voice after disconnect")
+        session_id = await self._remote_message_protocol.start_voice(timeout)
+        return VoiceStream(self._remote_message_protocol, session_id)

{androidtvremote2-0.2.2 → androidtvremote2-0.3.0}/src/androidtvremote2/base.py

@@ -3,20 +3,22 @@
 from __future__ import annotations
 
 import asyncio
-from typing import cast
+from typing import TYPE_CHECKING, cast
 
 from google.protobuf import text_format
 from google.protobuf.internal.decoder import _DecodeVarint  # type: ignore[attr-defined]
 from google.protobuf.internal.encoder import _EncodeVarint  # type: ignore[attr-defined]
-from google.protobuf.message import Message
 
 from .const import LOGGER
 
+if TYPE_CHECKING:
+    from google.protobuf.message import Message
+
 
 class ProtobufProtocol(asyncio.Protocol):
     """Protocol for receiving and sending protobuf messages."""
 
-    def __init__(self, on_con_lost: asyncio.Future) -> None:
+    def __init__(self, on_con_lost: asyncio.Future[Exception | None]) -> None:
         """Initialize.
 
         :param on_con_lost: callback for when the connection is lost or closed.
@@ -29,7 +31,7 @@ class ProtobufProtocol(asyncio.Protocol):
     def connection_made(self, transport: asyncio.BaseTransport) -> None:
         """Store transport when a connection is made."""
         LOGGER.debug("Connected to %s", transport.get_extra_info("peername"))
-        self.transport = cast(asyncio.Transport, transport)
+        self.transport = cast("asyncio.Transport", transport)
 
     def connection_lost(self, exc: Exception | None) -> None:
         """Notify on_con_lost when the connection is lost or closed."""
@@ -69,9 +71,7 @@ class ProtobufProtocol(asyncio.Protocol):
         This does not block; it buffers the data and arranges for it to be sent out asynchronously.
         """
         if should_debug_log:
-            LOGGER.debug(
-                "Sending: %s", text_format.MessageToString(msg, as_one_line=True)
-            )
+            LOGGER.debug("Sending: %s", text_format.MessageToString(msg, as_one_line=True))
         if not self.transport or self.transport.is_closing():
             LOGGER.debug("Connection is closed!")
             return

androidtvremote2-0.3.0/src/androidtvremote2/model.py

@@ -0,0 +1,21 @@
+"""Data models for AndroidTVRemote."""
+
+from __future__ import annotations
+
+from typing import TypedDict
+
+
+class DeviceInfo(TypedDict):
+    """A TypedDict for device information."""
+
+    manufacturer: str
+    model: str
+    sw_version: str
+
+
+class VolumeInfo(TypedDict):
+    """A TypedDict for volume information."""
+
+    level: int
+    max: int
+    muted: bool