use constants

This commit is contained in:
mib1185
2025-06-22 18:36:31 +00:00
parent 301dfeb07e
commit 816ece9506

View File

@@ -15,12 +15,16 @@ from .coordinator import ImmichDataUpdateCoordinator
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
CONF_ALBUM_ID = "album_id"
CONF_CONFIG_ENTRY_ID = "config_entry_id"
CONF_FILE = "file"
SERVICE_UPLOAD_FILE = "upload_file" SERVICE_UPLOAD_FILE = "upload_file"
SERVICE_SCHEMA_UPLOAD_FILE = vol.Schema( SERVICE_SCHEMA_UPLOAD_FILE = vol.Schema(
{ {
vol.Required("config_entry_id"): str, vol.Required(CONF_CONFIG_ENTRY_ID): str,
vol.Required("file"): str, vol.Required(CONF_FILE): str,
vol.Optional("album_id"): str, vol.Optional(CONF_ALBUM_ID): str,
} }
) )
@@ -34,9 +38,9 @@ async def _async_upload_file(service_call: ServiceCall) -> None:
) )
hass = service_call.hass hass = service_call.hass
target_entry = hass.config_entries.async_get_entry( target_entry = hass.config_entries.async_get_entry(
service_call.data["config_entry_id"] service_call.data[CONF_CONFIG_ENTRY_ID]
) )
target_file = service_call.data["file"] target_file = service_call.data[CONF_FILE]
if not target_entry: if not target_entry:
raise ServiceValidationError( raise ServiceValidationError(
@@ -64,7 +68,7 @@ async def _async_upload_file(service_call: ServiceCall) -> None:
coordinator: ImmichDataUpdateCoordinator = target_entry.runtime_data coordinator: ImmichDataUpdateCoordinator = target_entry.runtime_data
if target_album := service_call.data.get("album_id"): if target_album := service_call.data.get(CONF_ALBUM_ID):
try: try:
await coordinator.api.albums.async_get_album_info(target_album, True) await coordinator.api.albums.async_get_album_info(target_album, True)
except ImmichError as ex: except ImmichError as ex: