"""add max image and video count to video engines

Revision ID: f7a3b2c1d4e5
Revises: e5f260ed1459
Create Date: 2026-07-01 00:00:00.000000

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


revision: str = 'f7a3b2c1d4e5'
down_revision: Union[str, None] = 'e5f260ed1459'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.add_column('video_engines', sa.Column('max_image_count', sa.Integer(), server_default='2', nullable=False))
    op.add_column('video_engines', sa.Column('max_video_count', sa.Integer(), server_default='0', nullable=False))


def downgrade() -> None:
    op.drop_column('video_engines', 'max_video_count')
    op.drop_column('video_engines', 'max_image_count')