Try to be more careful with imports

IATkachenko · IATkachenko · commit c5f0a8617edd · 2020-06-09T20:40:24.000+03:00
diff --git a/tion/s3.py b/tion/s3.py
@@ -11,6 +11,7 @@
 
 _LOGGER = logging.getLogger(__name__)
 
+
 class s3(tion):
     uuid = "6e400001-b5a3-f393-e0a9-e50e24dcca9e"
     uuid_write = "6e400002-b5a3-f393-e0a9-e50e24dcca9e"
@@ -29,14 +30,13 @@ class s3(tion):
     command_SET_PARAMS = 2
 
     def __init__(self, mac: str):
-        self._btle = btle.Peripheral(None)
-        self._mac = mac
+        super().__init__(mac)
 
     def __try_get_state(self) -> bytearray:
         return self._btle.getServiceByUUID(self.uuid).getCharacteristics()[0].read()
 
     def pair(self):
-        def get_pair_command(self) -> bytearray:
+        def get_pair_command() -> bytearray:
             return self.create_command(self.command_PAIR)
 
         self._btle.connect(self.mac, btle.ADDR_TYPE_RANDOM)
diff --git a/tion/tion.py b/tion/tion.py
@@ -2,8 +2,11 @@
 import logging
 from typing import Callable
 
+from bluepy import btle
+
 _LOGGER = logging.getLogger(__name__)
 
+
 class TionException(Exception):
     def __init__(self, expression, message):
         self.expression = expression
@@ -17,6 +20,7 @@ class tion:
 
     def __init__(self, mac: str):
         self._mac = mac
+        self._btle: btle.Peripheral = btle.Peripheral(None)
 
     @abc.abstractmethod
     def _send_request(self, request: bytearray) -> bytearray:
