Add support for getting the bluetooth MAC form the primary MAC (#815)

* Add support for getting the bluetooth MAC form the primary MAC

* Add support for getting the bluetooth MAC form the primary MAC

* Add support for getting the bluetooth MAC form the primary MAC

* Update aioshelly/rpc_device/utils.py

Author: bdraco

aioshelly/rpc_device/__init__.py

@@ -1,6 +1,7 @@
 """Shelly Gen2 RPC based device."""
 
 from .device import RpcDevice, RpcUpdateType
+from .utils import bluetooth_mac_from_primary_mac
 from .wsrpc import WsServer
 
-__all__ = ["RpcDevice", "RpcUpdateType", "WsServer"]
+__all__ = ["RpcDevice", "RpcUpdateType", "WsServer", "bluetooth_mac_from_primary_mac"]

aioshelly/rpc_device/utils.py

@@ -0,0 +1,17 @@
+"""Utilities for RPC devices."""
+
+from __future__ import annotations
+
+# The Bluetooth MAC address is the primary MAC address plus 2.
+# https://docs.espressif.com/projects/esp-idf/en/latest/esp32/api-reference/system/misc_system_api.html#mac-address
+
+
+def bluetooth_mac_from_primary_mac(primary_mac: str) -> str:
+    """Get Bluetooth MAC from primary MAC.
+
+    MAC address must be in format "[0-F]{16}"
+
+    :param primary_mac: Primary MAC address
+    :return: Bluetooth MAC address
+    """
+    return f"{(int(primary_mac, 16) + 2):012X}"

tests/rpc_device/test_init.py

@@ -0,0 +1,9 @@
+from aioshelly import rpc_device
+
+
+def test_exports() -> None:
+    """Test objects are available at top level of rpc_device."""
+    assert hasattr(rpc_device, "bluetooth_mac_from_primary_mac")
+    assert hasattr(rpc_device, "RpcDevice")
+    assert hasattr(rpc_device, "RpcUpdateType")
+    assert hasattr(rpc_device, "WsServer")

tests/rpc_device/test_utils.py

@@ -0,0 +1,9 @@
+from aioshelly.rpc_device.utils import bluetooth_mac_from_primary_mac
+
+
+def test_bluetooth_mac_from_primary_mac() -> None:
+    """Test bluetooth_mac_from_primary_mac."""
+    assert bluetooth_mac_from_primary_mac("0A1B2C3D4E5F") == "0A1B2C3D4E61"
+    assert bluetooth_mac_from_primary_mac("0A1B2C3D4EA0") == "0A1B2C3D4EA2"
+    assert bluetooth_mac_from_primary_mac("0A1B2C3D4EF0") == "0A1B2C3D4EF2"
+    assert bluetooth_mac_from_primary_mac("0A1B2C3D4EC9") == "0A1B2C3D4ECB"