Refactor

Author: PiotrMachowski

custom_components/xiaomi_cloud_map_extractor/binary_sensor.py

@@ -16,8 +16,8 @@
 from homeassistant.helpers.typing import StateType
 from vacuum_map_parser_base.map_data import MapData
 
-from .entity import XiaomiCloudMapExtractorEntity
 from .coordinator import XiaomiCloudMapExtractorDataUpdateCoordinator
+from .entity import XiaomiCloudMapExtractorEntity
 from .types import XiaomiCloudMapExtractorConfigEntry
 
 
@@ -51,6 +51,7 @@ async def async_setup_entry(
         for description in SENSOR_TYPES
     )
 
+
 class XiaomiCloudMapExtractorBinarySensorEntity(XiaomiCloudMapExtractorEntity, BinarySensorEntity):
     entity_description: XiaomiCloudMapExtractorBinarySensorEntityDescription
 

custom_components/xiaomi_cloud_map_extractor/button.py

@@ -15,8 +15,8 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from .entity import XiaomiCloudMapExtractorEntity
 from .coordinator import XiaomiCloudMapExtractorDataUpdateCoordinator
+from .entity import XiaomiCloudMapExtractorEntity
 from .types import XiaomiCloudMapExtractorConfigEntry
 
 
@@ -50,6 +50,7 @@ async def async_setup_entry(
         for description in BUTTON_TYPES
     )
 
+
 class XiaomiCloudMapExtractorButtonEntity(XiaomiCloudMapExtractorEntity, ButtonEntity):
     entity_description: XiaomiCloudMapExtractorButtonEntityDescription
 

custom_components/xiaomi_cloud_map_extractor/config_flow.py

@@ -179,7 +179,7 @@ def session_creator() -> ClientSession:
                 qr_image_b64 = base64.b64encode(qr_image).decode("utf-8")
                 placeholders["qr_image_b64"] = qr_image_b64
                 placeholders["login_link"] = login_link
-            except FailedLoginException as ex:
+            except FailedLoginException:
                 return self.async_abort(reason="qr_code_unavailable")
 
             _wait_for_login = self._connector.login_with_qr_wait_for_completion

custom_components/xiaomi_cloud_map_extractor/connector/xiaomi_cloud/utils.py

@@ -69,5 +69,6 @@ def decrypt_rc4(password: str, payload: str) -> bytes:
     r.encrypt(bytes(1024))
     return r.encrypt(base64.b64decode(payload))
 
+
 def random_text(chr_from: int, chr_to: int, length: int) -> str:
     return "".join([chr(random.randint(chr_from, chr_to)) for _ in range(length)])

custom_components/xiaomi_cloud_map_extractor/sensor.py

@@ -18,10 +18,10 @@
 from homeassistant.helpers.typing import StateType
 from vacuum_map_parser_base.map_data import MapData
 
-from .entity import XiaomiCloudMapExtractorEntity
+from .connector.utils.list_operations import as_list_of_dict, len_len
 from .coordinator import XiaomiCloudMapExtractorDataUpdateCoordinator
+from .entity import XiaomiCloudMapExtractorEntity
 from .types import XiaomiCloudMapExtractorConfigEntry
-from .connector.utils.list_operations import as_list_of_dict, len_len
 
 
 @dataclass(frozen=True, kw_only=True)
@@ -262,6 +262,7 @@ async def async_setup_entry(
         for description in SENSOR_TYPES
     )
 
+
 class XiaomiCloudMapExtractorSensorEntity(XiaomiCloudMapExtractorEntity, SensorEntity):
     entity_description: XiaomiCloudMapExtractorSensorEntityDescription