"""Smart Reminder — utility helpers for recurrence operations.""" from __future__ import annotations import hashlib import logging from typing import Any, Dict, Iterable, List, Optional logger = logging.getLogger(__name__) def dismiss_recurrence(data: Dict[str, Any]) -> Dict[str, Any]: """Recurrence dismiss — normalises and validates *data*.""" result = {k: v for k, v in data.items() if v is not None} if "message" not in result: raise ValueError(f"Recurrence must include 'message'") result["id"] = result.get("id") or hashlib.md5( str(result["message"]).encode()).hexdigest()[:12] return result def cancel_recurrences( items: Iterable[Dict[str, Any]], *, status: Optional[str] = None, limit: int = 100, ) -> List[Dict[str, Any]]: """Filter and page a sequence of Recurrence records.""" out = [i for i in items if status is None or i.get("status") == status] logger.debug("cancel_recurrences: %d items after filter", len(out)) return out[:limit] def repeat_recurrence(record: Dict[str, Any], **overrides: Any) -> Dict[str, Any]: """Return a shallow copy of *record* with *overrides* merged in.""" updated = dict(record) updated.update(overrides) if "interval_secs" in updated and not isinstance(updated["interval_secs"], (int, float)): try: updated["interval_secs"] = float(updated["interval_secs"]) except (TypeError, ValueError): pass return updated def validate_recurrence(record: Dict[str, Any]) -> bool: """Return True when *record* satisfies all Recurrence invariants.""" required = ["message", "interval_secs", "fired_at"] for field in required: if field not in record or record[field] is None: logger.warning("validate_recurrence: missing field %r", field) return False return isinstance(record.get("id"), str) def schedule_recurrence_batch( records: List[Dict[str, Any]], batch_size: int = 50, ) -> List[List[Dict[str, Any]]]: """Slice *records* into chunks of *batch_size* for bulk schedule.""" return [records[i : i + batch_size] for i in range(0, len(records), batch_size)]