Fix lint

custom_components/truenas/config_flow.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 import logging
+from typing import Any
 
 import voluptuous as vol
 from truenaspy import (

custom_components/truenas/sensor.py

@@ -2,7 +2,8 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from datetime import datetime as dt
+from datetime import date, datetime
+from decimal import Decimal
 from logging import getLogger
 from typing import Any, Final, Mapping
 
@@ -23,6 +24,7 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import entity_platform
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
+from homeassistant.helpers.typing import StateType
 
 from .const import (
     DOMAIN,
@@ -307,7 +309,7 @@ async def async_setup_entry(
 
 class Sensor(TruenasEntity, SensorEntity):
     @property
-    def native_value(self) -> StateType | datetime | Decimal:
+    def native_value(self) -> StateType | date | datetime | Decimal:
         """Return the value reported by the sensor."""
         return self.datas.get(self.entity_description.attr)
 
@@ -316,9 +318,9 @@ class UptimeSensor(Sensor):
     """Define an Truenas Uptime sensor."""
 
     @property
-    def native_value(self) -> StateType | datetime | Decimal:
+    def native_value(self) -> StateType | date | datetime | Decimal:
         """Return the value reported by the sensor."""
-        return dt.strptime(
+        return datetime.strptime(
             self.datas.get(self.entity_description.attr), "%Y-%m-%dT%H:%M:%S%z"
         )
 
@@ -336,7 +338,7 @@ class DatasetSensor(Sensor):
 
     async def snapshot(self) -> None:
         """Create dataset snapshot."""
-        ts = dt.now().isoformat(sep="_", timespec="microseconds")
+        ts = datetime.now().isoformat(sep="_", timespec="microseconds")
         await self.coordinator.api.query(
             "zfs/snapshot",
             method="post",