mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-04 20:05:16 +02:00
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
"""add document_files table for stored original uploads
|
|
|
|
Revision ID: 152
|
|
Revises: 151
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
from alembic import op
|
|
|
|
revision: str = "152"
|
|
down_revision: str | None = "151"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
# The enum type must precede the table that references it.
|
|
op.execute(
|
|
"""
|
|
DO $$
|
|
BEGIN
|
|
IF NOT EXISTS (
|
|
SELECT 1 FROM pg_type WHERE typname = 'document_file_kind'
|
|
) THEN
|
|
CREATE TYPE document_file_kind AS ENUM (
|
|
'ORIGINAL', 'REDACTED', 'FILLED_FORM'
|
|
);
|
|
END IF;
|
|
END
|
|
$$;
|
|
"""
|
|
)
|
|
|
|
op.execute(
|
|
"""
|
|
CREATE TABLE IF NOT EXISTS document_files (
|
|
id SERIAL PRIMARY KEY,
|
|
document_id INTEGER NOT NULL
|
|
REFERENCES documents(id) ON DELETE CASCADE,
|
|
search_space_id INTEGER NOT NULL
|
|
REFERENCES searchspaces(id) ON DELETE CASCADE,
|
|
kind document_file_kind NOT NULL DEFAULT 'ORIGINAL',
|
|
storage_backend VARCHAR(32) NOT NULL,
|
|
storage_key TEXT NOT NULL,
|
|
original_filename TEXT NOT NULL,
|
|
mime_type TEXT,
|
|
size_bytes BIGINT NOT NULL,
|
|
checksum_sha256 VARCHAR(64),
|
|
created_by_id UUID
|
|
REFERENCES "user"(id) ON DELETE SET NULL,
|
|
created_at TIMESTAMP WITH TIME ZONE NOT NULL DEFAULT NOW()
|
|
);
|
|
"""
|
|
)
|
|
|
|
op.execute(
|
|
"CREATE INDEX IF NOT EXISTS ix_document_files_document_id "
|
|
"ON document_files(document_id);"
|
|
)
|
|
op.execute(
|
|
"CREATE INDEX IF NOT EXISTS ix_document_files_search_space_id "
|
|
"ON document_files(search_space_id);"
|
|
)
|
|
op.execute(
|
|
"CREATE INDEX IF NOT EXISTS ix_document_files_kind "
|
|
"ON document_files(kind);"
|
|
)
|
|
op.execute(
|
|
"CREATE INDEX IF NOT EXISTS ix_document_files_created_by_id "
|
|
"ON document_files(created_by_id);"
|
|
)
|
|
op.execute(
|
|
"CREATE INDEX IF NOT EXISTS ix_document_files_created_at "
|
|
"ON document_files(created_at);"
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.execute("DROP INDEX IF EXISTS ix_document_files_created_at;")
|
|
op.execute("DROP INDEX IF EXISTS ix_document_files_created_by_id;")
|
|
op.execute("DROP INDEX IF EXISTS ix_document_files_kind;")
|
|
op.execute("DROP INDEX IF EXISTS ix_document_files_search_space_id;")
|
|
op.execute("DROP INDEX IF EXISTS ix_document_files_document_id;")
|
|
op.execute("DROP TABLE IF EXISTS document_files;")
|
|
op.execute("DROP TYPE IF EXISTS document_file_kind;")
|