refactor: remove unused import, fix style

Author: jalcaras

custom_components/weback_vacuum/camera.py

@@ -1,32 +1,16 @@
 """Support for Weback Vacuum Robot map camera."""
+
+from __future__ import annotations
+
 import logging
-import resource
 
-import homeassistant.helpers.config_validation as cv
-import voluptuous as vol
 from homeassistant.components.camera import (
     ENTITY_ID_FORMAT,
-    PLATFORM_SCHEMA,
-    SUPPORT_ON_OFF,
     Camera,
-    CameraEntityFeature,
-)
-from homeassistant.components.vacuum import (
-    STATE_CLEANING,
-    STATE_DOCKED,
-    STATE_ERROR,
-    STATE_IDLE,
-    STATE_PAUSED,
-    STATE_RETURNING,
-    StateVacuumEntity,
-    VacuumEntityFeature,
 )
-from homeassistant.helpers import entity_platform
 from homeassistant.helpers.entity import generate_entity_id
-from homeassistant.helpers.icon import icon_for_battery_level
 
 from . import DOMAIN, VacDevice
-from .vacmap import VacMap, VacMapDraw
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -38,7 +22,6 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info=
     for device in hass.data[DOMAIN]:
         entity_id = generate_entity_id(ENTITY_ID_FORMAT, device.name, hass=hass)
         vacuums.append(WebackVacuumCamera(device, entity_id))
-        # hass.loop.create_task(device.watch_state())
 
     _LOGGER.debug("Adding Weback Vacuums Maps to Home Assistant: %s", vacuums)
 
@@ -55,15 +38,9 @@ def __init__(self, device: VacDevice, entity_id):
         super().__init__()
         self._vacdevice = device
         self._vacdevice.register_map_camera(self)
-        # self.entity_id = entity_id
         self.content_type = "image/png"
-
-        # self._vacdevice.subscribe(lambda vacdevice: self.schedule_update_ha_state(False))
-
         self._error = None
-
-        # self._attr_supported_features = ()
-        _LOGGER.info(f"Vacuum Camera initialized: {self.name}")
+        _LOGGER.info("Vacuum Camera initialized: %s", self.name)
 
     @property
     def name(self):
@@ -85,13 +62,14 @@ def extra_state_attributes(self):
         return attributes
 
     def camera_image(
-        self, width: int | None = None, height: int | None = None
+        self,
+        width: int | None = None,
+        height: int | None = None,
     ) -> bytes | None:
         """Return bytes of camera image."""
         if self._vacdevice.map:
             return self.generate_image()
-        else:
-            return None
+        return None
 
     def generate_image(self):
         return self._vacdevice.map_image_buffer