mirror of
https://github.com/hanzoai/insights.git
synced 2026-11-03 00:52:32 +00:00
zeekay
1b367e2b62
The tree was a debranded snapshot of upstream taken around 2026-02-20, and upstream has moved about five months since. This replaces it with upstream master, mechanically re-derived: bin/debrand does the rename, and the pipeline and squashed migrations are restored from the shipped tree. Not a merge. The previous import replaced the tree wholesale in a single commit, so there is no shared content history to merge against. Reproducing the rename on the new tree is what actually converges. What survives: .hanzo/workflows, the Dockerfiles and bin/docker* entrypoints, compose files, LLM.md, and insights/migrations. The squashed initial migration is load-bearing — the live database records it by name, so upstream's ~918 individual migrations cannot come back. Nine products are dropped as unreferenced. Twenty-one more that we do not run are kept anyway because the core app still imports them; unpicking those is a refactor and does not belong in a rebase. This base does not build yet, and the reasons are not in the rename: upstream now wants Python 3.13.13 against our 3.12.12 and Django 5.2 against our 4.2, and the dependency manifests still name upstream packages that we republish under our own names. Those are decisions, not renames.
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
"""Local rrule helpers for scheduled vision actions.
|
|
|
|
Behaviorally identical to `products/workflows/backend/utils/rrule_utils.py` — copied rather
|
|
than imported so replay_vision doesn't take a cross-product dependency on the Workflows
|
|
backend just for date math. If these ever need to share an implementation, lift them into a
|
|
common module rather than wiring a product-to-product import.
|
|
"""
|
|
|
|
from datetime import UTC, datetime
|
|
from zoneinfo import ZoneInfo, ZoneInfoNotFoundError
|
|
|
|
from dateutil.rrule import rrulestr
|
|
|
|
|
|
def validate_rrule(rrule_string: str) -> None:
|
|
"""Validate an RRULE string. Raises ValueError if invalid."""
|
|
# Case-insensitive: dateutil normalizes property names to uppercase, so a lowercase
|
|
# `dtstart:` would otherwise slip past and override the managed start.
|
|
if "DTSTART" in rrule_string.upper():
|
|
raise ValueError("RRULE must not contain DTSTART (starts_at is managed separately)")
|
|
try:
|
|
rrulestr(rrule_string)
|
|
except (TypeError, ValueError) as e:
|
|
# rrulestr raises ValueError for most malformed input (bad FREQ/BYDAY/INTERVAL, empty
|
|
# string) and TypeError only for the missing-FREQ case — normalize both to ValueError.
|
|
raise ValueError(str(e)) from e
|
|
|
|
|
|
def validate_timezone(timezone_str: str) -> None:
|
|
"""Validate an IANA timezone name. Raises ValueError if unknown/invalid.
|
|
|
|
Without this, an unvalidated TZ string is accepted at the API and only blows up later in
|
|
`compute_next_occurrences` (ZoneInfo lookup) inside the scheduling workflow.
|
|
"""
|
|
try:
|
|
ZoneInfo(timezone_str)
|
|
except (ZoneInfoNotFoundError, ValueError) as e:
|
|
raise ValueError(f"Unknown timezone: {timezone_str!r}") from e
|
|
|
|
|
|
def compute_next_occurrences(
|
|
rrule_string: str,
|
|
starts_at: datetime,
|
|
timezone_str: str = "UTC",
|
|
after: datetime | None = None,
|
|
count: int = 1,
|
|
) -> list[datetime]:
|
|
"""Compute the next `count` occurrences from an RRULE string.
|
|
|
|
Expands the RRULE in naive local time so that e.g. "9 AM Europe/Prague" stays at 9 AM
|
|
local across DST changes, then converts to UTC for storage.
|
|
"""
|
|
tz = ZoneInfo(timezone_str)
|
|
|
|
starts_local = starts_at.astimezone(tz).replace(tzinfo=None)
|
|
rule = rrulestr(rrule_string, dtstart=starts_local, ignoretz=True)
|
|
after_local = (after or datetime.now(UTC)).astimezone(tz).replace(tzinfo=None)
|
|
|
|
occurrences: list[datetime] = []
|
|
current = after_local
|
|
for _ in range(count * 10): # safety limit against unbounded gaps
|
|
next_dt = rule.after(current, inc=False)
|
|
if next_dt is None:
|
|
break
|
|
occurrences.append(next_dt.replace(tzinfo=tz).astimezone(UTC))
|
|
if len(occurrences) >= count:
|
|
break
|
|
current = next_dt
|
|
|
|
return occurrences
|