46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""Subscription information."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import logging
|
|
from typing import Any
|
|
|
|
from aiohttp.client_exceptions import ClientError
|
|
from hass_nabucasa import Cloud, cloud_api
|
|
from hass_nabucasa.payments_api import PaymentsApiError, SubscriptionInfo
|
|
|
|
from .client import CloudClient
|
|
from .const import REQUEST_TIMEOUT
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_subscription_info(cloud: Cloud[CloudClient]) -> SubscriptionInfo | None:
|
|
"""Fetch the subscription info."""
|
|
try:
|
|
async with asyncio.timeout(REQUEST_TIMEOUT):
|
|
return await cloud.payments.subscription_info()
|
|
except PaymentsApiError as exception:
|
|
_LOGGER.error("Failed to fetch subscription information - %s", exception)
|
|
|
|
return None
|
|
|
|
|
|
async def async_migrate_paypal_agreement(
|
|
cloud: Cloud[CloudClient],
|
|
) -> dict[str, Any] | None:
|
|
"""Migrate a paypal agreement from legacy."""
|
|
try:
|
|
async with asyncio.timeout(REQUEST_TIMEOUT):
|
|
return await cloud_api.async_migrate_paypal_agreement(cloud)
|
|
except TimeoutError:
|
|
_LOGGER.error(
|
|
"A timeout of %s was reached while trying to start agreement migration",
|
|
REQUEST_TIMEOUT,
|
|
)
|
|
except ClientError as exception:
|
|
_LOGGER.error("Failed to start agreement migration - %s", exception)
|
|
|
|
return None
|