60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
"""The NZBGet integration."""
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.core import HomeAssistant, ServiceCall, callback
|
|
from homeassistant.exceptions import ServiceValidationError
|
|
from homeassistant.helpers import config_validation as cv
|
|
|
|
from .const import (
|
|
ATTR_SPEED,
|
|
DATA_COORDINATOR,
|
|
DEFAULT_SPEED_LIMIT,
|
|
DOMAIN,
|
|
SERVICE_PAUSE,
|
|
SERVICE_RESUME,
|
|
SERVICE_SET_SPEED,
|
|
)
|
|
from .coordinator import NZBGetDataUpdateCoordinator
|
|
|
|
SPEED_LIMIT_SCHEMA = vol.Schema(
|
|
{vol.Optional(ATTR_SPEED, default=DEFAULT_SPEED_LIMIT): cv.positive_int}
|
|
)
|
|
|
|
|
|
def _get_coordinator(call: ServiceCall) -> NZBGetDataUpdateCoordinator:
|
|
"""Service call to pause downloads in NZBGet."""
|
|
entries = call.hass.config_entries.async_loaded_entries(DOMAIN)
|
|
if not entries:
|
|
raise ServiceValidationError(
|
|
translation_domain=DOMAIN,
|
|
translation_key="invalid_config_entry",
|
|
)
|
|
return call.hass.data[DOMAIN][entries[0].entry_id][DATA_COORDINATOR]
|
|
|
|
|
|
def pause(call: ServiceCall) -> None:
|
|
"""Service call to pause downloads in NZBGet."""
|
|
_get_coordinator(call).nzbget.pausedownload()
|
|
|
|
|
|
def resume(call: ServiceCall) -> None:
|
|
"""Service call to resume downloads in NZBGet."""
|
|
_get_coordinator(call).nzbget.resumedownload()
|
|
|
|
|
|
def set_speed(call: ServiceCall) -> None:
|
|
"""Service call to rate limit speeds in NZBGet."""
|
|
_get_coordinator(call).nzbget.rate(call.data[ATTR_SPEED])
|
|
|
|
|
|
@callback
|
|
def async_setup_services(hass: HomeAssistant) -> None:
|
|
"""Register integration-level services."""
|
|
|
|
hass.services.async_register(DOMAIN, SERVICE_PAUSE, pause, schema=vol.Schema({}))
|
|
hass.services.async_register(DOMAIN, SERVICE_RESUME, resume, schema=vol.Schema({}))
|
|
hass.services.async_register(
|
|
DOMAIN, SERVICE_SET_SPEED, set_speed, schema=SPEED_LIMIT_SCHEMA
|
|
)
|