sibi-dst 2025.9.9__py3-none-any.whl → 2025.9.10__py3-none-any.whl

sibi_dst/utils/update_planner.py

@@ -1,54 +1,52 @@
-# update_planner.py
 from __future__ import annotations
 
 import datetime as dt
+import re
 from concurrent.futures import ThreadPoolExecutor, wait
 from typing import List, Optional, Dict, Union, Tuple, Set, Iterator, ClassVar, Any, Callable
 
 import pandas as pd
 
 from sibi_dst.utils import ManagedResource
-from . import FileAgeChecker  # Assuming FileAgeChecker is in the same package
+from . import FileAgeChecker
 
 
 class UpdatePlanner(ManagedResource):
     """
-    Scans date-partitioned storage and builds an 'update plan' for dates that need processing.
-    Backward compatible: public API and legacy attributes preserved; enhancements are opt-in via kwargs.
+    Update planner for datasets organized either as:
 
-    Enhancements:
-      - Batch listings via fsspec.find(..., detail=True) to avoid N×exists() roundtrips.
-      - Age computed from the NEWEST data file (ignoring control files).
-      - Optional completeness check: partitions with files but no _SUCCESS => 'incomplete'.
-      - Real timeouts using concurrent.futures.wait(...).
-      - Future dates marked as 'future' (not actionable).
+    - Legacy layout: /YYYY/MM/DD/file.parquet
+    - Hive layout:   /partition_date=YYYY-MM-DD/[other=val]/file.parquet
+
+    Public API is unchanged (`generate_plan`, `show_update_plan`, etc.).
     """
 
-    # -------- Defaults (extended, but original keys retained) --------
     DEFAULT_PRIORITY_MAP: ClassVar[Dict[str, int]] = {
         "file_is_recent": 0,
         "missing_ignored": 0,
         "overwrite_forced": 1,
-        "incomplete": 1,  # new: prioritize just under overwrite
+        "incomplete": 1,
         "create_missing": 2,
         "missing_in_history": 3,
         "stale_in_history": 4,
-        "future": 99,  # new: not actionable
+        "future": 99,
     }
 
     DEFAULT_MAX_AGE_MINUTES: int = 1440
     DEFAULT_HISTORY_DAYS_THRESHOLD: int = 30
 
-    # Data/Control file heuristics (can be overridden)
     DATA_FILE_PATTERNS: ClassVar[Tuple[str, ...]] = (".parquet", ".orc", ".csv", ".json")
     CONTROL_BASENAMES: ClassVar[Set[str]] = {"_SUCCESS", "_metadata", "_common_metadata"}
 
+    HIVE_PARTITION_RE: ClassVar[re.Pattern] = re.compile(r"([^/=]+)=([^/]+)")
+
     logger_extra = {"sibi_dst_component": __name__}
 
     def __init__(
         self,
         parquet_storage_path: str,
-        parquet_filename: str,
+        *,
+        partition_on: Optional[List[str]] = None,
         description: str = "Update Planner",
         reference_date: Union[str, dt.date, None] = None,
         history_days_threshold: int = DEFAULT_HISTORY_DAYS_THRESHOLD,
@@ -58,57 +56,61 @@ class UpdatePlanner(ManagedResource):
         custom_priority_map: Optional[Dict[str, int]] = None,
         reverse_order: bool = False,
         show_progress: bool = False,
+        hive_style: bool = False,
         skipped: Optional[List[Union[str, dt.date]]] = None,
         **kwargs,
     ):
         super().__init__(**kwargs)
 
-        # ---- Core Configuration ----
-        self.description: str = description
+        # ---- core config ----
         self.data_path: str = self._ensure_trailing_slash(parquet_storage_path)
-        self.filename: str = parquet_filename
+        self.description: str = description
         self.reverse_order: bool = reverse_order
         self.show_progress: bool = show_progress
         self.overwrite: bool = overwrite
         self.ignore_missing: bool = ignore_missing
         self.history_days_threshold: int = history_days_threshold
         self.max_age_minutes: int = max_age_minutes
-        # Copy to avoid shared mutation
         self.priority_map: Dict[str, int] = dict(custom_priority_map) if custom_priority_map else dict(self.DEFAULT_PRIORITY_MAP)
 
-        # ---- Execution Parameters ----
+        # ---- NEW: Hive partition support ----
+        self.hive_style: bool = hive_style
+        self.partition_on: List[str] = list(partition_on or ["partition_date"] if self.hive_style else ["year", "month", "day"])
+
+        # ---- execution knobs ----
         self.max_threads: int = int(kwargs.get("max_threads", 3))
-        self.timeout: float = float(kwargs.get("timeout", 30.0))  # legacy overall timeout
+        self.timeout: float = float(kwargs.get("timeout", 30.0))
+        self.list_timeout: float = float(kwargs.get("list_timeout", self.timeout))
+        self.total_timeout: float = float(kwargs.get("total_timeout", self.timeout))
 
-        # ---- Date Window ----
+        # ---- date window ----
         self.start_date = kwargs.get("parquet_start_date")
         self.end_date = kwargs.get("parquet_end_date")
 
-        # ---- Reference Date ----
+        # ---- reference date ----
         if reference_date is not None:
             self.reference_date: dt.date = pd.to_datetime(reference_date).date()
         else:
             self.reference_date: dt.date = dt.date.today()
 
-        # ---- Feature Flags / Advanced Knobs ----
-        self.check_completeness: bool = bool(kwargs.get("check_completeness", True))
-        self.require_success_marker: bool = bool(kwargs.get("require_success_marker", True))
+        # ---- completeness/heuristics ----
+        self.check_completeness: bool = bool(kwargs.get("check_completeness", False))
+        self.require_success_marker: bool = bool(kwargs.get("require_success_marker", False))
         self.list_granularity: str = str(kwargs.get("list_granularity", "month"))
         self.data_file_suffixes: Tuple[str, ...] = tuple(kwargs.get("data_file_suffixes", self.DATA_FILE_PATTERNS))
-        self.list_timeout: float = float(kwargs.get("list_timeout", self.timeout))
-        self.total_timeout: float = float(kwargs.get("total_timeout", self.timeout))
-        # Dependency-injected clock (UTC) for tests
+
+        # ---- clock for tests ----
         self._utcnow: Callable[[], dt.datetime] = kwargs.get("utcnow_func", None) or (lambda: dt.datetime.utcnow())
 
