92 lines
3.7 KiB
Python
92 lines
3.7 KiB
Python
"""Upload batches and their items (US05-02).
|
|
|
|
Revision ID: 0008_upload_batches
|
|
Revises: 0007_rename_plans
|
|
Create Date: 2026-08-16
|
|
|
|
One row per approved album folder handed to immich-go, plus the per-asset
|
|
pre-upload hashes that later stories verify the result against.
|
|
"""
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
|
|
revision = "0008_upload_batches"
|
|
down_revision = "0007_rename_plans"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"upload_batches",
|
|
sa.Column("id", sa.String(), primary_key=True),
|
|
sa.Column("album", sa.String(), nullable=False),
|
|
sa.Column("folder", sa.String(), nullable=False),
|
|
sa.Column("album_name", sa.String(), nullable=False),
|
|
# planned | running | cancelling | cancelled | succeeded | failed
|
|
# | unknown_requires_verification
|
|
sa.Column("state", sa.String(), nullable=False, server_default="planned"),
|
|
# Preflight token this batch was approved against; re-checked before every
|
|
# attempt, so changed bytes or decisions cannot be uploaded silently.
|
|
sa.Column("preflight_token", sa.String(), nullable=False),
|
|
sa.Column("allow_partial", sa.Boolean(), nullable=False, server_default="0"),
|
|
sa.Column("command", sa.String(), nullable=True), # JSON array, redacted
|
|
sa.Column("uploader_version", sa.String(), nullable=True),
|
|
sa.Column("asset_count", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("attempt_count", sa.Integer(), nullable=False, server_default="0"),
|
|
# Bumped on every claim and used as the fencing token.
|
|
sa.Column("version", sa.Integer(), nullable=False, server_default="1"),
|
|
sa.Column("worker_id", sa.String(), nullable=True),
|
|
sa.Column("report_path", sa.String(), nullable=True),
|
|
sa.Column("report_bytes", sa.Integer(), nullable=True),
|
|
sa.Column("report_truncated", sa.Boolean(), nullable=False, server_default="0"),
|
|
sa.Column("exit_code", sa.Integer(), nullable=True),
|
|
sa.Column("error_code", sa.String(), nullable=True),
|
|
sa.Column("error_message", sa.String(), nullable=True),
|
|
sa.Column("started_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("finished_at", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column(
|
|
"created_at",
|
|
sa.DateTime(timezone=True),
|
|
nullable=False,
|
|
server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
nullable=False,
|
|
server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
),
|
|
)
|
|
op.create_index("ix_upload_batches_album", "upload_batches", ["album"])
|
|
op.create_index("ix_upload_batches_state", "upload_batches", ["state"])
|
|
|
|
op.create_table(
|
|
"upload_items",
|
|
sa.Column(
|
|
"batch_id",
|
|
sa.String(),
|
|
sa.ForeignKey("upload_batches.id", ondelete="CASCADE"),
|
|
primary_key=True,
|
|
),
|
|
sa.Column("asset_id", sa.String(), primary_key=True),
|
|
sa.Column("path", sa.String(), nullable=False),
|
|
# The bytes as they were when the batch was created; SHA-1 is what Immich
|
|
# uses to recognise a file it already holds.
|
|
sa.Column("sha256", sa.String(), nullable=True),
|
|
sa.Column("sha1", sa.String(), nullable=True),
|
|
sa.Column("state", sa.String(), nullable=False, server_default="pending"),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
nullable=False,
|
|
server_default=sa.text("CURRENT_TIMESTAMP"),
|
|
),
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("upload_items")
|
|
op.drop_table("upload_batches")
|