Rotation order
- Independent rotation, filled in order. Reordering never rewrites assignments that already exist.
+ Independent rotation, filled in order. After changing the order, use Apply rotation above to update this month's automatic shifts.
{% for member in schedule.rotation_members %}
-
@@ -93,4 +106,4 @@
{% endif %}
-{% endblock %}
\ No newline at end of file
+{% endblock %}
diff --git a/app/web.py b/app/web.py
index f2cf5ca..4f2f52d 100644
--- a/app/web.py
+++ b/app/web.py
@@ -21,7 +21,7 @@ from .models import (
User,
)
from .notifications import send_email_notification, send_many
-from .scheduling import assignments_by_date, ensure_month_assignments, month_bounds
+from .scheduling import assignments_by_date, ensure_month_assignments, iter_month_dates, month_bounds, rotation_user_for_date
from .security import hash_password, new_csrf_token, verify_password
router = APIRouter()
@@ -956,6 +956,11 @@ def _save_assignments(db: Session, assignments: Sequence[Assignment], form) -> i
changed.append((assignment, old_user, new_user))
db.commit()
+ _notify_assignment_changes(db, changed)
+ return len(changed)
+
+
+def _notify_assignment_changes(db: Session, changed: Sequence[tuple[Assignment, User | None, User | None]]) -> None:
for assignment, old_user, new_user in changed:
subject = f"Bar rota updated: {assignment.date:%d %b} – {assignment.schedule.name}"
if new_user:
@@ -976,7 +981,6 @@ def _save_assignments(db: Session, assignments: Sequence[Assignment], form) -> i
f"You are no longer assigned to {assignment.schedule.name} on {assignment.date:%A, %d %B %Y}.\n\n"
f"View the rota: {settings.normalized_base_url}/",
)
- return len(changed)
def _month_assignments(db: Session, year: int, month: int) -> list[Assignment]:
@@ -1009,6 +1013,91 @@ async def admin_schedule_assignments(
return _redirect(f"/admin/schedules/{schedule.id}?year={year}&month={month}")
+def _apply_rotation_to_rows(db: Session, schedule: Schedule, assignments: Sequence[Assignment]) -> int:
+ """Explicit admin edit: retain assignment IDs and invalidate requests when ownership changes."""
+ changed: list[tuple[Assignment, User | None, User | None]] = []
+ touched = 0
+ for assignment in assignments:
+ new_user = rotation_user_for_date(db, schedule, assignment.date)
+ new_user_id = new_user.id if new_user else None
+ if assignment.user_id == new_user_id and assignment.source == "generated":
+ continue
+ if assignment.user_id != new_user_id:
+ old_user = db.get(User, assignment.user_id) if assignment.user_id is not None else None
+ changed.append((assignment, old_user, new_user))
+ if assignment.id is not None:
+ requests = db.scalars(select(SwapRequest).where(
+ SwapRequest.assignment_id == assignment.id,
+ SwapRequest.status.in_(["open", "pending_approval"]),
+ )).all()
+ for swap in requests:
+ swap.status = "cancelled"
+ assignment.user_id = new_user_id
+ assignment.user = new_user
+ assignment.source = "generated"
+ touched += 1
+ db.commit()
+ _notify_assignment_changes(db, changed)
+ return touched
+
+
+@router.post("/admin/schedules/{schedule_id}/apply-rotation")
+def admin_apply_rotation(
+ schedule_id: int,
+ request: Request,
+ year: int = Form(...),
+ month: int = Form(...),
+ csrf_token: str = Form(...),
+ db: Session = Depends(get_db),
+):
+ _require_admin(request, db)
+ _check_csrf(request, csrf_token)
+ schedule = _get_schedule(db, schedule_id)
+ if schedule.schedule_type != "rotation":
+ raise HTTPException(status_code=400, detail="This schedule does not use a rotation")
+ year, month = _parse_month(year, month)
+ rows = {row.date: row for row in _month_assignments(db, year, month) if row.schedule_id == schedule.id}
+ assignments = []
+ created = 0
+ for day in iter_month_dates(year, month):
+ if day.weekday() not in schedule.weekday_set:
+ continue
+ row = rows.get(day)
+ if row is None:
+ row = Assignment(schedule=schedule, date=day, source="generated")
+ db.add(row)
+ created += 1
+ if row.source == "generated":
+ assignments.append(row)
+ changed = _apply_rotation_to_rows(db, schedule, assignments)
+ _flash(request, f"Applied rotation for {calendar.month_name[month]} {year}: "
+ f"created {created} shift(s), updated {changed} assignment(s). Manual assignments and swaps were kept.")
+ return _redirect(f"/admin/schedules/{schedule.id}?year={year}&month={month}")
+
+
+@router.post("/admin/schedules/{schedule_id}/assignments/{assignment_id}/use-rotation")
+def admin_assignment_use_rotation(
+ schedule_id: int,
+ assignment_id: int,
+ request: Request,
+ csrf_token: str = Form(...),
+ db: Session = Depends(get_db),
+):
+ _require_admin(request, db)
+ _check_csrf(request, csrf_token)
+ schedule = _get_schedule(db, schedule_id)
+ if schedule.schedule_type != "rotation":
+ raise HTTPException(status_code=400, detail="This schedule does not use a rotation")
+ assignment = db.get(Assignment, assignment_id)
+ if assignment is None or assignment.schedule_id != schedule.id:
+ raise HTTPException(status_code=404, detail="Assignment not found")
+ if assignment.source != "manual" or assignment.date.weekday() not in schedule.weekday_set:
+ raise HTTPException(status_code=400, detail="Only manual rotation assignments can be reset")
+ _apply_rotation_to_rows(db, schedule, [assignment])
+ _flash(request, "Manual assignment removed; this shift now uses the rotation.")
+ return _redirect(f"/admin/schedules/{schedule.id}?year={assignment.date.year}&month={assignment.date.month}")
+
+
@router.get("/admin/notifications", response_class=HTMLResponse)
def admin_notifications(request: Request, db: Session = Depends(get_db)):
_require_admin(request, db)
diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml
index 7568bc4..327e4f1 100644
--- a/docker-compose.dev.yml
+++ b/docker-compose.dev.yml
@@ -6,17 +6,18 @@ services:
build:
context: .
target: dev
- container_name: bar-rota-dev
env_file:
- path: .env
required: false
ports:
- - "8000:8000"
+ - "${PORT:-8000}:8000"
volumes:
- ./app:/app/app
- ./tests:/app/tests
+ - ./migrations:/app/migrations
+ - ./alembic.ini:/app/alembic.ini:ro
- ${DATA_DIR:-./data}:/app/data
- command: ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]
+ command: ["uvicorn", "app.main:app", "--host", "0.0.0.0", "--port", "8000", "--reload", "--reload-dir", "app", "--reload-dir", "migrations"]
checks:
build:
@@ -28,9 +29,12 @@ services:
SECRET_KEY: test-only-secret-key
DATABASE_URL: sqlite:///./data/test.db
INITIAL_ADMIN_PASSWORD: test-only-password
+ SMTP_HOST: ""
PYTHONPATH: /app
volumes:
- ./app:/app/app
- ./tests:/app/tests
- ./pyproject.toml:/app/pyproject.toml:ro
- command: ["sh", "-c", "python -m compileall -q app && pytest -ra"]
\ No newline at end of file
+ - ./migrations:/app/migrations:ro
+ - ./alembic.ini:/app/alembic.ini:ro
+ command: ["sh", "-c", "python -m compileall -q app && pytest -ra"]
diff --git a/docker-compose.yml b/docker-compose.yml
index 27ed663..4d60e2b 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -1,12 +1,20 @@
services:
bar-rota:
- build: .
+ build:
+ context: .
+ target: production
container_name: bar-rota
restart: unless-stopped
env_file:
- path: .env
required: false
ports:
- - "${PORT}:8000"
+ - "${PORT:-8000}:8000"
volumes:
- - ${DATA_DIR}:/app/data
+ - ${DATA_DIR:-./data}:/app/data
+ healthcheck:
+ test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/health', timeout=5)"]
+ interval: 30s
+ timeout: 10s
+ start_period: 10s
+ retries: 3
diff --git a/docs/ARCHITECTURE.md b/docs/ARCHITECTURE.md
index b6cc338..b956dbf 100644
--- a/docs/ARCHITECTURE.md
+++ b/docs/ARCHITECTURE.md
@@ -16,7 +16,7 @@ Single FastAPI process with server-rendered HTML and a SQLite database volume. T
## Data lifecycle
1. Visiting a month calls `ensure_month_assignments`.
-2. Missing Night rows are created unassigned.
+2. Missing Night rows use the repeating monthly pattern, or remain unassigned for unconfigured days.
3. Missing lunch rows are created from each independent rotation.
4. Admin assigns/overrides rows on each schedule's own configuration page (the night pattern, or a lunch month view).
5. Member change requests operate on concrete assignment IDs.
@@ -26,4 +26,6 @@ Single FastAPI process with server-rendered HTML and a SQLite database volume. T
`docker-compose.yml` mounts `./data` at `/app/data`, so the SQLite DB persists across image rebuilds. It is the service stack only. `docker-compose.dev.yml` is the development stack: it builds the `dev` image target, bind-mounts `app/` and `tests/` for auto-reload, and hosts the containerized test runner.
+The deployment stack builds the `production` target (also the default Docker image target), which inherits the non-reloading runtime from `base`. Development also mounts Alembic configuration and migrations so schema changes can be authored in a container. `make` database commands use the host virtualenv by default, or a one-off development container with `RUNTIME=docker`. Both use the same `data/` directory by default. The checks service has independent settings and no persistent data mount, allowing app data entry and automated tests to run concurrently.
+
For larger installations, the SQLAlchemy configuration can point to PostgreSQL, but the project intentionally ships only the SQLite deployment path today.
diff --git a/docs/DOMAIN.md b/docs/DOMAIN.md
index f8f4ae6..931d41b 100644
--- a/docs/DOMAIN.md
+++ b/docs/DOMAIN.md
@@ -38,6 +38,10 @@ For a rotation schedule, the member is chosen from the ordered active rotation m
Only missing assignments are generated. Existing rows are never rewritten.
+An administrator can explicitly **Apply rotation** on a lunch schedule's month page. This recalculates that month's `generated` assignments using the current active rotation members and fills missing dates for that schedule. Manual assignments and completed swaps are preserved. Simply editing the rotation order still does not rewrite any assignment.
+
+An administrator can choose **Use rotation** on a `manual` assignment to remove that override and immediately recalculate that date from the current rotation. Its source becomes `generated`; if there are no active rotation members, it remains unassigned until an administrator applies a configured rotation. These actions preserve assignment IDs and notes. If ownership changes, open or pending-approval change requests for the assignment are cancelled, and assignment-change notifications are attempted after saving.
+
## People and logins
- `email` and `phone` are optional.
diff --git a/tests/test_admin_schedule_assignments.py b/tests/test_admin_schedule_assignments.py
index 31f9b70..a2f3135 100644
--- a/tests/test_admin_schedule_assignments.py
+++ b/tests/test_admin_schedule_assignments.py
@@ -9,7 +9,7 @@ from sqlalchemy.pool import StaticPool
from app.database import Base, get_db
from app.main import app
-from app.models import Assignment, MonthlyPattern, RotationMember, Schedule, User
+from app.models import Assignment, MonthlyPattern, NotificationLog, RotationMember, Schedule, SwapRequest, User
from app.scheduling import ensure_month_assignments
from app.security import hash_password
@@ -261,3 +261,184 @@ def test_month_page_redirects_to_the_schedules_hub(client, db):
assert response.status_code == 303
assert response.headers["location"] == "/admin/schedules"
+
+
+def test_apply_rotation_updates_only_automatic_rows_in_selected_schedule_and_month(client, db):
+ night, sat, sun, ana, bruno, admin = seed(db)
+ token = sign_in(client)
+ ensure_month_assignments(db, YEAR, MONTH)
+ ensure_month_assignments(db, YEAR, MONTH + 1)
+ manual = assignment_on(db, sat, date(YEAR, MONTH, 10))
+ manual.user_id, manual.source = admin.id, "manual"
+ swapped = assignment_on(db, sat, date(YEAR, MONTH, 17))
+ swapped.user_id, swapped.source = bruno.id, "swap"
+ unassigned = assignment_on(db, sat, date(YEAR, MONTH, 24))
+ unassigned.user_id = None
+ first = assignment_on(db, sat, date(YEAR, MONTH, 3))
+ assert first.user_id == bruno.id
+ first.note = "Keep this note"
+ db.commit()
+ untouched = {
+ row.id: (row.user_id, row.source)
+ for row in db.scalars(select(Assignment)).all()
+ if row.schedule_id != sat.id or row.date.month != MONTH or row.source != "generated"
+ }
+ members = db.scalars(select(RotationMember).where(
+ RotationMember.schedule_id == sat.id
+ ).order_by(RotationMember.position)).all()
+ members[0].position = 99
+ db.flush()
+ members[1].position = 0
+ db.flush()
+ members[0].position = 1
+ db.commit()
+
+ # Editing order alone leaves the published assignment stable.
+ ensure_month_assignments(db, YEAR, MONTH)
+ assert db.get(Assignment, first.id).user_id == bruno.id
+ response = client.post(f"/admin/schedules/{sat.id}/apply-rotation", data={
+ "csrf_token": token, "year": YEAR, "month": MONTH,
+ }, follow_redirects=False)
+
+ assert response.status_code == 303
+ assert response.headers["location"] == f"/admin/schedules/{sat.id}?year={YEAR}&month={MONTH}"
+ db.expire_all()
+ assert db.get(Assignment, first.id).user_id == ana.id
+ assert db.get(Assignment, first.id).note == "Keep this note"
+ assert db.get(Assignment, unassigned.id).user_id == bruno.id
+ for assignment_id, expected in untouched.items():
+ row = db.get(Assignment, assignment_id)
+ assert (row.user_id, row.source) == expected
+ log_count = len(db.scalars(select(NotificationLog)).all())
+ client.post(f"/admin/schedules/{sat.id}/apply-rotation", data={
+ "csrf_token": token, "year": YEAR, "month": MONTH,
+ })
+ assert len(db.scalars(select(NotificationLog)).all()) == log_count
+
+
+def test_apply_rotation_creates_only_missing_dates_for_selected_schedule(client, db):
+ _, sat, _, _, _, _ = seed(db)
+ token = sign_in(client)
+ response = client.post(f"/admin/schedules/{sat.id}/apply-rotation", data={
+ "csrf_token": token, "year": 2027, "month": 1,
+ }, follow_redirects=False)
+ assert response.status_code == 303
+ rows = db.scalars(select(Assignment).where(Assignment.date >= date(2027, 1, 1))).all()
+ assert len(rows) == 5
+ assert all(row.schedule_id == sat.id and row.date.weekday() == 5 and row.user_id is not None for row in rows)
+
+
+@pytest.mark.parametrize("status", ["open", "pending_approval"])
+@pytest.mark.parametrize("action", ["apply-rotation", "use-rotation"])
+def test_rotation_actions_cancel_stale_requests_and_notify_after_reassignment(client, db, status, action):
+ _, sat, _, ana, bruno, _ = seed(db)
+ token = sign_in(client)
+ ensure_month_assignments(db, YEAR, MONTH)
+ row = assignment_on(db, sat, date(YEAR, MONTH, 3)) # Bruno's date in the seeded order
+ row.user_id = ana.id
+ row.source = "manual" if action == "use-rotation" else "generated"
+ row.note = "Keep assignment history"
+ db.add(SwapRequest(assignment_id=row.id, requester_id=ana.id, accepted_by_id=bruno.id, status=status))
+ db.commit()
+ url = f"/admin/schedules/{sat.id}/apply-rotation" if action == "apply-rotation" else (
+ f"/admin/schedules/{sat.id}/assignments/{row.id}/use-rotation"
+ )
+ response = client.post(url, data={"csrf_token": token, "year": YEAR, "month": MONTH}, follow_redirects=False)
+
+ assert response.status_code == 303
+ db.expire_all()
+ restored = db.get(Assignment, row.id)
+ assert (restored.user_id, restored.source, restored.note) == (bruno.id, "generated", "Keep assignment history")
+ request = db.scalar(select(SwapRequest).where(SwapRequest.assignment_id == row.id))
+ assert request.status == "cancelled"
+ logs = db.scalars(select(NotificationLog)).all()
+ assert {(log.user_id, log.event_type) for log in logs} == {
+ (ana.id, "assignment_changed"), (bruno.id, "assignment_updated"),
+ }
+
+
+def test_use_rotation_removes_override_even_when_person_is_already_correct(client, db):
+ _, sat, _, ana, bruno, _ = seed(db)
+ token = sign_in(client)
+ ensure_month_assignments(db, YEAR, MONTH)
+ row = assignment_on(db, sat, date(YEAR, MONTH, 3))
+ row.source = "manual"
+ db.add(SwapRequest(assignment_id=row.id, requester_id=bruno.id, target_user_id=ana.id, status="open"))
+ db.commit()
+ client.post(f"/admin/schedules/{sat.id}/assignments/{row.id}/use-rotation", data={"csrf_token": token})
+ db.expire_all()
+ assert (row.user_id, row.source) == (bruno.id, "generated")
+ assert db.scalar(select(SwapRequest).where(SwapRequest.assignment_id == row.id)).status == "open"
+ assert not db.scalars(select(NotificationLog)).all()
+
+
+def test_reset_unassigned_manual_row_and_apply_later_when_members_become_active(client, db):
+ _, sat, _, ana, bruno, _ = seed(db)
+ token = sign_in(client)
+ ensure_month_assignments(db, YEAR, MONTH)
+ row = assignment_on(db, sat, date(YEAR, MONTH, 3))
+ row.user_id, row.source = None, "manual"
+ ana.is_active = bruno.is_active = False
+ db.commit()
+ client.post(f"/admin/schedules/{sat.id}/assignments/{row.id}/use-rotation", data={"csrf_token": token})
+ db.expire_all()
+ assert (row.user_id, row.source) == (None, "generated")
+ ana.is_active = True
+ db.commit()
+ client.post(f"/admin/schedules/{sat.id}/apply-rotation", data={
+ "csrf_token": token, "year": YEAR, "month": MONTH,
+ })
+ db.expire_all()
+ assert (row.user_id, row.source) == (ana.id, "generated")
+
+
+def test_rotation_controls_render_manual_rows_as_separate_forms(client, db):
+ night, sat, _, _, _, _ = seed(db)
+ sign_in(client)
+ ensure_month_assignments(db, YEAR, MONTH)
+ row = assignment_on(db, sat, date(YEAR, MONTH, 3))
+ row.source = "manual"
+ db.commit()
+ page = client.get(f"/admin/schedules/{sat.id}?year={YEAR}&month={MONTH}").text
+ assert "Apply rotation to October" in page
+ assert f'form="use-rotation-{row.id}"' in page
+ assert f'action="/admin/schedules/{sat.id}/assignments/{row.id}/use-rotation"' in page
+ assert page.index(f'id="use-rotation-{row.id}"') > page.index("Save October")
+ assert "Apply rotation" not in client.get(f"/admin/schedules/{night.id}").text
+
+
+def test_rotation_actions_reject_invalid_schedule_row_and_month(client, db):
+ night, sat, sun, _, _, _ = seed(db)
+ token = sign_in(client)
+ ensure_month_assignments(db, YEAR, MONTH)
+ row = assignment_on(db, sat, date(YEAR, MONTH, 3))
+ base = {"csrf_token": token, "year": YEAR, "month": MONTH}
+ assert client.post(f"/admin/schedules/{night.id}/apply-rotation", data=base).status_code == 400
+ assert client.post(f"/admin/schedules/{sat.id}/apply-rotation", data={**base, "month": 13}).status_code == 400
+ assert client.post(f"/admin/schedules/{sun.id}/assignments/{row.id}/use-rotation", data=base).status_code == 404
+ assert client.post(f"/admin/schedules/{sat.id}/assignments/{row.id}/use-rotation", data=base).status_code == 400
+ row.source = "swap"
+ db.commit()
+ assert client.post(f"/admin/schedules/{sat.id}/assignments/{row.id}/use-rotation", data=base).status_code == 400
+
+
+@pytest.mark.parametrize("action", ["apply-rotation", "use-rotation"])
+def test_rotation_actions_require_admin_and_csrf(client, db, action):
+ _, sat, _, _, _, _ = seed(db)
+ ensure_month_assignments(db, YEAR, MONTH)
+ row = assignment_on(db, sat, date(YEAR, MONTH, 3))
+ row.source = "manual"
+ member = User(name="Member", username="member", password_hash=hash_password("member-password"), is_active=True)
+ db.add(member)
+ db.commit()
+ url = f"/admin/schedules/{sat.id}/apply-rotation" if action == "apply-rotation" else (
+ f"/admin/schedules/{sat.id}/assignments/{row.id}/use-rotation"
+ )
+ data = {"csrf_token": "invalid", "year": YEAR, "month": MONTH}
+ assert client.post(url, data=data).status_code == 401
+ token = sign_in(client)
+ assert client.post(url, data=data).status_code == 400
+ client.post("/logout", data={"csrf_token": token})
+ token = CSRF_RE.search(client.get("/login").text).group(1)
+ client.post("/login", data={"username": "member", "password": "member-password", "csrf_token": token})
+ assert client.post(url, data={**data, "csrf_token": token}).status_code == 403
diff --git a/tests/test_cli.py b/tests/test_cli.py
new file mode 100644
index 0000000..3efd12e
--- /dev/null
+++ b/tests/test_cli.py
@@ -0,0 +1,76 @@
+from datetime import date, timedelta
+
+import pytest
+from sqlalchemy import create_engine, select
+from sqlalchemy.orm import sessionmaker
+
+from app import cli, database
+from app.models import Assignment, MonthlyPattern, RotationMember, Schedule, User
+from app.security import verify_password
+
+
+@pytest.fixture()
+def sessions(tmp_path, monkeypatch):
+ url = f"sqlite:///{tmp_path / 'demo.db'}"
+ engine = create_engine(url)
+ factory = sessionmaker(bind=engine, expire_on_commit=False)
+ monkeypatch.setattr(database.settings, "database_url", url)
+ monkeypatch.setattr(database, "engine", engine)
+ monkeypatch.setattr(cli, "SessionLocal", factory)
+ yield factory
+ engine.dispose()
+
+
+def test_demo_seeds_repeating_schedules_and_is_repeatable(sessions):
+ cli.seed_demo()
+ today = date.today()
+ next_month = (today.replace(day=28) + timedelta(days=4)).replace(day=1)
+ with sessions() as db:
+ people = db.scalars(select(User).where(User.username.in_(["alice", "bruno", "carla", "diogo"]))).all()
+ assert len(people) == 4
+ assert all(verify_password("demo-password", person.password_hash) for person in people)
+ assert len(db.scalars(select(MonthlyPattern)).all()) == 31
+ assert len(db.scalars(select(RotationMember)).all()) == 8
+ assignments = db.scalars(select(Assignment)).all()
+ assert all(row.user_id is not None for row in assignments)
+ assert {row.date.replace(day=1) for row in assignments} == {today.replace(day=1), next_month}
+ original = [(row.id, row.date, row.user_id, row.source) for row in assignments]
+
+ cli.seed_demo()
+ with sessions() as db:
+ assert len(db.scalars(select(User)).all()) == 5
+ assert len(db.scalars(select(MonthlyPattern)).all()) == 31
+ assert len(db.scalars(select(RotationMember)).all()) == 8
+ assert [(row.id, row.date, row.user_id, row.source) for row in db.scalars(select(Assignment)).all()] == original
+
+
+def test_demo_preserves_existing_people_configuration_and_assignments(sessions):
+ cli.init_db()
+ today = date.today()
+ assigned_day = today.replace(day=1) if today.day != 1 else today + timedelta(days=1)
+ with sessions() as db:
+ night = db.scalar(select(Schedule).where(Schedule.slug == "night"))
+ saturday = db.scalar(select(Schedule).where(Schedule.slug == "saturday-lunch"))
+ alice = User(name="My Alice", username="alice", password_hash="keep-this-password", is_active=True)
+ db.add(alice)
+ db.flush()
+ db.add_all([
+ MonthlyPattern(schedule_id=night.id, day_of_month=1, user_id=alice.id),
+ RotationMember(schedule_id=saturday.id, position=10, user_id=alice.id),
+ Assignment(schedule_id=night.id, date=today, user_id=None, source="manual", note="Keep unassigned"),
+ Assignment(schedule_id=night.id, date=assigned_day, user_id=alice.id, source="swap"),
+ ])
+ db.commit()
+ original = [(row.id, row.user_id, row.source, row.note) for row in db.scalars(select(Assignment)).all()]
+
+ cli.seed_demo()
+ with sessions() as db:
+ alice = db.scalar(select(User).where(User.username == "alice"))
+ assert (alice.name, alice.password_hash) == ("My Alice", "keep-this-password")
+ pattern = db.scalars(select(MonthlyPattern).where(MonthlyPattern.schedule_id == night.id)).all()
+ assert [(row.day_of_month, row.user_id) for row in pattern] == [(1, alice.id)]
+ rotation = db.scalars(select(RotationMember).where(RotationMember.schedule_id == saturday.id)).all()
+ assert [(row.position, row.user_id) for row in rotation] == [(10, alice.id)]
+ for assignment_id, user_id, source, note in original:
+ row = db.get(Assignment, assignment_id)
+ assert (row.user_id, row.source, row.note) == (user_id, source, note)