fix precommit

Author: square-spade

.github/workflows/lint_test.yml

@@ -3,7 +3,7 @@ name: Lint & Test
 on:
   pull_request:
   push:
-    branches: [base]      
+    branches: [base]
 
 jobs:
   pre-commit:

custom_components/ttlock/__init__.py

@@ -42,7 +42,12 @@
 from .models import WebhookEvent
 from .services import Services
 
-PLATFORMS: list[Platform] = [Platform.LOCK, Platform.SENSOR, Platform.BINARY_SENSOR, Platform.SWITCH]
+PLATFORMS: list[Platform] = [
+    Platform.LOCK,
+    Platform.SENSOR,
+    Platform.BINARY_SENSOR,
+    Platform.SWITCH,
+]
 
 _LOGGER = logging.getLogger(__name__)
 

custom_components/ttlock/api.py

@@ -164,8 +164,7 @@ async def get_sensor(self, lock_id: int) -> Sensor:
             _LOGGER.error("Error setting up sensor", lock_id, res["errmsg"])
             pass
         return Sensor.parse_obj(res)
-      
-    
+
     async def get_lock_state(self, lock_id: int) -> LockState:
         """Get the state of a lock."""
         async with GW_LOCK:
@@ -272,7 +271,7 @@ async def delete_passcode(self, lock_id: int, passcode_id: int) -> bool:
             return False
 
         return True
-    
+
     async def set_auto_lock(self, lock_id: int, seconds: int) -> bool:
         """Set the AutoLock feature of the lock."""
 

custom_components/ttlock/binary_sensor.py

@@ -3,7 +3,10 @@
 
 import logging
 
-from homeassistant.components.binary_sensor import BinarySensorDeviceClass, BinarySensorEntity
+from homeassistant.components.binary_sensor import (
+    BinarySensorDeviceClass,
+    BinarySensorEntity,
+)
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
@@ -32,15 +35,18 @@ async def async_setup_entry(
         ]
     )
 
+
 class Sensor(BaseLockEntity, BinarySensorEntity):
     """Current sensor state."""
+
     _attr_device_class = BinarySensorDeviceClass.DOOR
 
     def _update_from_coordinator(self) -> None:
         """Fetch state of device."""
         self._attr_name = f"{self.coordinator.data.name} Sensor"
         self._attr_is_on = self.coordinator.data.opened
 
+
 class PassageMode(BaseLockEntity, BinarySensorEntity):
     """Current passage mode state."""
 

custom_components/ttlock/const.py

@@ -21,4 +21,4 @@
 SVC_CONFIG_PASSAGE_MODE = "configure_passage_mode"
 SVC_CREATE_PASSCODE = "create_passcode"
 SVC_CLEANUP_PASSCODES = "cleanup_passcodes"
-SVC_SET_AUTOLOCK =  "set_autolock"
+SVC_SET_AUTOLOCK = "set_autolock"

custom_components/ttlock/coordinator.py

@@ -44,7 +44,7 @@ class LockState:
     opened: bool | None = None
     sensor_battery: int | None = None
 
-    auto_lock_seconds: int | None = None
+    auto_lock_seconds: int = 0
     auto_lock: bool | None = None
     passage_mode_config: PassageModeConfig | None = None
 
@@ -74,9 +74,10 @@ def auto_lock_delay(self, current_date: datetime) -> int | None:
 
         if self.passage_mode_active(current_date):
             return None
-        
+
         return self.auto_lock_seconds
 
+
 @contextmanager
 def lock_action(controller: LockUpdateCoordinator):
     """Wrap a lock action so that in-progress state is managed correctly."""
@@ -150,21 +151,26 @@ async def _async_update_data(self) -> LockState:
                     pass
 
             new_data.auto_lock_seconds = details.autoLockTime
-            new_data.auto_lock = True if details.autoLockTime > 0 else False
+            if details.autoLockSeconds > 0:
+                new_data.auto_lock = True
+            else:
+                new_data.auto_lock = False
             new_data.lock_sound = details.lockSound == 1
 
             new_data.passage_mode_config = await self.api.get_lock_passage_mode_config(
                 self.lock_id
             )
-            
-            if Features.door_sensor in Features.from_feature_value(details.featureValue):
+
+            if Features.door_sensor in Features.from_feature_value(
+                details.featureValue
+            ):
                 sensor = await self.api.get_sensor(self.lock_id)
                 if sensor.battery_level is not None:
                     new_data.sensor_battery = sensor.battery_level
                     new_data.sensor = True
                 else:
                     new_data.sensor = False
-            
+
             return new_data
         except Exception as err:
             raise UpdateFailed(err) from err

custom_components/ttlock/models.py

@@ -50,12 +50,15 @@ class State(Enum):
     unlocked = 1
     unknown = 2
 
+
 class SensorState(Enum):
     """State of the sensor."""
+
     opened = 0
     closed = 1
     unknown = None
 
+
 class Lock(BaseModel):
     """Lock details."""
 
