Fixing expired IDs bug

global_func/contest_lobby_cache.py

@@ -1,11 +1,50 @@
 """Persist pricing-step contest lists in Mongo (Contest_Information DB)."""
-from datetime import datetime
+from datetime import date, datetime
 from typing import List, Optional
 
 import pandas as pd
+import pytz
 
 COLLECTION_NAME = "DFS_PM_pricing_contest_cache"
 _INDEX_ENSURED = False
+_EASTERN = pytz.timezone("US/Eastern")
+
+
+def _eastern_today() -> date:
+    return datetime.now(_EASTERN).date()
+
+
+def _parse_contest_slate_date(contest: dict) -> Optional[date]:
+    """Slate date from cached contest dict (DRAFTKINGS YYYYMMDD or YYYY-MM-DD display), or None if unknown."""
+    cd = contest.get("contest_date")
+    if isinstance(cd, str) and len(cd) == 8 and cd.isdigit():
+        try:
+            return datetime.strptime(cd, "%Y%m%d").date()
+        except ValueError:
+            pass
+    disp = contest.get("contest_date_display")
+    if isinstance(disp, str):
+        s = disp.strip()
+        if len(s) >= 10:
+            try:
+                return datetime.strptime(s[:10], "%Y-%m-%d").date()
+            except ValueError:
+                pass
+    return None
+
+
+def filter_contests_from_today_forward(contests: List[dict]) -> List[dict]:
+    """
+    Keep contests whose slate date is today (US/Eastern) or later.
+    Entries with no parseable date are kept (e.g. FanDuel CSV uploads without a date column).
+    """
+    today = _eastern_today()
+    out: List[dict] = []
+    for c in contests:
+        d = _parse_contest_slate_date(c)
+        if d is None or d >= today:
+            out.append(c)
+    return out
 
 
 def _collection(db):
@@ -28,32 +67,40 @@ def ensure_contest_cache_indexes(db) -> None:
 
 
 def cache_has_contests(db, site: str, sport: str, game_type: str) -> bool:
-    ensure_contest_cache_indexes(db)
-    return (
-        _collection(db).find_one({"site": site, "sport": sport, "type": game_type}, {"_id": 1})
-        is not None
-    )
+    """True if cache has at least one non-expired contest (slate date >= today US/Eastern)."""
+    cached = get_cached_contests(db, site, sport, game_type)
+    return cached is not None and len(cached) > 0
 
 
 def get_cached_contests(db, site: str, sport: str, game_type: str) -> Optional[List[dict]]:
-    """Return cached contest list, or None if no document exists."""
+    """Return cached contest list (today-forward only), or None if no document exists."""
     ensure_contest_cache_indexes(db)
-    doc = _collection(db).find_one({"site": site, "sport": sport, "type": game_type})
+    key = {"site": site, "sport": sport, "type": game_type}
+    doc = _collection(db).find_one(key)
     if doc is None:
         return None
     contests = doc.get("contests")
     if contests is None:
         return []
-    return list(contests)
+    raw = list(contests)
+    filtered = filter_contests_from_today_forward(raw)
+    if len(filtered) != len(raw):
+        _collection(db).update_one(
+            key,
+            {"$set": {"contests": filtered, "updated_at": datetime.utcnow()}},
+        )
+    return filtered
 
 
 def save_contests_cache(db, site: str, sport: str, game_type: str, contests: List[dict]) -> None:
+    """Replace cached contests with the payload after removing prior-day slates (US/Eastern)."""
     ensure_contest_cache_indexes(db)
+    to_store = filter_contests_from_today_forward(list(contests))
     _collection(db).update_one(
         {"site": site, "sport": sport, "type": game_type},
         {
             "$set": {
-                "contests": contests,
+                "contests": to_store,
                 "updated_at": datetime.utcnow(),
             }
         },