"""Add LOCAL status to PPR records

Revision ID: 011_ppr_local_status
Revises: 010_drone_request_agl_altitude
Create Date: 2026-06-29 00:00:00.000000

"""
from alembic import op

revision = '011_ppr_local_status'
down_revision = '010_drone_request_agl_altitude'
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.execute("ALTER TABLE submitted CHANGE COLUMN departed_dt takeoff_dt DATETIME NULL")
    op.execute("ALTER TABLE submitted ADD COLUMN qsy_dt DATETIME NULL AFTER takeoff_dt")
    op.execute(
        "ALTER TABLE submitted MODIFY COLUMN status "
        "ENUM('NEW','CONFIRMED','CANCELED','LANDED','LOCAL','DELETED','DEPARTED','ACTIVATED') "
        "NOT NULL DEFAULT 'NEW'"
    )


def downgrade() -> None:
    op.execute("UPDATE submitted SET status = 'LANDED' WHERE status = 'LOCAL'")
    op.execute("ALTER TABLE submitted DROP COLUMN qsy_dt")
    op.execute("ALTER TABLE submitted CHANGE COLUMN takeoff_dt departed_dt DATETIME NULL")
    op.execute(
        "ALTER TABLE submitted MODIFY COLUMN status "
        "ENUM('NEW','CONFIRMED','CANCELED','LANDED','DELETED','DEPARTED','ACTIVATED') "
        "NOT NULL DEFAULT 'NEW'"
    )