-        # ---- Backward-Compatible Skip Handling ----
-        # Keep legacy attribute and derive new internal canonical sets.
+        # ---- skipped (back-compat) ----
         self.skipped = list(skipped or kwargs.get("skipped", []) or [])
         self.skipped_paths: Set[str] = {p.rstrip("/") + "/" for p in self.skipped if isinstance(p, str)}
         self.skipped_dates: Set[dt.date] = {p for p in self.skipped if isinstance(p, dt.date)}
 
-        # ---- Helpers & State ----
         if not getattr(self, "fs", None):
             raise ValueError("UpdatePlanner requires a valid fsspec filesystem (fs).")
+
+        # ---- state ----
         self.age_checker = FileAgeChecker(debug=self.debug, logger=self.logger)
         self.plan: pd.DataFrame = pd.DataFrame()
         self.df_req: pd.DataFrame = pd.DataFrame()
@@ -117,41 +119,20 @@ class UpdatePlanner(ManagedResource):
     # --------------------- Back-compat property bridge ---------------------
     @property
     def skipped(self) -> List[Union[str, dt.date]]:
-        """
-        Backward-compatible view of skip configuration.
-        Returns a merged list of path-strings and dates.
-        """
-        paths = sorted(self.skipped_paths)
-        dates = sorted(self.skipped_dates)
-        return [*paths, *dates]
+        return [*sorted(self.skipped_paths), *sorted(self.skipped_dates)]
 
     @skipped.setter
     def skipped(self, value: List[Union[str, dt.date]]) -> None:
-        """
-        Accepts legacy assignments like:
-            planner.skipped = ["s3://.../2025/01/03/", date(2025,1,4)]
-        and keeps new internals in sync.
-        """
-        value = list(value or [])
         self.skipped_paths = {p.rstrip("/") + "/" for p in value if isinstance(p, str)}
         self.skipped_dates = {p for p in value if isinstance(p, dt.date)}
 
     # --------------------- Public API ---------------------
-    def has_plan(self) -> bool:
-        """Check if a plan DataFrame exists and is not empty."""
-        return isinstance(self.plan, pd.DataFrame) and not self.plan.empty
-
-    def required_count(self) -> int:
-        """Get the number of dates that require an update."""
-        return len(self.df_req) if isinstance(self.df_req, pd.DataFrame) else 0
-
     def generate_plan(
         self,
         start: Union[str, dt.date, None] = None,
         end: Union[str, dt.date, None] = None,
         freq: str = "D",
     ) -> pd.DataFrame:
-        """Build a plan for [start, end]. Returns rows that require update (df_req)."""
         start = start or self.start_date
         end = end or self.end_date
         if start is None or end is None:
@@ -162,330 +143,196 @@ class UpdatePlanner(ManagedResource):
         if sd > ed:
             raise ValueError(f"Start date ({sd}) must be on or before end date ({ed}).")
 
-        log_extra = self._log_extra()
-        self.logger.info(f"Generating update plan for {self.description} from {sd} to {ed}", extra=log_extra)
+        self.logger.info(f"Generating update plan for {self.description} from {sd} to {ed}", extra=self._log_extra())
         self._generate_plan(sd, ed, freq=freq)
-        self.logger.info(
-            f"Plan built for {self.description}: {len(self.plan)} dates evaluated, "
-            f"{len(self.df_req)} require update",
-            extra=log_extra,
-        )
         return self.df_req
 
     def show_update_plan(self) -> None:
-        """Pretty-print the current plan once per run."""
-        if not self.has_plan():
-            self.logger.info("No update plan to show.", extra=self._log_extra())
+        if not self.has_plan() or self._printed_this_run:
             return
-        if self._printed_this_run:
-            return
-
         try:
             from rich.console import Console
             from rich.table import Table
-
             console = Console()
-            terminal_width = console.size.width
-
             table = Table(
-                title=f"Update Plan for {self.data_path}",
-                show_header=True,
-                header_style="bold magenta",
-                expand=True,
-                pad_edge=False,
+                title=f"Update Plan for {self.data_path} [{'Hive' if 'partition_date' in self.partition_on else 'Legacy'}]",
+                show_header=True, header_style="bold magenta", expand=True, pad_edge=False,
             )
-            max_w = max(terminal_width - 50, 640)
             for col in self.plan.columns:
-                if col in {"date", "update_category", "update_priority", "update_required", "file_exists"}:
-                    table.add_column(col, justify="left", no_wrap=True, overflow="fold", max_width=max_w)
-                elif col == "description":
-                    table.add_column(col, justify="left", overflow="fold", max_width=max_w)
-                else:
-                    table.add_column(col, justify="left", overflow="fold")
-
+                table.add_column(col, justify="left", overflow="fold")
             for _, row in self.plan.iterrows():
                 table.add_row(*(str(row[c]) for c in self.plan.columns))
-
-            with console.capture() as cap:
-                console.print(table)
-            self.logger.info(f"Full Update Plan:\n{cap.get().strip()}", extra=self._log_extra())
-
-        except Exception as e:
-            self.logger.debug(f"Falling back to plain text plan display due to: {e}", extra=self._log_extra())
-            preview = self.plan.head(200).to_string(index=False)
-            self.logger.info(f"Update Plan (first 200 rows):\n{preview}", extra=self._log_extra())
-
+            console.print(table)
+        except Exception:
+            self.logger.info(f"Update Plan:\n{self.plan.head(50)}", extra=self._log_extra())
         self._printed_this_run = True
 
     def get_tasks_by_priority(self) -> Iterator[Tuple[int, List[dt.date]]]:
-        """Yield (priority, [dates...]) batches, smallest priority first."""
         if not self.has_plan():
             return
         req = self.plan[self.plan["update_required"]]
-        if req.empty:
-            return
         for priority in sorted(req["update_priority"].unique()):
-            dates_df = req[req["update_priority"] == priority]
-            dates_df = dates_df.sort_values(by="date", ascending=not self.reverse_order)
-            dates = dates_df["date"].tolist()
+            dates = req[req["update_priority"] == priority].sort_values(
+                by="date", ascending=not self.reverse_order
+            )["date"].tolist()
             if dates:
                 yield int(priority), dates
 
-    # --------------------- Plan Generation Internals ---------------------
+    def has_plan(self) -> bool:
+        return not self.plan.empty
+
+    def required_count(self) -> int:
+        return len(self.df_req)
+
+    # --------------------- Internals ---------------------
     def _generate_plan(self, start: dt.date, end: dt.date, freq: str = "D") -> None:
-        """
-        Populate self.plan with all dates and self.df_req with the subset to update.
-        - Pre-lists months or days (configurable) with timeouts that actually apply
-        - Computes staleness from newest *data* file
-        - Flags partitions without _SUCCESS as 'incomplete' (unless disabled)
-        - Marks future dates as 'future' (not actionable)
-        """
         dates: List[dt.date] = pd.date_range(start=start, end=end, freq=freq).date.tolist()
         history_start = self.reference_date - dt.timedelta(days=self.history_days_threshold)
         rows: List[Dict[str, Any]] = []
 
