"""make policy zone ids nullable (support 'all')

Revision ID: 0007
Revises: 0006
Create Date: 2026-03-01
"""
from alembic import op
import sqlalchemy as sa

revision = "0007"
down_revision = "0006"
branch_labels = None
depends_on = None


def upgrade() -> None:
    op.alter_column("policies", "src_zone_id", existing_type=sa.Integer(), nullable=True)
    op.alter_column("policies", "dst_zone_id", existing_type=sa.Integer(), nullable=True)


def downgrade() -> None:
    op.alter_column("policies", "src_zone_id", existing_type=sa.Integer(), nullable=False)
    op.alter_column("policies", "dst_zone_id", existing_type=sa.Integer(), nullable=False)