camera.py

Go to the documentation of this file.

"""Camera entity for PrusaLink."""
 
from __future__ import annotations
 
from pyprusalink.types import PrinterState
 
from homeassistant.components.camera import Camera
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from .const import DOMAIN
from .coordinator import JobUpdateCoordinator
from .entity import PrusaLinkEntity
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    entry: ConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up PrusaLink camera."""
    coordinator: JobUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]["job"]
    async_add_entities([PrusaLinkJobPreviewEntity(coordinator)])
 
 
class PrusaLinkJobPreviewEntity(PrusaLinkEntity, Camera):
    """Defines a PrusaLink camera."""
 
    last_path = ""
    last_image: bytes
    _attr_translation_key = "job_preview"
 
    def __init__(self, coordinator: JobUpdateCoordinator) -> None:
        """Initialize a PrusaLink camera entity."""
        super().__init__(coordinator)
        Camera.__init__(self)
        self._attr_unique_id_attr_unique_id = f"{self.coordinator.config_entry.entry_id}_job_preview"
 
    @property
    def available(self) -> bool:
        """Get if camera is available."""
        return (
            super().available
            and self.coordinator.data.get("state") != PrinterState.IDLE.value
            and (file := self.coordinator.data.get("file"))
            and file.get("refs", {}).get("thumbnail")
        )
 
    async def async_camera_image(
        self, width: int | None = None, height: int | None = None
    ) -> bytes | None:
        """Return a still image from the camera."""
        if not self.availableavailableavailableavailableavailable:
            return None
 
        path = self.coordinator.data["file"]["refs"]["thumbnail"]
 
        if self.last_pathlast_pathlast_path == path:
            return self.last_imagelast_image
 
        self.last_imagelast_image = await self.coordinator.api.get_file(path)
        self.last_pathlast_pathlast_path = path
        return self.last_imagelast_image