-        def is_future(d: dt.date) -> bool:
-            return d > self.reference_date
-
-        # Choose listing units
-        units: List[Tuple[str, dt.date]] = []
-        if self.list_granularity == "day":
-            units = [("day", d) for d in dates]
-        else: # Default to month
+        if "partition_date" in self.partition_on:
+            caches: Dict[dt.date, Dict[str, Any]] = self._list_prefix(self.data_path)
+        else:
+            caches: Dict[dt.date, Dict[str, Any]] = {}
             months = list(self._iter_month_starts(self._month_floor(start), self._month_floor(end)))
-            units = [("month", m) for m in months]
-
-        self.logger.info(
-            f"Pre-listing {len(units)} {'days' if self.list_granularity=='day' else 'month prefixes'} for {self.description}",
-            extra=self._log_extra(),
-        )
-
-        # --- Parallel File Listing with Realistic Timeouts ---
-        caches: Dict[dt.date, Dict[dt.date, Dict[str, Any]]] = {}
-        max_workers = max(1, self.max_threads) # Ensure at least 1 worker
-
-        with ThreadPoolExecutor(max_workers=max_workers, thread_name_prefix="update_planner") as executor:
-            future_to_unit: Dict[Any, Tuple[str, dt.date]] = {}
-            for kind, val in units:
-                prefix = self._day_prefix(val) if kind == "day" else self._month_prefix(val)
-                future = executor.submit(self._list_prefix, prefix)
-                future_to_unit[future] = (kind, val)
-
-            # Wait for all futures with a total timeout
-            done_futures, not_done_futures = wait(future_to_unit.keys(), timeout=self.total_timeout or None)
-
-            # Process completed futures
-            for future in done_futures:
-                kind, val = future_to_unit[future]
-                try:
-                    # Get the result with a per-listing timeout
-                    cache = future.result(timeout=self.list_timeout or None)
-                except Exception as e:
-                    self.logger.warning(f"Listing failed for {kind}:{val} — {e}", extra=self._log_extra())
-                    cache = {}
-
-                if kind == "month":
-                    caches[val] = cache
-                else: # day
-                    # Store day listing results in its month's bucket for summarization
-                    month_key = val.replace(day=1)
-                    caches.setdefault(month_key, {}).update(cache)
-
-            # Handle timed-out futures
-            for future in not_done_futures:
-                kind, val = future_to_unit[future]
-                self.logger.error(f"Listing timed out for {kind}:{val}", extra=self._log_extra())
-                if kind == "month":
-                    caches[val] = {}
-                else: # day
-                    month_key = val.replace(day=1)
-                    caches.setdefault(month_key, {})
+            with ThreadPoolExecutor(max_workers=max(1, self.max_threads)) as ex:
+                future_to_unit = {ex.submit(self._list_prefix, self._month_prefix(m)): m for m in months}
+                done, _ = wait(future_to_unit.keys(), timeout=self.total_timeout or None)
+                for fut in done:
+                    m = future_to_unit[fut]
+                    try:
+                        caches[m] = fut.result(timeout=self.list_timeout or None)
+                    except Exception:
+                        caches[m] = {}
 
-        # --- Summarize Each Date and Build Plan ---
         for d in dates:
-            if is_future(d):
-                rows.append({
-                    "date": d, "file_exists": False, "file_age_minutes": None,
-                    "update_category": "future", "update_priority": self.priority_map.get("future", 99),
-                    "update_required": False, "description": self.description,
-                })
+            if d > self.reference_date:
+                rows.append(self._row_future(d))
                 continue
-
             if self._is_skipped(d):
-                self.logger.debug(f"Skipping {d}: in skipped set.", extra=self._log_extra())
-                # Append a row even for skipped dates, using default policy logic
                 rows.append(self._make_row(d, history_start, False, None))
                 continue
 
-            # Get the cache for the month containing this date
-            month_key = d.replace(day=1)
-            cache = caches.get(month_key, {})
+            cache = caches if "partition_date" in self.partition_on else caches.get(d.replace(day=1), {})
             exists, age_min, incomplete = self._summarize_partition(d, cache)
-
-            # Incomplete partitions get their own category (unless overwrite forces update)
             if incomplete and not self.overwrite:
-                rows.append({
-                    "date": d, "file_exists": True, "file_age_minutes": age_min,
-                    "update_category": "incomplete", "update_priority": self.priority_map.get("incomplete", 1),
-                    "update_required": True, "description": self.description,
-                })
-                continue
-
-            # Fall back to the standard policy logic (overwrite / history / staleness / missing)
-            rows.append(self._make_row(d, history_start, exists, age_min))
+                rows.append(self._row_incomplete(d, age_min))
+            else:
+                rows.append(self._make_row(d, history_start, exists, age_min))
 
-        # --- Finalize DataFrame ---
         df = pd.DataFrame.from_records(rows)
         if not df.empty:
             df["date"] = pd.to_datetime(df["date"]).dt.date
             df["update_priority"] = df["update_priority"].astype(int)
