Address late code review comment of Prusa MK3 support (#114455)

Address code review comment from #114210 (Prusa MK3 support)
This commit is contained in:
Tereza Tomcova 2024-03-29 21:57:08 +01:00 committed by GitHub
parent 0f710f9fe0
commit 0554ac18b8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -146,15 +146,19 @@ SENSORS: dict[str, tuple[PrusaLinkSensorEntityDescription, ...]] = {
translation_key="progress",
native_unit_of_measurement=PERCENTAGE,
value_fn=lambda data: cast(float, data["progress"]),
available_fn=lambda data: data.get("progress") is not None
and data.get("state") != PrinterState.IDLE.value,
available_fn=lambda data: (
data.get("progress") is not None
and data.get("state") != PrinterState.IDLE.value
),
),
PrusaLinkSensorEntityDescription[JobInfo](
key="job.filename",
translation_key="filename",
value_fn=lambda data: cast(str, data["file"]["display_name"]),
available_fn=lambda data: data.get("file") is not None
and data.get("state") != PrinterState.IDLE.value,
available_fn=lambda data: (
data.get("file") is not None
and data.get("state") != PrinterState.IDLE.value
),
),
PrusaLinkSensorEntityDescription[JobInfo](
key="job.start",
@ -164,8 +168,10 @@ SENSORS: dict[str, tuple[PrusaLinkSensorEntityDescription, ...]] = {
lambda data: (utcnow() - timedelta(seconds=data["time_printing"])),
timedelta(minutes=2),
),
available_fn=lambda data: data.get("time_printing") is not None
and data.get("state") != PrinterState.IDLE.value,
available_fn=lambda data: (
data.get("time_printing") is not None
and data.get("state") != PrinterState.IDLE.value
),
),
PrusaLinkSensorEntityDescription[JobInfo](
key="job.finish",
@ -175,8 +181,10 @@ SENSORS: dict[str, tuple[PrusaLinkSensorEntityDescription, ...]] = {
lambda data: (utcnow() + timedelta(seconds=data["time_remaining"])),
timedelta(minutes=2),
),
available_fn=lambda data: data.get("time_remaining") is not None
and data.get("state") != PrinterState.IDLE.value,
available_fn=lambda data: (
data.get("time_remaining") is not None
and data.get("state") != PrinterState.IDLE.value
),
),
),
}