@@ -82,6 +85,7 @@ class Lock(BaseModel):
     # sensitive fields
     noKeyPwd: str = Field(alias="adminPwd")
 
+
 class Sensor(BaseModel):
     """sensor details."""
 
@@ -90,12 +94,14 @@ class Sensor(BaseModel):
     battery_level: int | None = Field(None, alias="electricQuantity")
     mac: str = Field(..., alias="mac")
 
+
 class LockState(BaseModel):
     """Lock state."""
 
     locked: State | None = Field(State.unknown, alias="state")
     opened: SensorState | None = Field(SensorState.unknown, alias="sensorState")
 
+
 class PassageModeConfig(BaseModel):
     """The passage mode configuration of the lock."""
 
@@ -279,7 +285,7 @@ def state(self) -> LockState:
         elif self.success and self.event.action == Action.unlock:
             return LockState(state=State.unlocked)
         return LockState(state=None)
-    
+
     @property
     def sensorState(self) -> LockState:
         """The end state of the sensor after this event."""

custom_components/ttlock/sensor.py

@@ -92,13 +92,14 @@ async def async_added_to_hass(self) -> None:
 
         self._attr_native_value = last_state.state
 
+
 class SensorBattery(BaseLockEntity, SensorEntity):
-    """Representaion of sensor bettery."""
+    """Representation of sensor battery."""
 
     _attr_device_class = SensorDeviceClass.BATTERY
     _attr_native_unit_of_measurement = PERCENTAGE
 
     def _update_from_coordinator(self) -> None:
         """Fetch state from the device."""
         self._attr_name = f"{self.coordinator.data.name} Sensor Battery"
-        self._attr_native_value = self.coordinator.data.sensor_battery
+        self._attr_native_value = self.coordinator.data.sensor_battery

custom_components/ttlock/services.py

@@ -96,7 +96,9 @@ def register(self) -> None:
                 {
                     vol.Required(ATTR_ENTITY_ID): cv.entity_ids,
                     vol.Required(CONF_ENABLED): cv.boolean,
-                    vol.Optional(CONF_SECONDS): vol.All(vol.Coerce(int), vol.Range(min=0, max=60)),
+                    vol.Optional(CONF_SECONDS): vol.All(
+                        vol.Coerce(int), vol.Range(min=0, max=60)
+                    ),
                 }
             ),
         )
@@ -168,12 +170,14 @@ async def handle_cleanup_passcodes(self, call: ServiceCall) -> ServiceResponse:
                     removed.append(code.name)
 
         return {"removed": removed}
-    
+
     async def handle_set_autolock(self, call: ServiceCall):
-        """Set the autolock seconds"""
+        """Set the autolock seconds."""
 
         if call.data.get(CONF_ENABLED):
-            seconds = call.data.get(CONF_SECONDS) if call.data.get(CONF_SECONDS) > 0 else 10
+            seconds = (
+                call.data.get(CONF_SECONDS) if call.data.get(CONF_SECONDS) > 0 else 10
+            )
         else:
             seconds = 0
 
@@ -182,4 +186,3 @@ async def handle_set_autolock(self, call: ServiceCall):
                 coordinator.data.auto_lock = call.data.get(CONF_ENABLED)
                 coordinator.data.auto_lock_seconds = seconds
                 coordinator.async_update_listeners()
-

custom_components/ttlock/services.yaml

@@ -125,7 +125,7 @@ set_autolock:
       integration: ttlock
       domain: lock
   fields:
-    enabled: 
+    enabled:
       name: Enabled
       description: Should the Autolock feature for the lock be enabled?
       required: true
@@ -141,5 +141,3 @@ set_autolock:
           min: 0
           max: 60
           unit_of_measurement: secs
-          
-      

custom_components/ttlock/switch.py

@@ -2,7 +2,6 @@
 
 from __future__ import annotations
 
-import logging
 from typing import Any
 
 from homeassistant.components.switch import SwitchDeviceClass, SwitchEntity
@@ -26,14 +25,15 @@ async def async_setup_entry(
             entity
             for coordinator in lock_coordinators(hass, entry)
             for entity in (
-                Autolock(coordinator),
-                Locksound(coordinator),
+                AutoLock(coordinator),
+                LockSound(coordinator),
             )
         ]
     )
 
+
 class AutoLock(BaseLockEntity, SwitchEntity):
-    """The entity object for a switch"""
+    """The entity object for a switch."""
 
     _attr_device_class = SwitchDeviceClass.SWITCH
 
@@ -50,8 +50,9 @@ async def async_turn_off(self, **kwargs: Any) -> None:
         """Turn the entity off."""
         await self.coordinator.auto_lock_off()
 
+
 class LockSound(BaseLockEntity, SwitchEntity):
-    """The entity object for a switch"""
+    """The entity object for a switch."""
 
     _attr_device_class = SwitchDeviceClass.SWITCH
 
@@ -67,4 +68,3 @@ async def async_turn_on(self, **kwargs: Any) -> None:
     async def async_turn_off(self, **kwargs: Any) -> None:
         """Turn the entity off."""
         await self.coordinator.lock_sound_off()
-