-
-        df = df.sort_values(
+        self.plan = df.sort_values(
             by=["update_priority", "date"],
             ascending=[True, not self.reverse_order],
-            kind="mergesort", # Stable sort
+            kind="mergesort",
         ).reset_index(drop=True)
+        self.df_req = self.plan[self.plan["update_required"]].copy()
 
-        self.plan = df
-        self.df_req = df[df["update_required"]].copy()
-        self._printed_this_run = False
-
-    # --------------------- File System Interaction ---------------------
     def _list_prefix(self, prefix: str) -> Dict[dt.date, Dict[str, Any]]:
-        """
-        Return {date: {'files': [paths], 'has_success': bool, 'newest_ts': datetime|None}} under prefix.
-        Uses fsspec.find(detail=True) for one-shot listing with metadata (mtime).
-        """
         try:
-            # Returns {path: info_dict} when detail=True
             items: Dict[str, Any] = self.fs.find(prefix, withdirs=False, detail=True)
-        except Exception as e:
-            self.logger.warning(f"Listing failed for {prefix}: {e}", extra=self._log_extra())
+        except Exception:
             return {}
 
         out: Dict[dt.date, Dict[str, Any]] = {}
         for path, info in items.items():
-            # Extract date from path structure (e.g., .../YYYY/MM/DD/file)
-            parts = path.strip("/").split("/")
-            if len(parts) < 3: # Need at least year, month, day
-                continue
-            try:
-                y, m, dd = int(parts[-3]), int(parts[-2]), int(parts[-1])
-                d = dt.date(y, m, dd)
-            except (ValueError, IndexError):
-                # Not a date-partitioned path, skip
+            d: Optional[dt.date] = None
+            if "partition_date" in self.partition_on:
+                parts = self._extract_partitions(path)
+                if "partition_date" in parts:
+                    try:
+                        d = dt.date.fromisoformat(parts["partition_date"])
+                    except Exception:
+                        continue
+            else:
+                segs = path.strip("/").split("/")
+                if len(segs) >= 3:
+                    try:
+                        y, m, dd = int(segs[-3]), int(segs[-2]), int(segs[-1])
+                        d = dt.date(y, m, dd)
+                    except Exception:
+                        continue
+            if d is None:
                 continue
 
-            # Initialize or get the record for this date
             rec = out.setdefault(d, {"files": [], "has_success": False, "newest_ts": None})
-            base_name = path.rsplit("/", 1)[-1]
-
-            # Check for _SUCCESS marker
-            if base_name == "_SUCCESS":
+            base = path.rsplit("/", 1)[-1]
+            if base == "_SUCCESS":
                 rec["has_success"] = True
-
-            # Check if it's a relevant data file
             if self._is_data_file(path):
                 rec["files"].append(path)
-                # Determine the modification time
-                mtime = info.get("mtime") or info.get("LastModified") or info.get("last_modified")
-                ts = None
-                if isinstance(mtime, (int, float)):
-                    ts = dt.datetime.utcfromtimestamp(mtime)
-                elif isinstance(mtime, str):
-                    try:
-                        ts = pd.to_datetime(mtime, utc=True).to_pydatetime()
-                    except Exception:
-                        ts = None
-                elif isinstance(mtime, dt.datetime):
-                    # Ensure timezone awareness for comparison
-                    ts = mtime if mtime.tzinfo else mtime.replace(tzinfo=dt.timezone.utc)
-
-                # Update the newest timestamp for this partition
-                if ts:
-                    current_newest = rec["newest_ts"]
-                    # Naive comparison after ensuring tz awareness
-                    ts_naive = ts.replace(tzinfo=None) if ts.tzinfo else ts
-                    current_naive = current_newest.replace(tzinfo=None) if current_newest and current_newest.tzinfo else current_newest
-                    if current_naive is None or ts_naive > current_naive:
-                        rec["newest_ts"] = ts
-
+                ts = self._extract_mtime(info)
+                if ts and (rec["newest_ts"] is None or ts > rec["newest_ts"]):
+                    rec["newest_ts"] = ts
         return out
 
-    def _summarize_partition(
-        self, d: dt.date, cache: Dict[dt.date, Dict[str, Any]]
-    ) -> Tuple[bool, Optional[float], bool]:
-        """
-        Summarize the state of a partition for a given date.
+    def _extract_partitions(self, path: str) -> Dict[str, str]:
+        out: Dict[str, str] = {}
+        for seg in path.strip("/").split("/"):
+            m = self.HIVE_PARTITION_RE.match(seg)
+            if m:
+                out[m.group(1)] = m.group(2)
+        return out
 
-        Returns:
-            Tuple[bool, Optional[float], bool]: (exists, age_minutes, incomplete)
-            - exists: True iff at least one *data* file is present for day `d`
-            - age_minutes: minutes since the NEWEST data file (UTC 'now'), or None if not determinable
-            - incomplete: True if files exist but required _SUCCESS is missing (and checks are enabled)
-        """
+    def _summarize_partition(self, d: dt.date, cache: Dict[dt.date, Dict[str, Any]]) -> Tuple[bool, Optional[float], bool]:
         rec = cache.get(d, {})
         files = rec.get("files", [])
-        has_success = bool(rec.get("has_success", False))
-        exists = len(files) > 0
-
+        exists = bool(files)
         if not exists:
             return False, None, False
-
+        has_success = rec.get("has_success", False)
         newest_ts = rec.get("newest_ts")
-        age_min: Optional[float] = None
+        age_min = None
         if newest_ts:
-            now_utc = self._utcnow().replace(tzinfo=None) # Get current UTC time (naive)
-            ts_naive = newest_ts.replace(tzinfo=None) if newest_ts.tzinfo else newest_ts # Make mtime naive
-            age_min = max(0.0, (now_utc - ts_naive).total_seconds() / 60.0)
-
+            now = self._utcnow().replace(tzinfo=None)
+            ts = newest_ts.replace(tzinfo=None) if newest_ts.tzinfo else newest_ts
+            age_min = max(0.0, (now - ts).total_seconds() / 60.0)
         incomplete = self.check_completeness and self.require_success_marker and not has_success
         return exists, age_min, incomplete
 
-    # --------------------- Policy Logic ---------------------
-    def _make_row(
-        self,
-        date: dt.date,
-        history_start: dt.date,
-        file_exists: bool,
-        file_age: Optional[float],
-    ) -> Dict[str, Any]:
-        """
-        Build a single plan row based on flags and thresholds.
-        (Categories 'future'/'incomplete' are injected earlier by _generate_plan.)
-        """
-        within_history = history_start <= date <= self.reference_date
-        update_required = False
-        category = "unknown"
-
+    def _make_row(self, d: dt.date, history_start: dt.date, exists: bool, age_min: Optional[float]) -> Dict[str, Any]:
+        within_history = history_start <= d <= self.reference_date
+        category, update_required = "unknown", False
         if self.overwrite:
-            category = "overwrite_forced"
-            update_required = True
+            category, update_required = "overwrite_forced", True
         elif within_history:
-            if not file_exists:
-                category = "missing_in_history"
-                update_required = True
-            elif file_age is not None and file_age > self.max_age_minutes:
-                category = "stale_in_history"
-                update_required = True
+            if not exists:
+                category, update_required = "missing_in_history", True
+            elif age_min is not None and age_min > self.max_age_minutes:
+                category, update_required = "stale_in_history", True
             else:
                 category = "file_is_recent"
-        elif not file_exists and not self.ignore_missing:
-            category = "create_missing"
-            update_required = True
+        elif not exists and not self.ignore_missing:
+            category, update_required = "create_missing", True
         else:
-            category = "missing_ignored" if not file_exists else "file_is_recent"
-
+            category = "missing_ignored" if not exists else "file_is_recent"
         return {
-            "date": date,
-            "file_exists": file_exists,
-            "file_age_minutes": file_age,
+            "date": d,
+            "file_exists": exists,
+            "file_age_minutes": age_min,
             "update_category": category,
             "update_priority": self.priority_map.get(category, 99),
             "update_required": update_required,
             "description": self.description,
         }
 
+    def _row_future(self, d: dt.date) -> Dict[str, Any]:
+        return {
+            "date": d, "file_exists": False, "file_age_minutes": None,
+            "update_category": "future", "update_priority": self.priority_map.get("future", 99),
+            "update_required": False, "description": self.description,
+        }
+
+    def _row_incomplete(self, d: dt.date, age_min: Optional[float]) -> Dict[str, Any]:
+        return {
+            "date": d, "file_exists": True, "file_age_minutes": age_min,
+            "update_category": "incomplete", "update_priority": self.priority_map.get("incomplete", 1),
+            "update_required": True, "description": self.description,
+        }
+
     # --------------------- Utilities ---------------------
     @staticmethod
     def _ensure_trailing_slash(path: str) -> str:
@@ -501,29 +348,36 @@ class UpdatePlanner(ManagedResource):
         while cur <= end:
             yield cur
             y, m = cur.year, cur.month
-            # Move to the first day of the next month
-            if m == 12:
-                cur = dt.date(y + 1, 1, 1)
-            else:
-                cur = dt.date(y, m + 1, 1)
+            cur = dt.date(y + 1, 1, 1) if m == 12 else dt.date(y, m + 1, 1)
 
     def _month_prefix(self, month_start: dt.date) -> str:
         return f"{self.data_path}{month_start.year}/{month_start.month:02d}/"
 
-    def _day_prefix(self, d: dt.date) -> str:
-        return f"{self.data_path}{d.year}/{d.month:02d}/{d.day:02d}/"
-
     def _is_data_file(self, path: str) -> bool:
         base = path.rsplit("/", 1)[-1]
-        # Skip hidden files, directories, and control files
         if not base or base.startswith(".") or base in self.CONTROL_BASENAMES:
             return False
-        lower_base = base.lower()
-        return any(lower_base.endswith(suf) for suf in self.data_file_suffixes)
+        return any(base.lower().endswith(suf) for suf in self.data_file_suffixes)
+
+    @staticmethod
+    def _extract_mtime(info: Dict[str, Any]) -> Optional[dt.datetime]:
+        mtime = info.get("mtime") or info.get("LastModified") or info.get("last_modified")
+        if isinstance(mtime, (int, float)):
+            return dt.datetime.utcfromtimestamp(mtime)
+        if isinstance(mtime, str):
+            try:
+                return pd.to_datetime(mtime, utc=True).to_pydatetime()
+            except Exception:
+                return None
+        if isinstance(mtime, dt.datetime):
+            return mtime if mtime.tzinfo else mtime.replace(tzinfo=dt.timezone.utc)
+        return None
 
     def _is_skipped(self, d: dt.date) -> bool:
-        """True if the date or its canonical path is in the skip config."""
-        canonical_path = f"{self.data_path}{d.year}/{d.month:02d}/{d.day:02d}/"
+        if "partition_date" in self.partition_on:
+            canonical_path = f"{self.data_path}partition_date={d.isoformat()}/"
+        else:
+            canonical_path = f"{self.data_path}{d.year}/{d.month:02d}/{d.day:02d}/"
         return (d in self.skipped_dates) or (canonical_path in self.skipped_paths)
 
     def _log_extra(self, **overrides) -> Dict[str, Any]:
@@ -534,502 +388,4 @@ class UpdatePlanner(ManagedResource):
             "action_module_name": "update_plan",
         }
         base.update(overrides)
