Make image links update only when media changes, fix login errors, and reuse HA sessions for better performance

Author: mkanet

custom_components/plex_recently_added/__init__.py

@@ -50,7 +50,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
         raise ConfigEntryNotReady("Failed to Log-in") from err
     coordinator = PlexDataCoordinator(hass, client)
 
-    hass.http.register_view(ImagesRedirect(config_entry))
+    hass.http.register_view(ImagesRedirect(hass, config_entry))
     await coordinator.async_config_entry_first_refresh()
     hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = coordinator
 

custom_components/plex_recently_added/const.py

@@ -2,6 +2,8 @@
 
 DOMAIN: Final = "plex_recently_added"
 TIMEOUT_MINUTES: Final = 10
+POLL_INTERVAL_MINUTES: Final = 10
+SIGN_URL_TTL_MINUTES: Final = 10080  # 7 days
 
 
 DEFAULT_NAME: Final = 'Plex Recently Added'

custom_components/plex_recently_added/coordinator.py

@@ -6,7 +6,7 @@
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryError
 
-from .const import DOMAIN, TIMEOUT_MINUTES
+from .const import DOMAIN, POLL_INTERVAL_MINUTES
 from .plex_api import (
     PlexApi,
     FailedToLogin,
@@ -23,11 +23,11 @@ def __init__(self, hass: HomeAssistant, client: PlexApi):
             _LOGGER,
             name=DOMAIN,
             update_method=self._async_update_data,
-            update_interval=timedelta(minutes=TIMEOUT_MINUTES),
+            update_interval=timedelta(minutes=POLL_INTERVAL_MINUTES),
         )
 
     async def _async_update_data(self) -> Dict[str, Any]:
         try:
             return await self._client.update()
-        except FailedToLogin:
+        except FailedToLogin as err:
             raise ConfigEntryError("Failed to Log-in") from err

custom_components/plex_recently_added/manifest.json

@@ -9,5 +9,5 @@
   "iot_class": "local_polling",
   "issue_tracker": "https://github.com/custom-components/sensor.plex_recently_added/issues",
   "requirements": [],
-  "version": "0.4.8"
+  "version": "0.5.4"
 }

custom_components/plex_recently_added/parser.py

@@ -9,8 +9,26 @@
 from datetime import datetime, timedelta
 from homeassistant.core import HomeAssistant
 from homeassistant.components.http.auth import async_sign_path
-
-from .const import TIMEOUT_MINUTES
+from time import time
+
+# Stable signed-path cache: prevents new URLs on each 10-min refresh
+# Re-signs only when near expiry, so browsers don't re-download unchanged images.
+_SIGNED_URL_CACHE = {}           # key: raw_path (without authSig), value: (signed_url, expires_epoch)
+_STALE_MARGIN_SEC = 24 * 3600    # renew 1 day before expiry to be safe
+
+def _stable_signed_path(hass: HomeAssistant, raw_path: str, ttl_minutes: int) -> str:
+    now = time()
+    entry = _SIGNED_URL_CACHE.get(raw_path)
+    if entry:
+        signed_url, exp = entry
+        if exp - now > _STALE_MARGIN_SEC:
+            return signed_url
+    # Sign a fresh URL and remember its approximate expiry
+    signed = async_sign_path(hass, raw_path, timedelta(minutes=ttl_minutes))
+    _SIGNED_URL_CACHE[raw_path] = (signed, now + ttl_minutes * 60)
+    return signed
+
+from .const import SIGN_URL_TTL_MINUTES
 
 import logging
 _LOGGER = logging.getLogger(__name__)
@@ -94,11 +112,19 @@ def parse_data(hass: HomeAssistant, data, max, base_url, token, identifier, sect
 
 
         thumb_IDs = extract_metadata_and_type(thumb)
-        data_output["poster"] = async_sign_path(hass, f'{images_base_url}?metadata={thumb_IDs[0]}&thumb={thumb_IDs[2]}', timedelta(minutes=TIMEOUT_MINUTES)) if thumb_IDs else ""
+        data_output["poster"] = _stable_signed_path(
+            hass,
+            f'{images_base_url}?metadata={thumb_IDs[0]}&thumb={thumb_IDs[2]}&v={int(item.get("updatedAt", item.get("addedAt", 0)))}',
+            SIGN_URL_TTL_MINUTES
+        ) if thumb_IDs else ""
 
 
         art_IDs = extract_metadata_and_type(art)
-        data_output["fanart"] = async_sign_path(hass, f'{images_base_url}?metadata={art_IDs[0]}&art={art_IDs[2]}', timedelta(minutes=TIMEOUT_MINUTES)) if art_IDs else ""
+        data_output["fanart"] = _stable_signed_path(
+            hass,
+            f'{images_base_url}?metadata={art_IDs[0]}&art={art_IDs[2]}&v={int(item.get("updatedAt", item.get("addedAt", 0)))}',
+            SIGN_URL_TTL_MINUTES
+        ) if art_IDs else ""
 
 
         data_output["deep_link"] = deep_link if identifier else None

custom_components/plex_recently_added/redirect.py

@@ -1,6 +1,7 @@
 from homeassistant.components.http import HomeAssistantView
 from homeassistant.config_entries import ConfigEntry
-from aiohttp import web, ClientSession
+from aiohttp import web
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
 import requests
 import os
 
@@ -15,12 +16,13 @@
 from .const import DOMAIN
 
 class ImagesRedirect(HomeAssistantView):
-    def __init__(self, config_entry: ConfigEntry):
+    def __init__(self, hass, config_entry: ConfigEntry):
         super().__init__()
         self._token = config_entry.data[CONF_API_KEY]
         self._base_url = f'http{'s' if config_entry.data[CONF_SSL] else ''}://{config_entry.data[CONF_HOST]}:{config_entry.data[CONF_PORT]}'
         self.name = f'{self._token}_Plex_Recently_Added'
         self.url = f'/{config_entry.data[CONF_NAME].lower() + "_" if len(config_entry.data[CONF_NAME]) > 0 else ""}plex_recently_added'
+        self._session = async_get_clientsession(hass)
 
     async def get(self, request):
         metadataId = int(request.query.get("metadata", 0))
@@ -34,12 +36,33 @@ async def get(self, request):
 
         url = f'{self._base_url}/library/metadata/{metadataId}/{image_type}/{image_id}?X-Plex-Token={self._token}'
 
-        async with ClientSession() as session:
-            async with session.get(url) as res:
-                if res.ok:
-                    content = await res.read()
-                    return web.Response(body=content, content_type=res.content_type)
+        fwd_headers = {}
+        if_modified = request.headers.get("If-Modified-Since")
+        if_none = request.headers.get("If-None-Match")
+        if if_modified:
+            fwd_headers["If-Modified-Since"] = if_modified
+        if if_none:
+            fwd_headers["If-None-Match"] = if_none
 
-                return web.HTTPNotFound()
+        async with self._session.get(url, headers=fwd_headers, timeout=10) as res:
+            if res.status == 304:
+                return web.Response(status=304)
+
+            if res.status == 200:
+                body = await res.read()
+                headers = {
+                    "Content-Type": res.headers.get("Content-Type", "image/jpeg"),
+                    # Strong client caching: immutable for a year cuts repeat requests
+                    "Cache-Control": "public, max-age=31536000, immutable",
+                }
+                etag = res.headers.get("ETag")
+                last_mod = res.headers.get("Last-Modified")
+                if etag:
+                    headers["ETag"] = etag
+                if last_mod:
+                    headers["Last-Modified"] = last_mod
+                return web.Response(body=body, headers=headers)
+
+            return web.HTTPNotFound()