Updated apps

custom_components/mail_and_packages/helpers.py

@@ -0,0 +1,131 @@
+"""Helper functions for Mail and Packages."""
+
+from __future__ import annotations
+
+import datetime
+import logging
+import os
+from pathlib import Path
+from shutil import copyfile
+from typing import Any
+
+from aioimaplib import IMAP4_SSL
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+
+from .const import (
+    AMAZON_DELIVERED_SUBJECT,
+    AMAZON_EXCEPTION_SUBJECT,
+    AMAZON_HUB_SUBJECT,
+    AMAZON_OTP_SUBJECT,
+    ATTR_COUNT,
+    ATTR_TRACKING,
+    BINARY_SENSORS,
+    CONF_PATH,
+    SENSOR_TYPES,
+)
+from .shippers import SHIPPER_REGISTRY
+from .shippers.usps import USPSShipper
+from .utils.image import (
+    default_image_path,
+)
+
+_LOGGER = logging.getLogger(__name__)
+
+# Legacy constants for tests
+
+amazon_exception = AMAZON_EXCEPTION_SUBJECT
+amazon_hub = AMAZON_HUB_SUBJECT
+amazon_otp = AMAZON_OTP_SUBJECT
+amazon_search_legacy = AMAZON_DELIVERED_SUBJECT
+image_file_name = "mail_today.gif"
+
+
+async def get_count(
+    account: IMAP4_SSL,
+    sensor_type: str,
+    body_count: bool,
+    image_path: str,
+    hass: HomeAssistant,
+) -> dict[str, Any]:
+    """Legacy get_count wrapper for tests."""
+    shipper_class = SHIPPER_REGISTRY.get(sensor_type)
+    if not shipper_class:
+        _LOGGER.error("No shipper found for %s", sensor_type)
+        return {ATTR_COUNT: 0, ATTR_TRACKING: []}
+
+    shipper = shipper_class(hass, {CONF_PATH: image_path})
+    today = datetime.datetime.now().strftime("%d-%b-%Y")
+    return await shipper.process(account, today, sensor_type)
+
+
+async def get_mails(
+    account: IMAP4_SSL,
+    address_list: list[str],
+    hass: HomeAssistant,
+    config: dict[str, Any],
+) -> int:
+    """Legacy get_mails wrapper for tests."""
+    shipper = USPSShipper(hass, config)
+    today = datetime.datetime.now().strftime("%d-%b-%Y")
+    result = await shipper.process(account, today, "usps_mail")
+    return result.get(ATTR_COUNT, 0)
+
+
+async def fetch(
+    hass: HomeAssistant,
+    config: dict[str, Any],
+    account: IMAP4_SSL,
+    sensor: str,
+) -> int:
+    """Legacy fetch wrapper for tests."""
+    shipper_class = SHIPPER_REGISTRY.get(sensor)
+    if not shipper_class:
+        return 0
+
+    shipper = shipper_class(hass, config)
+    today = datetime.datetime.now().strftime("%d-%b-%Y")
+    data = await shipper.process(account, today, sensor)
+    return data.get(ATTR_COUNT, 0)
+
+
+async def get_items(
+    hass: HomeAssistant,
+    config: dict[str, Any],
+    account: IMAP4_SSL,
+    sensor: str,
+) -> dict[str, Any]:
+    """Legacy get_items wrapper for tests."""
+    shipper_class = SHIPPER_REGISTRY.get(sensor)
+    if not shipper_class:
+        return {ATTR_COUNT: 0, ATTR_TRACKING: []}
+
+    shipper = shipper_class(hass, config)
+    today = datetime.datetime.now().strftime("%d-%b-%Y")
+    return await shipper.process(account, today, sensor)
+
+
+def get_resources(hass: HomeAssistant | None = None) -> dict:
+    """Return resources from const."""
+    resources = {k: v.name for k, v in SENSOR_TYPES.items()}
+    resources.update({k: v.name for k, v in BINARY_SENSORS.items() if v.selectable})
+    return resources
+
+
+def copy_images(hass: HomeAssistant, config: ConfigEntry) -> None:
+    """Copy processed images to www directory."""
+    image_path = Path(hass.config.path()) / default_image_path(hass, config)
+    www_path = Path(hass.config.path()) / "www" / "mail_and_packages"
+
+    if not www_path.is_dir():
+        www_path.mkdir(parents=True, exist_ok=True)
+
+    for root, _, files in os.walk(image_path):
+        for file in files:
+            if file.endswith((".gif", ".jpg", ".png")):
+                src = Path(root) / file
+                dest = www_path / file
+                try:
+                    copyfile(str(src), str(dest))
+                except OSError as err:
+                    _LOGGER.error("Error copying image: %s", err)