Files
gh-christianlouis-dmarq/backend/alembic/versions/e2f3a4b5c6d7_add_dmarcbis_aggregate_fields.py
T
2026-05-23 14:17:09 +02:00

55 lines
2.6 KiB
Python

"""add dmarcbis aggregate fields
Revision ID: e2f3a4b5c6d7
Revises: d1e2f3a4b5c6
Create Date: 2026-05-23 00:00:00.000000
"""
from typing import Sequence, Union
import sqlalchemy as sa
from alembic import op
# revision identifiers, used by Alembic.
revision: str = "e2f3a4b5c6d7"
down_revision: Union[str, Sequence[str], None] = "d1e2f3a4b5c6"
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None
def upgrade() -> None:
"""Persist optional RFC 9990 / DMARCbis aggregate metadata."""
op.add_column("dmarc_reports", sa.Column("extra_contact_info", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("generator", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("report_errors", sa.Text(), nullable=True))
op.add_column("dmarc_reports", sa.Column("non_subdomain_policy", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("failure_options", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("testing", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("discovery_method", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("schema_version", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("report_variant", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("xml_namespace", sa.String(), nullable=True))
op.add_column("dmarc_reports", sa.Column("report_extensions", sa.Text(), nullable=True))
op.add_column("report_records", sa.Column("envelope_to", sa.String(), nullable=True))
op.add_column("report_records", sa.Column("policy_override_reasons", sa.Text(), nullable=True))
op.add_column("report_records", sa.Column("record_extensions", sa.Text(), nullable=True))
def downgrade() -> None:
"""Remove optional RFC 9990 / DMARCbis aggregate metadata."""
op.drop_column("report_records", "record_extensions")
op.drop_column("report_records", "policy_override_reasons")
op.drop_column("report_records", "envelope_to")
op.drop_column("dmarc_reports", "report_extensions")
op.drop_column("dmarc_reports", "xml_namespace")
op.drop_column("dmarc_reports", "report_variant")
op.drop_column("dmarc_reports", "schema_version")
op.drop_column("dmarc_reports", "discovery_method")
op.drop_column("dmarc_reports", "testing")
op.drop_column("dmarc_reports", "failure_options")
op.drop_column("dmarc_reports", "non_subdomain_policy")
op.drop_column("dmarc_reports", "report_errors")
op.drop_column("dmarc_reports", "generator")
op.drop_column("dmarc_reports", "extra_contact_info")