epenet
d766aae436
Co-authored-by: copilot-swe-agent[bot] <198982749+Copilot@users.noreply.github.com> Co-authored-by: frenck <195327+frenck@users.noreply.github.com>
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
"""Diagnostics support for AWS S3."""
|
|
|
|
import dataclasses
|
|
from typing import Any
|
|
|
|
from homeassistant.components.backup import DATA_MANAGER as BACKUP_DATA_MANAGER
|
|
from homeassistant.components.diagnostics import async_redact_data
|
|
from homeassistant.core import HomeAssistant
|
|
|
|
from .const import (
|
|
CONF_ACCESS_KEY_ID,
|
|
CONF_BUCKET,
|
|
CONF_PREFIX,
|
|
CONF_SECRET_ACCESS_KEY,
|
|
DOMAIN,
|
|
)
|
|
from .coordinator import S3ConfigEntry
|
|
from .helpers import async_list_backups_from_s3
|
|
|
|
TO_REDACT = (CONF_ACCESS_KEY_ID, CONF_SECRET_ACCESS_KEY)
|
|
|
|
|
|
async def async_get_config_entry_diagnostics(
|
|
hass: HomeAssistant,
|
|
entry: S3ConfigEntry,
|
|
) -> dict[str, Any]:
|
|
"""Return diagnostics for a config entry."""
|
|
coordinator = entry.runtime_data
|
|
backup_manager = hass.data[BACKUP_DATA_MANAGER]
|
|
backups = await async_list_backups_from_s3(
|
|
coordinator.client,
|
|
bucket=entry.data[CONF_BUCKET],
|
|
prefix=entry.data.get(CONF_PREFIX, ""),
|
|
)
|
|
|
|
data = {
|
|
"coordinator_data": dataclasses.asdict(coordinator.data),
|
|
"config": {
|
|
**entry.data,
|
|
**entry.options,
|
|
},
|
|
"backup_agents": [
|
|
{"name": agent.name}
|
|
for agent in backup_manager.backup_agents.values()
|
|
if agent.domain == DOMAIN
|
|
],
|
|
"backup": [backup.as_dict() for backup in backups],
|
|
}
|
|
|
|
return async_redact_data(data, TO_REDACT)
|