feat(main): main

core/services/cache.py

@@ -40,44 +40,66 @@ def run_cache(hours: int = 24, prune_only: bool = False) -> dict:
     )
 
     youtube_items: dict[int, MediaItem] = {}
-    for airing in upcoming:
-        item = airing.media_item
-        if item.media_source and item.media_source.source_type in YOUTUBE_SOURCE_TYPES:
-            youtube_items[item.pk] = item
-
     downloaded = already_cached = failed = 0
     items_status = []
+    
+    for airing in upcoming:
+        item = airing.media_item
+        
+        # Determine if we are inside the 1-hour critical safety window
+        time_until_airing = airing.starts_at - now
+        in_safety_window = time_until_airing.total_seconds() < 3600
+        
+        if item.media_source and item.media_source.source_type in YOUTUBE_SOURCE_TYPES:
+            youtube_items[item.pk] = item
+            
+            # Skip if already cached
+            if item.cached_file_path and pathlib.Path(item.cached_file_path).exists():
+                already_cached += 1
+                items_status.append({
+                    "id": item.pk,
+                    "title": item.title,
+                    "status": "cached",
+                    "path": item.cached_file_path,
+                })
+                continue
+                
+            # If in the 1-hour safety valve window, DO NOT download. Replace the airing.
+            if in_safety_window:
+                logger.warning(f"Airing {airing.id} ({item.title}) is < 1h away and not cached! Triggering emergency replacement.")
+                from core.services.scheduler import ScheduleGenerator
+                generator = ScheduleGenerator(channel=airing.channel)
+                try:
+                    generator.replace_undownloaded_airings([airing])
+                    items_status.append({
+                        "id": item.pk,
+                        "title": item.title,
+                        "status": "replaced",
+                        "error": "Not downloaded in time",
+                    })
+                except Exception as e:
+                    logger.error(f"Emergency replacement failed for airing {airing.id}: {e}")
+                continue
 
-    for item in youtube_items.values():
-        # Skip if already cached
-        if item.cached_file_path and pathlib.Path(item.cached_file_path).exists():
-            already_cached += 1
-            items_status.append({
-                "id": item.pk,
-                "title": item.title,
-                "status": "cached",
-                "path": item.cached_file_path,
-            })
-            continue
-
-        try:
-            local_path = download_for_airing(item)
-            downloaded += 1
-            items_status.append({
-                "id": item.pk,
-                "title": item.title,
-                "status": "downloaded",
-                "path": str(local_path),
-            })
-        except Exception as exc:
-            failed += 1
-            items_status.append({
-                "id": item.pk,
-                "title": item.title,
-                "status": "failed",
-                "error": str(exc),
-            })
-            logger.error("download_for_airing(%s) failed: %s", item.pk, exc)
+            # Otherwise, attempt download normally
+            try:
+                local_path = download_for_airing(item)
+                downloaded += 1
+                items_status.append({
+                    "id": item.pk,
+                    "title": item.title,
+                    "status": "downloaded",
+                    "path": str(local_path),
+                })
+            except Exception as exc:
+                failed += 1
+                items_status.append({
+                    "id": item.pk,
+                    "title": item.title,
+                    "status": "failed",
+                    "error": str(exc),
+                })
+                logger.error("download_for_airing(%s) failed: %s", item.pk, exc)
 
     logger.info(
         "run_cache(hours=%d): pruned=%d downloaded=%d cached=%d failed=%d",