evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use new enums in brother (#61327)

Browse source 
Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-12-09 09:17:50 +01:00 • committed by GitHub
parent 9d25961bf3
commit 4818872591
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 51 additions and 55 deletions
Download patch file Download diff file Expand all files Collapse all files

106
homeassistant/components/brother/sensor.py
View file

@ -6,19 +6,15 @@ from datetime import datetime
from typing import Any, cast from typing import Any, cast
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
STATE_CLASS_MEASUREMENT, SensorDeviceClass,
SensorEntity, SensorEntity,
SensorEntityDescription, SensorEntityDescription,
SensorStateClass,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ( from homeassistant.const import CONF_HOST, PERCENTAGE
CONF_HOST,
DEVICE_CLASS_TIMESTAMP,
ENTITY_CATEGORY_DIAGNOSTIC,
PERCENTAGE,
)
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity import DeviceInfo from homeassistant.helpers.entity import DeviceInfo, EntityCategory
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType from homeassistant.helpers.typing import StateType
from homeassistant.helpers.update_coordinator import CoordinatorEntity from homeassistant.helpers.update_coordinator import CoordinatorEntity
@ -173,190 +169,190 @@ SENSOR_TYPES: tuple[BrotherSensorEntityDescription, ...] = (
key=ATTR_STATUS, key=ATTR_STATUS,
icon="mdi:printer", icon="mdi:printer",
name=ATTR_STATUS.title(), name=ATTR_STATUS.title(),
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_PAGE_COUNTER, key=ATTR_PAGE_COUNTER,
icon="mdi:file-document-outline", icon="mdi:file-document-outline",
name=ATTR_PAGE_COUNTER.replace("_", " ").title(), name=ATTR_PAGE_COUNTER.replace("_", " ").title(),
native_unit_of_measurement=UNIT_PAGES, native_unit_of_measurement=UNIT_PAGES,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_BW_COUNTER, key=ATTR_BW_COUNTER,
icon="mdi:file-document-outline", icon="mdi:file-document-outline",
name=ATTR_BW_COUNTER.replace("_", " ").title(), name=ATTR_BW_COUNTER.replace("_", " ").title(),
native_unit_of_measurement=UNIT_PAGES, native_unit_of_measurement=UNIT_PAGES,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_COLOR_COUNTER, key=ATTR_COLOR_COUNTER,
icon="mdi:file-document-outline", icon="mdi:file-document-outline",
name=ATTR_COLOR_COUNTER.replace("_", " ").title(), name=ATTR_COLOR_COUNTER.replace("_", " ").title(),
native_unit_of_measurement=UNIT_PAGES, native_unit_of_measurement=UNIT_PAGES,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_DUPLEX_COUNTER, key=ATTR_DUPLEX_COUNTER,
icon="mdi:file-document-outline", icon="mdi:file-document-outline",
name=ATTR_DUPLEX_COUNTER.replace("_", " ").title(), name=ATTR_DUPLEX_COUNTER.replace("_", " ").title(),
native_unit_of_measurement=UNIT_PAGES, native_unit_of_measurement=UNIT_PAGES,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_DRUM_REMAINING_LIFE, key=ATTR_DRUM_REMAINING_LIFE,
icon="mdi:chart-donut", icon="mdi:chart-donut",
name=ATTR_DRUM_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_DRUM_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_BLACK_DRUM_REMAINING_LIFE, key=ATTR_BLACK_DRUM_REMAINING_LIFE,
icon="mdi:chart-donut", icon="mdi:chart-donut",
name=ATTR_BLACK_DRUM_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_BLACK_DRUM_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_CYAN_DRUM_REMAINING_LIFE, key=ATTR_CYAN_DRUM_REMAINING_LIFE,
icon="mdi:chart-donut", icon="mdi:chart-donut",
name=ATTR_CYAN_DRUM_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_CYAN_DRUM_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_MAGENTA_DRUM_REMAINING_LIFE, key=ATTR_MAGENTA_DRUM_REMAINING_LIFE,
icon="mdi:chart-donut", icon="mdi:chart-donut",
name=ATTR_MAGENTA_DRUM_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_MAGENTA_DRUM_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_YELLOW_DRUM_REMAINING_LIFE, key=ATTR_YELLOW_DRUM_REMAINING_LIFE,
icon="mdi:chart-donut", icon="mdi:chart-donut",
name=ATTR_YELLOW_DRUM_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_YELLOW_DRUM_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_BELT_UNIT_REMAINING_LIFE, key=ATTR_BELT_UNIT_REMAINING_LIFE,
icon="mdi:current-ac", icon="mdi:current-ac",
name=ATTR_BELT_UNIT_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_BELT_UNIT_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_FUSER_REMAINING_LIFE, key=ATTR_FUSER_REMAINING_LIFE,
icon="mdi:water-outline", icon="mdi:water-outline",
name=ATTR_FUSER_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_FUSER_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_LASER_REMAINING_LIFE, key=ATTR_LASER_REMAINING_LIFE,
icon="mdi:spotlight-beam", icon="mdi:spotlight-beam",
name=ATTR_LASER_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_LASER_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_PF_KIT_1_REMAINING_LIFE, key=ATTR_PF_KIT_1_REMAINING_LIFE,
icon="mdi:printer-3d", icon="mdi:printer-3d",
name=ATTR_PF_KIT_1_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_PF_KIT_1_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_PF_KIT_MP_REMAINING_LIFE, key=ATTR_PF_KIT_MP_REMAINING_LIFE,
icon="mdi:printer-3d", icon="mdi:printer-3d",
name=ATTR_PF_KIT_MP_REMAINING_LIFE.replace("_", " ").title(), name=ATTR_PF_KIT_MP_REMAINING_LIFE.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_BLACK_TONER_REMAINING, key=ATTR_BLACK_TONER_REMAINING,
icon="mdi:printer-3d-nozzle", icon="mdi:printer-3d-nozzle",
name=ATTR_BLACK_TONER_REMAINING.replace("_", " ").title(), name=ATTR_BLACK_TONER_REMAINING.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_CYAN_TONER_REMAINING, key=ATTR_CYAN_TONER_REMAINING,
icon="mdi:printer-3d-nozzle", icon="mdi:printer-3d-nozzle",
name=ATTR_CYAN_TONER_REMAINING.replace("_", " ").title(), name=ATTR_CYAN_TONER_REMAINING.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_MAGENTA_TONER_REMAINING, key=ATTR_MAGENTA_TONER_REMAINING,
icon="mdi:printer-3d-nozzle", icon="mdi:printer-3d-nozzle",
name=ATTR_MAGENTA_TONER_REMAINING.replace("_", " ").title(), name=ATTR_MAGENTA_TONER_REMAINING.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_YELLOW_TONER_REMAINING, key=ATTR_YELLOW_TONER_REMAINING,
icon="mdi:printer-3d-nozzle", icon="mdi:printer-3d-nozzle",
name=ATTR_YELLOW_TONER_REMAINING.replace("_", " ").title(), name=ATTR_YELLOW_TONER_REMAINING.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_BLACK_INK_REMAINING, key=ATTR_BLACK_INK_REMAINING,
icon="mdi:printer-3d-nozzle", icon="mdi:printer-3d-nozzle",
name=ATTR_BLACK_INK_REMAINING.replace("_", " ").title(), name=ATTR_BLACK_INK_REMAINING.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_CYAN_INK_REMAINING, key=ATTR_CYAN_INK_REMAINING,
icon="mdi:printer-3d-nozzle", icon="mdi:printer-3d-nozzle",
name=ATTR_CYAN_INK_REMAINING.replace("_", " ").title(), name=ATTR_CYAN_INK_REMAINING.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_MAGENTA_INK_REMAINING, key=ATTR_MAGENTA_INK_REMAINING,
icon="mdi:printer-3d-nozzle", icon="mdi:printer-3d-nozzle",
name=ATTR_MAGENTA_INK_REMAINING.replace("_", " ").title(), name=ATTR_MAGENTA_INK_REMAINING.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_YELLOW_INK_REMAINING, key=ATTR_YELLOW_INK_REMAINING,
icon="mdi:printer-3d-nozzle", icon="mdi:printer-3d-nozzle",
name=ATTR_YELLOW_INK_REMAINING.replace("_", " ").title(), name=ATTR_YELLOW_INK_REMAINING.replace("_", " ").title(),
native_unit_of_measurement=PERCENTAGE, native_unit_of_measurement=PERCENTAGE,
state_class=STATE_CLASS_MEASUREMENT, state_class=SensorStateClass.MEASUREMENT,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
), ),
BrotherSensorEntityDescription( BrotherSensorEntityDescription(
key=ATTR_UPTIME, key=ATTR_UPTIME,
name=ATTR_UPTIME.title(), name=ATTR_UPTIME.title(),
entity_registry_enabled_default=False, entity_registry_enabled_default=False,
device_class=DEVICE_CLASS_TIMESTAMP, device_class=SensorDeviceClass.TIMESTAMP,
entity_category=ENTITY_CATEGORY_DIAGNOSTIC, entity_category=EntityCategory.DIAGNOSTIC,
entity_class=BrotherPrinterUptimeSensor, entity_class=BrotherPrinterUptimeSensor,
), ),
) )