-        return base
-
-
-# import datetime as dt
-# from concurrent.futures import ThreadPoolExecutor, wait
-# from typing import List, Optional, Dict, Union, Tuple, Set, Iterator, ClassVar
-#
-# import pandas as pd
-#
-# from sibi_dst.utils import ManagedResource
-# from . import FileAgeChecker
-#
-#
-# class UpdatePlanner(ManagedResource):
-#     """
-#     Scans date-partitioned storage and builds an 'update plan' for dates that need processing.
-#     Backward compatible: public API and legacy attributes preserved; enhancements are opt-in via kwargs.
-#
-#     Enhancements:
-#       - Batch listings via fsspec.find(..., detail=True) to avoid N×exists() roundtrips.
-#       - Age computed from the NEWEST data file (ignoring control files).
-#       - Optional completeness check: partitions with files but no _SUCCESS => 'incomplete'.
-#       - Real timeouts using concurrent.futures.wait(...).
-#       - Future dates marked as 'future' (not actionable).
-#     """
-#
-#     # -------- Defaults (extended, but original keys retained) --------
-#     DEFAULT_PRIORITY_MAP: ClassVar[Dict[str, int]] = {
-#         "file_is_recent": 0,
-#         "missing_ignored": 0,
-#         "overwrite_forced": 1,
-#         "incomplete": 1,            # new: prioritize just under overwrite
-#         "create_missing": 2,
-#         "missing_in_history": 3,
-#         "stale_in_history": 4,
-#         "future": 99,               # new: not actionable
-#     }
-#
-#     DEFAULT_MAX_AGE_MINUTES: int = 1440
-#     DEFAULT_HISTORY_DAYS_THRESHOLD: int = 30
-#
-#     # Data/Control file heuristics (can be overridden)
-#     DATA_FILE_PATTERNS: ClassVar[Tuple[str, ...]] = (".parquet", ".orc", ".csv", ".json")
-#     CONTROL_BASENAMES: ClassVar[Set[str]] = {"_SUCCESS", "_metadata", "_common_metadata"}
-#
-#     logger_extra = {"sibi_dst_component": __name__}
-#
-#     def __init__(
-#         self,
-#         parquet_storage_path: str,
-#         parquet_filename: str,
-#         description: str = "Update Planner",
-#         reference_date: Union[str, dt.date, None] = None,
-#         history_days_threshold: int = DEFAULT_HISTORY_DAYS_THRESHOLD,
-#         max_age_minutes: int = DEFAULT_MAX_AGE_MINUTES,
-#         overwrite: bool = False,
-#         ignore_missing: bool = False,
-#         custom_priority_map: Optional[Dict[str, int]] = None,
-#         reverse_order: bool = False,
-#         show_progress: bool = False,
-#         skipped: Optional[List[Union[str, dt.date]]] = None,
-#         **kwargs,
-#     ):
-#         super().__init__(**kwargs)
-#
-#         # ---- Existing public-ish attributes (unchanged) ----
-#         self.description = description
-#         self.data_path = self._ensure_trailing_slash(parquet_storage_path)
-#         self.filename = parquet_filename
-#         self.reverse_order = reverse_order
-#         self.show_progress = show_progress
-#         self.overwrite = overwrite
-#         self.ignore_missing = ignore_missing
-#         self.history_days_threshold = history_days_threshold
-#         self.max_age_minutes = max_age_minutes
-#         # copy to avoid shared mutation
-#         self.priority_map = dict(custom_priority_map) if custom_priority_map else dict(self.DEFAULT_PRIORITY_MAP)
-#
-#         # Execution knobs from kwargs (kept)
-#         self.max_threads: int = int(kwargs.get("max_threads", 3))
-#         self.timeout: float = float(kwargs.get("timeout", 30.0))  # legacy overall timeout
-#
-#         # Date window (kept)
-#         self.start_date = kwargs.get("parquet_start_date")
-#         self.end_date = kwargs.get("parquet_end_date")
-#
-#         # Reference date (kept; tolerant)
-#         self.reference_date = pd.to_datetime(reference_date).date() if reference_date is not None else dt.date.today()
-#
-#         # Helpers & state (kept)
-#         self.age_checker = FileAgeChecker(debug=self.debug, logger=self.logger)
-#         self.plan: pd.DataFrame = pd.DataFrame()
-#         self.df_req: pd.DataFrame = pd.DataFrame()
-#         self._printed_this_run: bool = False
-#
-#         # ---- New feature flags / knobs (all default to safe choices) ----
-#         # Completeness check via _SUCCESS
-#         self.check_completeness: bool = bool(kwargs.get("check_completeness", True))
-#         self.require_success_marker: bool = bool(kwargs.get("require_success_marker", True))
-#         # Listing granularity: 'month' (default) or 'day'
-#         self.list_granularity: str = str(kwargs.get("list_granularity", "month"))
-#         # Data file suffixes to consider for age (default common formats)
-#         self.data_file_suffixes: Tuple[str, ...] = tuple(kwargs.get("data_file_suffixes", self.DATA_FILE_PATTERNS))
-#         # Timeouts
-#         self.list_timeout: float = float(kwargs.get("list_timeout", self.timeout))       # per-future
-#         self.total_timeout: float = float(kwargs.get("total_timeout", self.timeout))     # across all listings
-#         # Dependency-injected clock (UTC) for tests
-#         self._utcnow = kwargs.get("utcnow_func", None) or (lambda: dt.datetime.utcnow())
-#
-#         # ------------ Backward-compatible skip handling ------------
-#         # Keep legacy attribute and derive new internal canonical sets.
-#         self.skipped = list(skipped or kwargs.get("skipped", []) or [])
-#         self.skipped_paths = {p.rstrip("/") + "/" for p in self.skipped if isinstance(p, str)}
-#         self.skipped_dates = {p for p in self.skipped if isinstance(p, dt.date)}
-#
-#         # Validate fs presence (you rely on it)
-#         if not getattr(self, "fs", None):
-#             raise ValueError("UpdatePlanner requires a valid fsspec filesystem (fs).")
-#
-#     # --------------------- Back-compat property bridge ---------------------
-#     @property
-#     def skipped(self) -> List[Union[str, dt.date]]:  # type: ignore[override]
-#         """
-#         Backward-compatible view of skip configuration.
-#         Returns a merged list of path-strings and dates.
-#         """
-#         paths = sorted(self.skipped_paths)
-#         dates = sorted(self.skipped_dates)
-#         return [*paths, *dates]
-#
-#     @skipped.setter
-#     def skipped(self, value: List[Union[str, dt.date]]) -> None:  # type: ignore[override]
-#         """
-#         Accepts legacy assignments like:
-#             planner.skipped = ["s3://.../2025/01/03/", date(2025,1,4)]
-#         and keeps new internals in sync.
-#         """
-#         value = list(value or [])
-#         self.skipped_paths = {p.rstrip("/") + "/" for p in value if isinstance(p, str)}
-#         self.skipped_dates = {p for p in value if isinstance(p, dt.date)}
-#
-#     # --------------------- public helpers (kept) ---------------------
-#     def has_plan(self) -> bool:
-#         return isinstance(self.plan, pd.DataFrame) and not self.plan.empty
-#
-#     def required_count(self) -> int:
-#         return 0 if not isinstance(self.df_req, pd.DataFrame) else len(self.df_req)
-#
-#     # --------------------- core API (kept) ---------------------
-#     def generate_plan(
-#         self,
-#         start: Union[str, dt.date, None] = None,
-#         end: Union[str, dt.date, None] = None,
-#         freq: str = "D",
-#     ) -> pd.DataFrame:
-#         """Build a plan for [start, end]. Returns rows that require update (df_req)."""
-#         start = start or self.start_date
-#         end = end or self.end_date
-#         if start is None or end is None:
-#             raise ValueError("start and end must be provided (or set via parquet_* kwargs).")
-#
-#         sd = pd.to_datetime(start).date()
-#         ed = pd.to_datetime(end).date()
-#         if sd > ed:
-#             raise ValueError(f"Start date ({sd}) must be on or before end date ({ed}).")
-#
-#         self.logger.info(
-#             f"Generating update plan for {self.description} from {sd} to {ed}",
-#             extra=self._log_extra(),
-#         )
-#         self._generate_plan(sd, ed, freq=freq)
-#         self.logger.info(
-#             f"Plan built for {self.description}: {len(self.plan)} dates evaluated, "
-#             f"{len(self.df_req)} require update",
-#             extra=self._log_extra(),
-#         )
-#         return self.df_req
-#
-#     def show_update_plan(self) -> None:
-#         """Pretty-print the current plan once per run, now respecting terminal width fully."""
-#         if not self.has_plan():
-#             self.logger.info("No update plan to show.", extra=self._log_extra())
-#             return
-#         if self._printed_this_run:
-#             return
-#
-#         try:
-#             from rich.console import Console
-#             from rich.table import Table
-#
-#             console = Console()  # auto-detect terminal size
-#             terminal_width = console.size.width
-#
-#             table = Table(
-#                 title=f"Update Plan for {self.data_path}",
-#                 show_header=True,
-#                 header_style="bold magenta",
-#                 expand=True,  # fill available width
-#                 pad_edge=False,
-#             )
-#             max_w = max(terminal_width - 50, 640)
-#             for col in self.plan.columns:
-#                 if col in {"date", "update_category", "update_priority", "update_required", "file_exists"}:
-#                     table.add_column(col, justify="left", no_wrap=True, overflow="fold", max_width=max_w)
-#                 elif col == "description":
-#                     # Let description wrap, but set a max width to avoid huge columns
-#                     table.add_column(col, justify="left", overflow="fold", max_width=max_w)
-#                 else:
-#                     table.add_column(col, justify="left", overflow="fold")
-#
-#             for _, row in self.plan.iterrows():
-#                 table.add_row(*(str(row[c]) for c in self.plan.columns))
-#
-#             # Capture with the same console so width stays consistent
-#             with console.capture() as cap:
-#                 console.print(table)
-#             self.logger.info(f"Full Update Plan:\n{cap.get().strip()}", extra=self._log_extra())
-#
-#         except Exception:
-#             preview = self.plan.head(200).to_string(index=False)
-#             self.logger.info(f"Update Plan (first 200 rows):\n{preview}", extra=self._log_extra())
-#
-#         self._printed_this_run = True
-#
-#     def get_tasks_by_priority(self) -> Iterator[Tuple[int, List[dt.date]]]:
-#         """Yield (priority, [dates...]) batches, smallest priority first."""
-#         if not self.has_plan():
-#             return
-#         req = self.plan[self.plan["update_required"]]
-#         if req.empty:
-#             return
-#         for priority in sorted(req["update_priority"].unique()):
-#             dates_df = req[req["update_priority"] == priority]
-#             dates_df = dates_df.sort_values(by="date", ascending=not self.reverse_order)
-#             dates = dates_df["date"].tolist()
-#             if dates:
-#                 yield int(priority), dates
-#
-#     # --------------------- internals ---------------------
-#     @staticmethod
-#     def _ensure_trailing_slash(path: str) -> str:
-#         return path.rstrip("/") + "/"
-#
-#     @staticmethod
-#     def _month_floor(d: dt.date) -> dt.date:
-#         return d.replace(day=1)
-#
-#     @staticmethod
-#     def _iter_month_starts(start: dt.date, end: dt.date) -> Iterator[dt.date]:
-#         cur = start.replace(day=1)
-#         while cur <= end:
-#             yield cur
-#             y, m = cur.year, cur.month
-#             cur = dt.date(y + (m == 12), 1 if m == 12 else m + 1, 1)
-#
-#     def _month_prefix(self, month_start: dt.date) -> str:
-#         return f"{self.data_path}{month_start.year}/{month_start.month:02d}/"
-#
-#     def _day_prefix(self, d: dt.date) -> str:
-#         return f"{self.data_path}{d.year}/{d.month:02d}/{d.day:02d}/"
-#
-#     def _log_extra(self, **overrides) -> dict:
-#         base = {
-#             "sibi_dst_component": __name__,
-#             "date_of_update": self.reference_date.strftime("%Y-%m-%d"),
-#             "dataclass": self.description,
-#             "action_module_name": "update_plan",
-#         }
-#         base.update(overrides)
-#         return base
-#
-#     def _is_data_file(self, path: str) -> bool:
-#         base = path.rsplit("/", 1)[-1]
-#         if not base or base.startswith(".") or base in self.CONTROL_BASENAMES:
-#             return False
-#         lower = base.lower()
-#         return any(lower.endswith(suf) for suf in self.data_file_suffixes)
-#
-#     def _is_skipped(self, d: dt.date) -> bool:
-#         """True if the date or its canonical path is in the skip config."""
-#         just_path = f"{self.data_path}{d.year}/{d.month:02d}/{d.day:02d}/"
-#         return (d in self.skipped_dates) or (just_path in self.skipped_paths)
-#
-#     def _list_prefix(self, prefix: str) -> Dict[dt.date, Dict[str, object]]:
-#         """
-#         Return {date: {'files': [paths], 'has_success': bool, 'newest_ts': datetime|None}} under prefix.
-#         Uses fsspec.find(detail=True) for one-shot listing with metadata (mtime).  [oai_citation:0‡fsspec](https://filesystem-spec.readthedocs.io/en/latest/api.html?utm_source=chatgpt.com) [oai_citation:1‡GitHub](https://github.com/fsspec/filesystem_spec/blob/master/fsspec%2Fspec.py?utm_source=chatgpt.com)
-#         """
-#         try:
-#             items = self.fs.find(prefix, withdirs=False, detail=True)  # returns {path: info} when detail=True
-#         except Exception as e:
-#             self.logger.warning(f"Listing failed for {prefix}: {e}", extra=self._log_extra())
-#             return {}
-#
-#         out: Dict[dt.date, Dict[str, object]] = {}
-#         for path, info in items.items():
-#             parts = path.strip("/").split("/")
-#             if len(parts) < 4:
-#                 continue
-#             try:
-#                 y, m, dd = int(parts[-4]), int(parts[-3]), int(parts[-2])
-#                 d = dt.date(y, m, dd)
-#             except Exception:
-#                 continue
-#
-#             rec = out.setdefault(d, {"files": [], "has_success": False, "newest_ts": None})
-#             base = path.rsplit("/", 1)[-1]
-#             if base == "_SUCCESS":
-#                 rec["has_success"] = True
-#
-#             if self._is_data_file(path):
-#                 rec["files"].append(path)
-#                 mtime = info.get("mtime") or info.get("LastModified") or info.get("last_modified")
-#                 ts = None
-#                 if isinstance(mtime, (int, float)):
-#                     ts = dt.datetime.utcfromtimestamp(mtime)
-#                 elif isinstance(mtime, str):
-#                     try:
-#                         ts = pd.to_datetime(mtime, utc=True).to_pydatetime()
-#                     except Exception:
-#                         ts = None
-#                 elif isinstance(mtime, dt.datetime):
-#                     ts = mtime if mtime.tzinfo else mtime.replace(tzinfo=dt.timezone.utc)
-#                 if ts:
-#                     cur = rec["newest_ts"]
-#                     rec["newest_ts"] = ts if (cur is None or ts > cur) else cur
-#         return out
-#
-#     def _summarize_partition(
-#         self, d: dt.date, cache: Dict[dt.date, Dict[str, object]]
-#     ) -> Tuple[bool, Optional[float], bool]:
-#         """
-#         (exists, age_minutes, incomplete)
-#         - exists: True iff at least one *data* file is present for day `d`
-#         - age_minutes: minutes since the NEWEST data file (UTC 'now')
-#         - incomplete: True if files exist but required _SUCCESS is missing
-#         """
-#         rec = cache.get(d, {})
-#         files = rec.get("files", [])
-#         has_success = bool(rec.get("has_success", False))
-#         exists = len(files) > 0
-#         if not exists:
-#             return False, None, False
-#         newest_ts = rec.get("newest_ts")
-#         if newest_ts:
-#             now_utc = self._utcnow().replace(tzinfo=None)
-#             ts_naive = newest_ts.replace(tzinfo=None) if newest_ts.tzinfo else newest_ts
-#             age_min = max(0.0, (now_utc - ts_naive).total_seconds() / 60.0)
-#         else:
-#             age_min = None
-#         incomplete = self.check_completeness and self.require_success_marker and not has_success
-#         return True, age_min, incomplete
-#
-#     def _generate_plan(self, start: dt.date, end: dt.date, freq: str = "D") -> None:
-#         """
-#         Populate self.plan with all dates and self.df_req with the subset to update.
-#         - Pre-lists months or days (configurable) with timeouts that actually apply
-#         - Computes staleness from newest *data* file
-#         - Flags partitions without _SUCCESS as 'incomplete' (unless disabled)
-#         - Marks future dates as 'future' (not actionable)
-#         """
-#         dates: List[dt.date] = pd.date_range(start=start, end=end, freq=freq).date.tolist()
-#         history_start = self.reference_date - dt.timedelta(days=self.history_days_threshold)
-#         rows: List[Dict] = []
-#
-#         def is_future(d: dt.date) -> bool:
-#             return d > self.reference_date
-#
-#         # Choose listing units
-#         if self.list_granularity == "day":
-#             units: List[Tuple[str, dt.date]] = [("day", d) for d in dates]
-#         else:
-#             months = list(self._iter_month_starts(self._month_floor(start), self._month_floor(end)))
-#             units = [("month", m) for m in months]
-#
-#         self.logger.info(
-#             f"Pre-listing {len(units)} {'days' if self.list_granularity=='day' else 'month prefixes'} for {self.description}",
-#             extra=self._log_extra(),
-#         )
-#
-#         # Parallel listing with real timeout (uses futures.wait)  [oai_citation:2‡Python documentation](https://docs.python.org/3/library/concurrent.futures.html?utm_source=chatgpt.com) [oai_citation:3‡alexwlchan.net](https://alexwlchan.net/2019/adventures-with-concurrent-futures/?utm_source=chatgpt.com)
-#         caches: Dict[dt.date, Dict[dt.date, Dict[str, object]]] = {}
-#         max_workers = max(1, int(self.max_threads))
-#         with ThreadPoolExecutor(max_workers=max_workers) as ex:
-#             futs = {}
-#             for kind, val in units:
-#                 prefix = self._day_prefix(val) if kind == "day" else self._month_prefix(val)
-#                 futs[ex.submit(self._list_prefix, prefix)] = (kind, val)
-#             done, not_done = wait(futs, timeout=self.total_timeout or None)
-#             for f in done:
-#                 kind, val = futs[f]
-#                 try:
-#                     cache = f.result(timeout=self.list_timeout or None)
-#                 except Exception as e:
-#                     self.logger.warning(f"Listing failed for {kind}:{val} — {e}", extra=self._log_extra())
-#                     cache = {}
-#                 if kind == "month":
-#                     caches[val] = cache
-#                 else:
-#                     # day → store into its month bucket for summarization reuse
-#                     mk = val.replace(day=1)
-#                     caches.setdefault(mk, {}).update(cache)
-#             for f in not_done:
-#                 kind, val = futs[f]
-#                 self.logger.error(f"Listing timed out for {kind}:{val}", extra=self._log_extra())
-#                 if kind == "month":
-#                     caches[val] = {}
-#                 else:
-#                     caches.setdefault(val.replace(day=1), {})
-#
-#         # Summarize each date
-#         for d in dates:
-#             if is_future(d):
-#                 rows.append({
-#                     "date": d, "file_exists": False, "file_age_minutes": None,
-#                     "update_category": "future", "update_priority": self.priority_map.get("future", 99),
-#                     "update_required": False, "description": self.description,
-#                 })
-#                 continue
-#
-#             if self._is_skipped(d):
-#                 self.logger.debug(f"Skipping {d}: in skipped set.", extra=self._log_extra())
-#                 rows.append(self._make_row(d, history_start, False, None))
-#                 continue
-#
-#             month_key = d.replace(day=1)
-#             cache = caches.get(month_key, {})
-#             exists, age_min, incomplete = self._summarize_partition(d, cache)
-#
-#             # Incomplete partitions get their own category (unless overwrite)
-#             if incomplete and not self.overwrite:
-#                 rows.append({
-#                     "date": d, "file_exists": True, "file_age_minutes": age_min,
-#                     "update_category": "incomplete", "update_priority": self.priority_map.get("incomplete", 1),
-#                     "update_required": True, "description": self.description,
-#                 })
-#                 continue
-#
-#             # Fall back to your existing policy (overwrite / history / staleness / missing)
-#             rows.append(self._make_row(d, history_start, exists, age_min))
-#
-#         df = pd.DataFrame.from_records(rows)
-#         if not df.empty:
-#             df["date"] = pd.to_datetime(df["date"]).dt.date
-#             df["update_priority"] = df["update_priority"].astype(int)
-#
-#         df = df.sort_values(
-#             by=["update_priority", "date"],
-#             ascending=[True, not self.reverse_order],
-#             kind="mergesort",
-#         ).reset_index(drop=True)
-#
-#         self.plan = df
-#         self.df_req = df[df["update_required"]].copy()
-#         self._printed_this_run = False
-#
-#     # --------------------- original policy (kept) ---------------------
-#     def _make_row(
-#         self,
-#         date: dt.date,
-#         history_start: dt.date,
-#         file_exists: bool,
-#         file_age: Optional[float],
-#     ) -> Dict:
-#         """
-#         Build a single plan row based on flags and thresholds.
-#         (Categories 'future'/'incomplete' are injected earlier.)
-#         """
-#         within_history = history_start <= date <= self.reference_date
-#         update_required = False
-#
-#         if self.overwrite:
-#             category = "overwrite_forced"
-#             update_required = True
-#         elif within_history:
-#             if not file_exists:
-#                 category = "missing_in_history"
-#                 update_required = True
-#             elif file_age is not None and file_age > self.max_age_minutes:
-#                 category = "stale_in_history"
-#                 update_required = True
-#             else:
-#                 category = "file_is_recent"
-#         elif not file_exists and not self.ignore_missing:
-#             category = "create_missing"
-#             update_required = True
-#         else:
-#             category = "missing_ignored" if not file_exists else "file_is_recent"
-#
-#         return {
-#             "date": date,
-#             "file_exists": bool(file_exists),
-#             "file_age_minutes": file_age,
-#             "update_category": category,
-#             "update_priority": self.priority_map.get(category, 99),
-#             "update_required": bool(update_required),
-#             "description": self.description,
-#         }
-#
+        return base