98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
"""folders and capture policy
|
|
|
|
Revision ID: b2f7c1a9d3e4
|
|
Revises: 77df960a31de
|
|
Create Date: 2026-05-29 17:30:00.000000
|
|
|
|
"""
|
|
|
|
from collections.abc import Sequence
|
|
|
|
import sqlalchemy as sa
|
|
from alembic import op
|
|
from sqlalchemy.dialects import postgresql
|
|
|
|
revision: str = "b2f7c1a9d3e4"
|
|
down_revision: str | None = "77df960a31de"
|
|
branch_labels: str | Sequence[str] | None = None
|
|
depends_on: str | Sequence[str] | None = None
|
|
|
|
_TOGGLES = (
|
|
"messages",
|
|
"media",
|
|
"self_destruct_media",
|
|
"stt",
|
|
"reactions",
|
|
"track_edits_deletes",
|
|
"profile_history",
|
|
"stories",
|
|
"presence",
|
|
"backfill",
|
|
)
|
|
|
|
|
|
def _toggle_columns() -> list[sa.Column]:
|
|
return [
|
|
sa.Column(name, sa.Boolean(), nullable=False, server_default=sa.false())
|
|
for name in _TOGGLES
|
|
]
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"folders",
|
|
sa.Column("account_id", sa.Integer(), nullable=False),
|
|
sa.Column("folder_id", sa.Integer(), nullable=False),
|
|
sa.Column("title", sa.String(), nullable=False),
|
|
sa.Column("order_index", sa.Integer(), nullable=False),
|
|
sa.Column("is_chatlist", sa.Boolean(), nullable=False),
|
|
sa.Column("raw", postgresql.JSONB(astext_type=sa.Text()), nullable=False),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("now()"),
|
|
nullable=False,
|
|
),
|
|
sa.PrimaryKeyConstraint("account_id", "folder_id"),
|
|
)
|
|
op.create_table(
|
|
"capture_policy",
|
|
sa.Column("id", sa.Integer(), nullable=False),
|
|
sa.Column("account_id", sa.Integer(), nullable=True),
|
|
sa.Column("scope_type", sa.String(), nullable=False),
|
|
sa.Column("scope_id", sa.BigInteger(), nullable=True),
|
|
*_toggle_columns(),
|
|
sa.Column(
|
|
"created_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("now()"),
|
|
nullable=False,
|
|
),
|
|
sa.Column(
|
|
"updated_at",
|
|
sa.DateTime(timezone=True),
|
|
server_default=sa.text("now()"),
|
|
nullable=False,
|
|
),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("account_id", "scope_type", "scope_id"),
|
|
)
|
|
op.execute(
|
|
"CREATE UNIQUE INDEX ix_capture_policy_default "
|
|
"ON capture_policy (scope_type) WHERE scope_type LIKE 'default_%'"
|
|
)
|
|
|
|
op.execute(
|
|
"INSERT INTO capture_policy "
|
|
"(scope_type, messages, media, self_destruct_media, stt, reactions, "
|
|
"track_edits_deletes, profile_history, stories, presence, backfill) VALUES "
|
|
"('default_channel', FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE), "
|
|
"('default_group', TRUE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE, FALSE), "
|
|
"('default_dm', TRUE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE, TRUE)"
|
|
)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("capture_policy")
|
|
op.drop_table("folders")
|