fixture.py

"""Fixture data for previewing the dashboard before the Staff Allocations DB exists."""

from datetime import date, timedelta

_today = date.today()


def _id(slug: str) -> str:
    return f"fixture-{slug}"


FIXTURE_TEAMS = [
    {
        "id": _id("heron-team-a"),
        "name": "Heron Team A — Adversarial Probing",
        "poc": "Person A",
        "status_tags": ["Fellowship | Leg 3"],
        "tags": ["Heron 🦅"],
        "next_deadline": _today + timedelta(weeks=4),
        "next_deadline_name": "Leg 3 Full Draft",
        "condition": "On track",
    },
    {
        "id": _id("heron-team-b"),
        "name": "Heron Team B — Eval Suite",
        "poc": "Person A",
        "status_tags": ["Fellowship | Leg 4"],
        "tags": ["Heron 🦅"],
        "next_deadline": _today + timedelta(weeks=2),
        "next_deadline_name": "Leg 4 Submission Checklist",
        "condition": "At risk",
    },
    {
        "id": _id("studio-team-c"),
        "name": "Studio Team C — Control Project",
        "poc": "Mackenzie",
        "status_tags": ["Studio | Active"],
        "tags": [],
        "next_deadline": _today + timedelta(weeks=6),
        "next_deadline_name": "Studio Mid-cycle Review",
        "condition": "Ahead of schedule",
    },
    {
        "id": _id("studio-team-d"),
        "name": "Studio Team D — Interp",
        "poc": "Mackenzie",
        "status_tags": ["Studio | Active"],
        "tags": ["Martian 👽"],
        "next_deadline": _today + timedelta(weeks=10),
        "next_deadline_name": "Studio Final",
        "condition": "On track",
    },
    {
        "id": _id("fellowship-team-e"),
        "name": "Fellowship Team E — Specs",
        "poc": "Person A",
        "status_tags": ["Fellowship | Leg 1"],
        "tags": [],
        "next_deadline": _today + timedelta(weeks=8),
        "next_deadline_name": "Leg 1 Prelim Experiment",
        "condition": "On track",
    },
]

_team = {t["name"]: t["id"] for t in FIXTURE_TEAMS}

FIXTURE_ALLOCATIONS = [
    # Team-bound rows
    {
        "id": _id("a-eitan-heron-a"),
        "person": "Person A",
        "hours_per_week": 3.0,
        "activity": "Heron Team A weekly",
        "program": "Heron",
        "team_id": _team["Heron Team A — Adversarial Probing"],
        "end_date": None,
        "active": True,
        "notes": "",
    },
    {
        "id": _id("a-eitan-heron-b"),
        "person": "Person A",
        "hours_per_week": 4.0,
        "activity": "Heron Team B Leg 4 push",
        "program": "Heron",
        "team_id": _team["Heron Team B — Eval Suite"],
        "end_date": None,
        "active": True,
        "notes": "",
    },
    {
        "id": _id("a-mackenzie-c"),
        "person": "Person B",
        "hours_per_week": 5.0,
        "activity": "Studio Team C mentoring",
        "program": "Studio",
        "team_id": _team["Studio Team C — Control Project"],
        "end_date": None,
        "active": True,
        "notes": "",
    },
    {
        "id": _id("a-mackenzie-d"),
        "person": "Person B",
        "hours_per_week": 4.0,
        "activity": "Studio Team D mentoring",
        "program": "Martian",
        "team_id": _team["Studio Team D — Interp"],
        "end_date": None,
        "active": True,
        "notes": "",
    },
    {
        "id": _id("a-eitan-fellowship-e"),
        "person": "Person A",
        "hours_per_week": 2.0,
        "activity": "Fellowship Team E mentoring",
        "program": "Fellowship",
        "team_id": _team["Fellowship Team E — Specs"],
        "end_date": None,
        "active": True,
        "notes": "",
    },
    # Program-ops rows (no team relation; manual end-dates where relevant)
    {
        "id": _id("a-mackenzie-sps-launch"),
        "person": "Person B",
        "hours_per_week": 8.0,
        "activity": "SPS launch ops",
        "program": "SPS",
        "team_id": None,
        "end_date": _today + timedelta(weeks=3),
        "active": True,
        "notes": "Tally form, website, mentor outreach",
    },
    {
        "id": _id("a-eitan-sps-launch"),
        "person": "Person A",
        "hours_per_week": 4.0,
        "activity": "SPS launch ops",
        "program": "SPS",
        "team_id": None,
        "end_date": _today + timedelta(weeks=3),
        "active": True,
        "notes": "",
    },
    {
        "id": _id("a-eitan-rpm-admin"),
        "person": "Person A",
        "hours_per_week": 3.0,
        "activity": "RPM ongoing duties",
        "program": "RPM Ops",
        "team_id": None,
        "end_date": None,
        "active": True,
        "notes": "WPF triage, lab@apart inbox, Discord",
    },
    {
        "id": _id("a-mackenzie-rpm-admin"),
        "person": "Person B",
        "hours_per_week": 6.0,
        "activity": "RPM ongoing duties",
        "program": "RPM Ops",
        "team_id": None,
        "end_date": None,
        "active": True,
        "notes": "",
    },
    {
        "id": _id("a-eitan-heron-v2-planning"),
        "person": "Person A",
        "hours_per_week": 2.0,
        "activity": "Heron v2 planning",
        "program": "Heron",
        "team_id": None,
        "end_date": _today + timedelta(weeks=12),
        "active": True,
        "notes": "Scoping for double cohort",
    },
    {
        "id": _id("a-clement-adhoc"),
        "person": "Person C",
        "hours_per_week": 4.0,
        "activity": "Ad-hoc reviews",
        "program": "Ad-hoc",
        "team_id": None,
        "end_date": None,
        "active": True,
        "notes": "Studio proposal reviews + studio teams",
    },
]