python-ember-mug 1.3.0b2__tar.gz → 1.3.0b4__tar.gz

{python_ember_mug-1.3.0b2 → python_ember_mug-1.3.0b4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-ember-mug
-Version: 1.3.0b2
+Version: 1.3.0b4
 Summary: Python Library for Ember Mugs.
 Project-URL: Changelog, https://sopelj.github.io/python-ember-mug/changelog/
 Project-URL: Documentation, https://sopelj.github.io/python-ember-mug/

{python_ember_mug-1.3.0b2 → python_ember_mug-1.3.0b4}/ember_mug/__init__.py

@@ -6,4 +6,4 @@ __all__ = ("EmberMug",)
 
 __author__ = """Jesse Sopel"""
 __email__ = "jesse.sopel@gmail.com"
-__version__ = "1.3.0b2"
+__version__ = "1.3.0b4"

{python_ember_mug-1.3.0b2 → python_ember_mug-1.3.0b4}/ember_mug/mug.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 import asyncio
 import contextlib
 import logging
+import os
 from datetime import UTC, datetime
 from enum import Enum
 from functools import cached_property
@@ -174,8 +175,6 @@ class EmberMug:
                     disconnected_callback=self._disconnect_callback,
                     ble_device_callback=lambda: self.device,
                 )
-                if self.debug:
-                    await discover_services(client)
                 self._expected_disconnect = False
             except (TimeoutError, BleakError) as error:
                 logger.debug("%s: Failed to connect to the mug: %s", self.device, error)
@@ -277,6 +276,25 @@ class EmberMug:
         """Get Battery percent from mug gatt."""
         return BatteryInfo.from_bytes(await self._read(MugCharacteristic.BATTERY))
 
+    async def make_writable(self) -> bool:
+        """Try to make device writable if need be."""
+        if self.can_write:
+            return True
+        try:
+            await self._ensure_connection()
+            # Attempt to write a random string
+            await self.set_udsk(os.urandom(14).hex())
+            return True
+        except BleakError as e:
+            logger.debug("Failed to make device writable: %s", e)
+            return False
+
+    async def pair(self) -> None:
+        """Attempt to pair."""
+        with contextlib.suppress(BleakError, EOFError):
+            await self._ensure_connection()
+            await self._client.pair()
+
     @require_attribute("led_colour")
     async def get_led_colour(self) -> Colour:
         """Get RGBA colours from mug gatt."""