mirror of
https://github.com/home-assistant/core.git
synced 2025-10-08 11:19:30 +00:00
23 lines
630 B
Python
23 lines
630 B
Python
"""Expose images as media sources."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.components.media_source import MediaSource, local_source
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .const import DATA_MEDIA_SOURCE, DOMAIN, IMAGE_DIR
|
|
|
|
|
|
async def async_get_media_source(hass: HomeAssistant) -> MediaSource:
|
|
"""Set up local media source."""
|
|
media_dir = hass.config.path(f"{DOMAIN}/{IMAGE_DIR}")
|
|
|
|
hass.data[DATA_MEDIA_SOURCE] = source = local_source.LocalSource(
|
|
hass,
|
|
DOMAIN,
|
|
"AI Generated Images",
|
|
{IMAGE_DIR: media_dir},
|
|
f"/{DOMAIN}",
|
|
)
|
|
return source
|