From b62824b7d452b14cb9d0d88c9e8b27a892b9022d Mon Sep 17 00:00:00 2001
From: Marius Gedminas <marius@gedmin.as>
Date: Wed, 8 Jan 2020 18:19:40 +0200
Subject: [PATCH] Only connect and authenticate once

My Viomi Smart Kettle SK-152A (product ID 1116, yunmi.kettle.v7) doesn't
like it when you try to connect too frequently, resulting in
bluepy.btle.BTLEDisconnectError: Failed to connect to peripheral
XX:XX:XX:XX:XX:XX, addr type: public.

Reusing the same connection makes demo.py work fine.

See #4.
---
 mikettle/mikettle.py | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/mikettle/mikettle.py b/mikettle/mikettle.py
index 4d75a3b..c212934 100644
--- a/mikettle/mikettle.py
+++ b/mikettle/mikettle.py
@@ -78,9 +78,13 @@ def __init__(self, mac, product_id, cache_timeout=600, retries=3, iface=None, to
             token = MiKettle.generateRandomToken()
         self._token = token
 
+        self._p = None
+        self._authenticated = False
+
     def connect(self):
-        self._p = Peripheral(deviceAddr=self._mac, iface=self._iface)
-        self._p.setDelegate(self)
+        if self._p is None:
+            self._p = Peripheral(deviceAddr=self._mac, iface=self._iface)
+            self._p.setDelegate(self)
 
     def name(self):
         """Return the name of the device."""
@@ -175,6 +179,8 @@ def bytes_to_int(bytes):
         return result
 
     def auth(self):
+        if self._authenticated:
+            return
         auth_service = self._p.getServiceByUUID(_UUID_SERVICE_KETTLE)
         auth_descriptors = auth_service.getDescriptors()
 
@@ -191,6 +197,7 @@ def auth(self):
         self._p.writeCharacteristic(_HANDLE_AUTH, MiKettle.cipher(self._token, _KEY2), "true")
 
         self._p.readCharacteristic(_HANDLE_VERSION)
+        self._authenticated = True
 
     def subscribeToData(self):
         controlService = self._p.getServiceByUUID(_UUID_SERVICE_KETTLE_DATA)