mirror of
https://github.com/TronoSfera/Law.git
synced 2026-05-18 18:13:46 +03:00
43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
"""attachment antivirus scan status
|
|
|
|
Revision ID: 0030_attachment_scan
|
|
Revises: 0029_admin_totp
|
|
Create Date: 2026-03-01 18:10:00.000000
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = "0030_attachment_scan"
|
|
down_revision = "0029_admin_totp"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.add_column(
|
|
"attachments",
|
|
sa.Column("scan_status", sa.String(length=20), nullable=True, server_default="CLEAN"),
|
|
)
|
|
op.add_column("attachments", sa.Column("scan_signature", sa.String(length=255), nullable=True))
|
|
op.add_column("attachments", sa.Column("scan_error", sa.String(length=500), nullable=True))
|
|
op.add_column("attachments", sa.Column("scanned_at", sa.DateTime(timezone=True), nullable=True))
|
|
op.add_column("attachments", sa.Column("content_sha256", sa.String(length=64), nullable=True))
|
|
op.add_column("attachments", sa.Column("detected_mime", sa.String(length=150), nullable=True))
|
|
op.execute("UPDATE attachments SET scan_status = 'CLEAN' WHERE scan_status IS NULL")
|
|
op.alter_column("attachments", "scan_status", nullable=False, server_default=None)
|
|
op.create_index("ix_attachments_scan_status", "attachments", ["scan_status"])
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_index("ix_attachments_scan_status", table_name="attachments")
|
|
op.drop_column("attachments", "detected_mime")
|
|
op.drop_column("attachments", "content_sha256")
|
|
op.drop_column("attachments", "scanned_at")
|
|
op.drop_column("attachments", "scan_error")
|
|
op.drop_column("attachments", "scan_signature")
|
|
op.drop_column("attachments", "scan_status")
|