Explorar o código

Add MDI icons to sensors and default wifi_ssid to 4114

ZiB hai 4 meses
pai
achega
ad623674ba

+ 1 - 1
custom_components/room_cal_ha/models.py

@@ -12,7 +12,7 @@ class ReservationData:
   check_in_time: time | None = time(14, 0)
   check_in_time: time | None = time(14, 0)
   check_out_date: date | None = None
   check_out_date: date | None = None
   check_out_time: time | None = time(12, 0)
   check_out_time: time | None = time(12, 0)
-  wifi_ssid: str | None = None
+  wifi_ssid: str | None = "4114"
   wifi_key: str | None = None
   wifi_key: str | None = None
   door_code: str | None = None
   door_code: str | None = None
 
 

+ 12 - 9
custom_components/room_cal_ha/sensor.py

@@ -42,14 +42,14 @@ async def async_setup_entry(
       name=device_name,
       name=device_name,
     )
     )
     entities += [
     entities += [
-      ReservationStrSensor(coordinator, entry, slot, device_info, "Guest Name", "guest_name"),
-      ReservationDateTimeSensor(coordinator, entry, slot, device_info, "Check-In Date", "check_in_date", "check_in_time"),
-      ReservationTimeSensor(coordinator, entry, slot, device_info, "Check-In Time", "check_in_time"),
-      ReservationDateTimeSensor(coordinator, entry, slot, device_info, "Check-Out Date", "check_out_date", "check_out_time"),
-      ReservationTimeSensor(coordinator, entry, slot, device_info, "Check-Out Time", "check_out_time"),
-      ReservationStrSensor(coordinator, entry, slot, device_info, "WiFi SSID", "wifi_ssid"),
-      ReservationStrSensor(coordinator, entry, slot, device_info, "WiFi Key", "wifi_key"),
-      ReservationStrSensor(coordinator, entry, slot, device_info, "Door Code", "door_code"),
+      ReservationStrSensor(coordinator, entry, slot, device_info, "Guest Name", "guest_name", "mdi:account-convert"),
+      ReservationDateTimeSensor(coordinator, entry, slot, device_info, "Check-In Date", "check_in_date", "check_in_time", "mdi:calendar-start"),
+      ReservationTimeSensor(coordinator, entry, slot, device_info, "Check-In Time", "check_in_time", "mdi:timer-check-outline"),
+      ReservationDateTimeSensor(coordinator, entry, slot, device_info, "Check-Out Date", "check_out_date", "check_out_time", "mdi:calendar-end"),
+      ReservationTimeSensor(coordinator, entry, slot, device_info, "Check-Out Time", "check_out_time", "mdi:timer-cancel-outline"),
+      ReservationStrSensor(coordinator, entry, slot, device_info, "WiFi SSID", "wifi_ssid", "mdi:wifi"),
+      ReservationStrSensor(coordinator, entry, slot, device_info, "WiFi Key", "wifi_key", "mdi:wifi-lock"),
+      ReservationStrSensor(coordinator, entry, slot, device_info, "Door Code", "door_code", "mdi:door-closed-lock"),
       LastRefreshSensor(coordinator, entry, slot, device_info),
       LastRefreshSensor(coordinator, entry, slot, device_info),
     ]
     ]
 
 
@@ -67,6 +67,7 @@ class _ReservationSensor(CoordinatorEntity[RoomCalCoordinator], SensorEntity):
     device_info: DeviceInfo,
     device_info: DeviceInfo,
     name: str,
     name: str,
     field: str,
     field: str,
+    icon: str | None = None,
   ) -> None:
   ) -> None:
     super().__init__(coordinator)
     super().__init__(coordinator)
     self._slot = slot
     self._slot = slot
@@ -74,6 +75,7 @@ class _ReservationSensor(CoordinatorEntity[RoomCalCoordinator], SensorEntity):
     self._attr_name = name
     self._attr_name = name
     self._attr_unique_id = f"{entry.entry_id}_{slot}_{field}"
     self._attr_unique_id = f"{entry.entry_id}_{slot}_{field}"
     self._attr_device_info = device_info
     self._attr_device_info = device_info
+    self._attr_icon = icon
 
 
   def _reservation(self) -> ReservationData:
   def _reservation(self) -> ReservationData:
     data: CoordinatorData = self.coordinator.data
     data: CoordinatorData = self.coordinator.data
@@ -126,8 +128,9 @@ class ReservationDateTimeSensor(_ReservationSensor):
     name: str,
     name: str,
     field: str,
     field: str,
     time_field: str,
     time_field: str,
+    icon: str | None = None,
   ) -> None:
   ) -> None:
-    super().__init__(coordinator, entry, slot, device_info, name, field)
+    super().__init__(coordinator, entry, slot, device_info, name, field, icon)
     self._time_field = time_field
     self._time_field = time_field
 
 
   @property
   @property