"""csv upload

Revision ID: b5f98ce6698e
Revises: 29b90a51efaf
Create Date: 2026-01-20 11:50:08.945906

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


# revision identifiers, used by Alembic.
revision: str = 'b5f98ce6698e'
down_revision: Union[str, None] = '29b90a51efaf'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.create_table('csv_upload_logs',
    sa.Column('id', sa.Integer(), autoincrement=True, nullable=False),
    sa.Column('company_id', sa.Integer(), nullable=False),
    sa.Column('upload_date', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    sa.Column('csv_type', sa.String(length=50), nullable=False),
    sa.Column('processing_status', sa.String(length=50), nullable=False),
    sa.Column('file_path', sa.String(length=500), nullable=False),
    sa.Column('error_message', sa.String(length=2000), nullable=True),
    sa.Column('rows_processed', sa.Integer(), nullable=True),
    sa.Column('rows_total', sa.Integer(), nullable=True),
    sa.Column('created_at', sa.DateTime(timezone=True), server_default=sa.text('now()'), nullable=False),
    sa.ForeignKeyConstraint(['company_id'], ['companies.id'], ),
    sa.PrimaryKeyConstraint('id')
    )
    op.create_index(op.f('ix_csv_upload_logs_company_id'), 'csv_upload_logs', ['company_id'], unique=False)
    op.create_index(op.f('ix_csv_upload_logs_csv_type'), 'csv_upload_logs', ['csv_type'], unique=False)
    op.create_index(op.f('ix_csv_upload_logs_id'), 'csv_upload_logs', ['id'], unique=False)
    op.create_index(op.f('ix_csv_upload_logs_processing_status'), 'csv_upload_logs', ['processing_status'], unique=False)
    op.create_index(op.f('ix_csv_upload_logs_upload_date'), 'csv_upload_logs', ['upload_date'], unique=False)
    op.add_column('celery_task_tracker', sa.Column('type', sa.String(length=50), nullable=True))
    op.add_column('celery_task_tracker', sa.Column('progress', sa.String(length=255), nullable=True))
    op.add_column('celery_task_tracker', sa.Column('file_path', sa.String(length=500), nullable=True))
    op.add_column('celery_task_tracker', sa.Column('upload_type', sa.String(length=50), nullable=True))
    op.create_index(op.f('ix_celery_task_tracker_upload_type'), 'celery_task_tracker', ['upload_type'], unique=False)
    # ### end Alembic commands ###


def downgrade() -> None:
    # ### commands auto generated by Alembic - please adjust! ###
    op.drop_index(op.f('ix_celery_task_tracker_upload_type'), table_name='celery_task_tracker')
    op.drop_column('celery_task_tracker', 'upload_type')
    op.drop_column('celery_task_tracker', 'file_path')
    op.drop_column('celery_task_tracker', 'progress')
    op.drop_column('celery_task_tracker', 'type')
    op.drop_index(op.f('ix_csv_upload_logs_upload_date'), table_name='csv_upload_logs')
    op.drop_index(op.f('ix_csv_upload_logs_processing_status'), table_name='csv_upload_logs')
    op.drop_index(op.f('ix_csv_upload_logs_id'), table_name='csv_upload_logs')
    op.drop_index(op.f('ix_csv_upload_logs_csv_type'), table_name='csv_upload_logs')
    op.drop_index(op.f('ix_csv_upload_logs_company_id'), table_name='csv_upload_logs')
    op.drop_table('csv_upload_logs')
    # ### end Alembic commands ###