101 lines
3.1 KiB
Python
101 lines
3.1 KiB
Python
"""The seventeentrack component."""
|
|
|
|
from py17track import Client as SeventeenTrackClient
|
|
from py17track.errors import SeventeenTrackError
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import (
|
|
ATTR_FRIENDLY_NAME,
|
|
ATTR_LOCATION,
|
|
CONF_PASSWORD,
|
|
CONF_USERNAME,
|
|
Platform,
|
|
)
|
|
from homeassistant.core import (
|
|
HomeAssistant,
|
|
ServiceCall,
|
|
ServiceResponse,
|
|
SupportsResponse,
|
|
)
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
from homeassistant.helpers import config_validation as cv
|
|
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
|
from homeassistant.helpers.typing import ConfigType
|
|
from homeassistant.util import slugify
|
|
|
|
from .const import (
|
|
ATTR_CONFIG_ENTRY_ID,
|
|
ATTR_INFO_TEXT,
|
|
ATTR_PACKAGE_STATE,
|
|
ATTR_STATUS,
|
|
ATTR_TIMESTAMP,
|
|
ATTR_TRACKING_NUMBER,
|
|
DOMAIN,
|
|
SERVICE_GET_PACKAGES,
|
|
)
|
|
from .coordinator import SeventeenTrackCoordinator
|
|
|
|
PLATFORMS: list[Platform] = [Platform.SENSOR]
|
|
|
|
CONFIG_SCHEMA = cv.empty_config_schema(DOMAIN)
|
|
|
|
|
|
async def async_setup(hass: HomeAssistant, config: ConfigType) -> bool:
|
|
"""Set up the 17Track component."""
|
|
|
|
async def get_packages(call: ServiceCall) -> ServiceResponse:
|
|
"""Get packages from 17Track."""
|
|
config_entry_id = call.data[ATTR_CONFIG_ENTRY_ID]
|
|
package_states = call.data.get(ATTR_PACKAGE_STATE, [])
|
|
seventeen_coordinator: SeventeenTrackCoordinator = hass.data[DOMAIN][
|
|
config_entry_id
|
|
]
|
|
live_packages = sorted(
|
|
await seventeen_coordinator.client.profile.packages(
|
|
show_archived=seventeen_coordinator.show_archived
|
|
)
|
|
)
|
|
|
|
return {
|
|
"packages": [
|
|
{
|
|
ATTR_TRACKING_NUMBER: package.tracking_number,
|
|
ATTR_LOCATION: package.location,
|
|
ATTR_STATUS: package.status,
|
|
ATTR_TIMESTAMP: package.timestamp,
|
|
ATTR_INFO_TEXT: package.info_text,
|
|
ATTR_FRIENDLY_NAME: package.friendly_name,
|
|
}
|
|
for package in live_packages
|
|
if slugify(package.status) in package_states or package_states == []
|
|
]
|
|
}
|
|
|
|
hass.services.async_register(
|
|
DOMAIN,
|
|
SERVICE_GET_PACKAGES,
|
|
get_packages,
|
|
supports_response=SupportsResponse.ONLY,
|
|
)
|
|
return True
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|
"""Set up 17Track from a config entry."""
|
|
|
|
session = async_get_clientsession(hass)
|
|
client = SeventeenTrackClient(session=session)
|
|
|
|
try:
|
|
await client.profile.login(entry.data[CONF_USERNAME], entry.data[CONF_PASSWORD])
|
|
except SeventeenTrackError as err:
|
|
raise ConfigEntryNotReady from err
|
|
|
|
seventeen_coordinator = SeventeenTrackCoordinator(hass, client)
|
|
|
|
await seventeen_coordinator.async_config_entry_first_refresh()
|
|
|
|
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = seventeen_coordinator
|
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
|
return True
|