Rework balances around per-parent days left and deduce the partner from FK
Manual entry now takes one parent's sjukpenningnivå, lägstanivå and dubbeldagar left (plus optional reserved days) and derives the rest in build_snapshot, shared with the scraper. The scraper also reads the family panel and exposes a partner deduction; with two parents it is saved as a 'deduced' snapshot (migration 002). Parent count is the single source of entitlement: one member gets 480 days and 96 savable at age 4, two get 240 and 48 each. Workspaces are capped at two parents. Also lands the in-flight squash of migrations 002–011 into the initial schema and the removal of planning_mode. Co-Authored-By: Claude Fable 5.1 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,134 @@
|
||||
"""Add 'deduced' to parental_leave_fetch.source's CHECK constraint.
|
||||
|
||||
`source` gains a third value alongside 'scraped'/'manual': 'deduced', for a
|
||||
snapshot the app derived on the user's behalf (e.g. defaulting an unentered
|
||||
total) rather than one FK reported or the user typed in directly. SQLite has
|
||||
no `ALTER TABLE ... DROP/ADD CONSTRAINT`, so the table is recreated with the
|
||||
new CHECK and its rows copied across — the standard SQLite migration dance
|
||||
(see e.g. https://sqlite.org/lang_altertable.html#otheralter). Recreating the
|
||||
table also drops its indexes, so they're recreated afterwards.
|
||||
|
||||
Must stay byte-for-byte equivalent, CHECK-constraint-wise, to
|
||||
`parental_leave_fetch` in `app/db/connection.py`'s `_FULL_SCHEMA` —
|
||||
`tests/integration/test_schema_parity.py` asserts exactly that.
|
||||
|
||||
Revision ID: 002
|
||||
Revises: 001
|
||||
Create Date: 2026-09-19
|
||||
"""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from alembic import op
|
||||
|
||||
revision: str = "002"
|
||||
down_revision: str | None = "001"
|
||||
branch_labels: str | None = None
|
||||
depends_on: str | None = None
|
||||
|
||||
_UPGRADE_SQL = """
|
||||
CREATE TABLE parental_leave_fetch_new (
|
||||
id TEXT PRIMARY KEY,
|
||||
personal_data_uuid TEXT NOT NULL,
|
||||
workspace_id TEXT NOT NULL,
|
||||
child_id TEXT NOT NULL,
|
||||
source TEXT NOT NULL CHECK (source IN ('scraped','manual','deduced')),
|
||||
fetched_at TIMESTAMP NOT NULL,
|
||||
total_days INTEGER NOT NULL,
|
||||
days_used INTEGER NOT NULL,
|
||||
days_left INTEGER NOT NULL,
|
||||
reserved_days_total INTEGER NOT NULL DEFAULT 90,
|
||||
reserved_days_left INTEGER NOT NULL,
|
||||
transferable_days_left INTEGER NOT NULL,
|
||||
dubbeldagar_left INTEGER NOT NULL,
|
||||
tier_data TEXT NOT NULL,
|
||||
created_at TIMESTAMP NOT NULL,
|
||||
updated_at TIMESTAMP NOT NULL,
|
||||
FOREIGN KEY (personal_data_uuid) REFERENCES personal_data(uuid) ON DELETE CASCADE,
|
||||
FOREIGN KEY (workspace_id) REFERENCES workspace(id) ON DELETE CASCADE,
|
||||
FOREIGN KEY (child_id) REFERENCES child(id) ON DELETE CASCADE
|
||||
);
|
||||
|
||||
INSERT INTO parental_leave_fetch_new (
|
||||
id, personal_data_uuid, workspace_id, child_id, source, fetched_at,
|
||||
total_days, days_used, days_left, reserved_days_total, reserved_days_left,
|
||||
transferable_days_left, dubbeldagar_left, tier_data, created_at, updated_at
|
||||
)
|
||||
SELECT
|
||||
id, personal_data_uuid, workspace_id, child_id, source, fetched_at,
|
||||
total_days, days_used, days_left, reserved_days_total, reserved_days_left,
|
||||
transferable_days_left, dubbeldagar_left, tier_data, created_at, updated_at
|
||||
FROM parental_leave_fetch;
|
||||
|
||||
DROP TABLE parental_leave_fetch;
|
||||
|
||||
ALTER TABLE parental_leave_fetch_new RENAME TO parental_leave_fetch;
|
||||
|
||||
CREATE INDEX IF NOT EXISTS idx_plf_parent_child_fetched
|
||||
ON parental_leave_fetch (personal_data_uuid, child_id, fetched_at);
|
||||
CREATE INDEX IF NOT EXISTS idx_plf_workspace
|
||||
ON parental_leave_fetch (workspace_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_plf_child
|
||||
ON parental_leave_fetch (child_id);
|
||||
"""
|
||||
|
||||
_DOWNGRADE_SQL = """
|
||||
CREATE TABLE parental_leave_fetch_old (
|
||||
id TEXT PRIMARY KEY,
|
||||
personal_data_uuid TEXT NOT NULL,
|
||||
workspace_id TEXT NOT NULL,
|
||||
child_id TEXT NOT NULL,
|
||||
source TEXT NOT NULL CHECK (source IN ('scraped','manual')),
|
||||
fetched_at TIMESTAMP NOT NULL,
|
||||
total_days INTEGER NOT NULL,
|
||||
days_used INTEGER NOT NULL,
|
||||
days_left INTEGER NOT NULL,
|
||||
reserved_days_total INTEGER NOT NULL DEFAULT 90,
|
||||
reserved_days_left INTEGER NOT NULL,
|
||||
transferable_days_left INTEGER NOT NULL,
|
||||
dubbeldagar_left INTEGER NOT NULL,
|
||||
tier_data TEXT NOT NULL,
|
||||
created_at TIMESTAMP NOT NULL,
|
||||
updated_at TIMESTAMP NOT NULL,
|
||||
FOREIGN KEY (personal_data_uuid) REFERENCES personal_data(uuid) ON DELETE CASCADE,
|
||||
FOREIGN KEY (workspace_id) REFERENCES workspace(id) ON DELETE CASCADE,
|
||||
FOREIGN KEY (child_id) REFERENCES child(id) ON DELETE CASCADE
|
||||
);
|
||||
|
||||
INSERT INTO parental_leave_fetch_old (
|
||||
id, personal_data_uuid, workspace_id, child_id, source, fetched_at,
|
||||
total_days, days_used, days_left, reserved_days_total, reserved_days_left,
|
||||
transferable_days_left, dubbeldagar_left, tier_data, created_at, updated_at
|
||||
)
|
||||
SELECT
|
||||
id, personal_data_uuid, workspace_id, child_id, source, fetched_at,
|
||||
total_days, days_used, days_left, reserved_days_total, reserved_days_left,
|
||||
transferable_days_left, dubbeldagar_left, tier_data, created_at, updated_at
|
||||
FROM parental_leave_fetch
|
||||
WHERE source != 'deduced';
|
||||
|
||||
DROP TABLE parental_leave_fetch;
|
||||
|
||||
ALTER TABLE parental_leave_fetch_old RENAME TO parental_leave_fetch;
|
||||
|
||||
CREATE INDEX IF NOT EXISTS idx_plf_parent_child_fetched
|
||||
ON parental_leave_fetch (personal_data_uuid, child_id, fetched_at);
|
||||
CREATE INDEX IF NOT EXISTS idx_plf_workspace
|
||||
ON parental_leave_fetch (workspace_id);
|
||||
CREATE INDEX IF NOT EXISTS idx_plf_child
|
||||
ON parental_leave_fetch (child_id);
|
||||
"""
|
||||
|
||||
|
||||
def upgrade() -> None:
|
||||
from sqlalchemy import text
|
||||
|
||||
for stmt in [s.strip() for s in _UPGRADE_SQL.split(";") if s.strip()]:
|
||||
op.execute(text(stmt))
|
||||
|
||||
|
||||
def downgrade() -> None:
|
||||
from sqlalchemy import text
|
||||
|
||||
for stmt in [s.strip() for s in _DOWNGRADE_SQL.split(";") if s.strip()]:
|
||||
op.execute(text(stmt))
|
||||
Reference in New Issue
Block a user