62 lines
3.0 KiB
Python
62 lines
3.0 KiB
Python
"""create scan ranges table
|
|
|
|
Revision ID: 0003_scan_ranges
|
|
Revises: 0002_audit_logs
|
|
Create Date: 2026-07-08
|
|
"""
|
|
|
|
from typing import Sequence, Union
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
|
|
revision: str = "0003_scan_ranges"
|
|
down_revision: Union[str, None] = "0002_audit_logs"
|
|
branch_labels: Union[str, Sequence[str], None] = None
|
|
depends_on: Union[str, Sequence[str], None] = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"scan_ranges",
|
|
sa.Column("id", postgresql.UUID(as_uuid=True), primary_key=True, server_default=sa.text("gen_random_uuid()")),
|
|
sa.Column("room_id", postgresql.UUID(as_uuid=True), sa.ForeignKey("rooms.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("name", sa.Text(), nullable=False),
|
|
sa.Column("range_type", sa.Text(), nullable=False, server_default=sa.text("'cidr'")),
|
|
sa.Column("cidr", sa.Text(), nullable=True),
|
|
sa.Column("start_ip", postgresql.INET(), nullable=True),
|
|
sa.Column("end_ip", postgresql.INET(), nullable=True),
|
|
sa.Column("enabled", sa.Boolean(), nullable=False, server_default=sa.text("true")),
|
|
sa.Column("scan_interval_seconds", sa.Integer(), nullable=False, server_default=sa.text("900")),
|
|
sa.Column("timeout_ms", sa.Integer(), nullable=False, server_default=sa.text("3000")),
|
|
sa.Column("retry_limit", sa.Integer(), nullable=False, server_default=sa.text("1")),
|
|
sa.Column("parallelism_limit", sa.Integer(), nullable=False, server_default=sa.text("10")),
|
|
sa.Column("stagger_ms", sa.Integer(), nullable=False, server_default=sa.text("250")),
|
|
sa.Column("api_port", sa.Integer(), nullable=False, server_default=sa.text("4028")),
|
|
sa.Column("last_scan_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("next_scan_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("created_by_user_id", postgresql.UUID(as_uuid=True), sa.ForeignKey("users.id", ondelete="SET NULL"), nullable=True),
|
|
sa.Column("created_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
|
|
sa.Column("updated_at", sa.DateTime(timezone=True), nullable=False, server_default=sa.text("now()")),
|
|
sa.CheckConstraint(
|
|
"""
|
|
(range_type = 'cidr' AND cidr IS NOT NULL)
|
|
OR
|
|
(range_type = 'start_end' AND start_ip IS NOT NULL AND end_ip IS NOT NULL)
|
|
""",
|
|
name="scan_range_has_valid_range",
|
|
),
|
|
)
|
|
|
|
op.create_index("idx_scan_ranges_room_id", "scan_ranges", ["room_id"])
|
|
op.create_index("idx_scan_ranges_enabled", "scan_ranges", ["enabled"])
|
|
op.create_index("idx_scan_ranges_next_scan_at", "scan_ranges", ["next_scan_at"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("idx_scan_ranges_next_scan_at", table_name="scan_ranges")
|
|
op.drop_index("idx_scan_ranges_enabled", table_name="scan_ranges")
|
|
op.drop_index("idx_scan_ranges_room_id", table_name="scan_ranges")
|
|
op.drop_table("scan_ranges") |