49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
"""Spook - Your homie."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.components.homeassistant import DOMAIN
|
|
from homeassistant.exceptions import HomeAssistantError
|
|
from homeassistant.helpers import config_validation as cv
|
|
|
|
from ....const import DOMAIN as SPOOK_DOMAIN
|
|
from ....errors import user_not_found
|
|
from ....services import AbstractSpookAdminService
|
|
|
|
if TYPE_CHECKING:
|
|
from homeassistant.core import ServiceCall
|
|
|
|
|
|
class SpookService(AbstractSpookAdminService):
|
|
"""Home Assistant Core integration service to disable a user."""
|
|
|
|
domain = DOMAIN
|
|
service = "disable_user"
|
|
schema = {vol.Required("user_id"): vol.All(cv.ensure_list, [cv.string])}
|
|
|
|
async def async_handle_service(self, call: ServiceCall) -> None:
|
|
"""Handle the service call."""
|
|
# Everything is looked up before anything is written. A bad one late
|
|
# in the list should not leave the first ones changed behind an error.
|
|
users = []
|
|
for user_id in call.data["user_id"]:
|
|
user = await self.hass.auth.async_get_user(user_id)
|
|
if user is None:
|
|
raise user_not_found(user_id)
|
|
if user.system_generated:
|
|
raise HomeAssistantError(
|
|
translation_domain=SPOOK_DOMAIN,
|
|
translation_key="cannot_disable_system_user",
|
|
translation_placeholders={
|
|
"user_id": user_id,
|
|
},
|
|
)
|
|
users.append(user)
|
|
|
|
for user in users:
|
|
await self.hass.auth.async_update_user(user, is_active=False)
|