e

M3U8/scrape/utils/__init__.py

@@ -4,7 +4,7 @@ from .config import (
     capture_req,
     get_base,
     get_logger,
-    load_ts_cache,
+    load_cache,
     now,
     safe_process_event,
 )
@@ -15,7 +15,7 @@ __all__ = [
     "capture_req",
     "get_base",
     "get_logger",
-    "load_ts_cache",
+    "load_cache",
     "now",
     "safe_process_event",
 ]

M3U8/scrape/utils/config.py

@@ -68,20 +68,23 @@ def get_logger(name: str | None = None) -> logging.Logger:
     return logger
 
 
-def load_ts_cache(
+def load_cache(
     file: Path,
-    cache_exp: int | float,
+    exp: int | float = None,
+    hour: int = None,
 ) -> dict[str, dict[str, str | float]]:
     try:
-        data: dict[str, dict[str, str | float]] = json.loads(
-            file.read_text(encoding="utf-8")
-        )
+        data = json.loads(file.read_text(encoding="utf-8"))
 
-        return {
-            k: v
-            for k, v in data.items()
-            if now.timestamp() - v.get("timestamp", 0) < cache_exp
-        }
+        if exp:
+            return {
+                k: v
+                for k, v in data.items()
+                if now.timestamp() - v.get("timestamp", 0) < exp
+            }
+
+        elif hour:
+            return {} if now.hour <= hour else data
     except (FileNotFoundError, json.JSONDecodeError):
         return {}