"""add payroll_runs and payroll_run_items

Revision ID: 002749047f04
Revises: 4fa9726b4e7c
Create Date: 2026-03-10 08:49:08.053868

"""
from typing import Sequence, Union

from alembic import op
import sqlalchemy as sa


revision: str = '002749047f04'
down_revision: Union[str, None] = '4fa9726b4e7c'
branch_labels: Union[str, Sequence[str], None] = None
depends_on: Union[str, Sequence[str], None] = None


def upgrade() -> None:
    op.create_table(
        'payroll_runs',
        sa.Column('id', sa.String(), nullable=False),
        sa.Column('run_date', sa.DateTime(timezone=True), nullable=False),
        sa.Column('month', sa.Integer(), nullable=False),
        sa.Column('year', sa.Integer(), nullable=False),
        sa.Column('export_type', sa.String(), nullable=False),
        sa.Column('company_id', sa.String(), nullable=True),
        sa.Column('company_name', sa.String(), nullable=True),
        sa.Column('total_amount', sa.Integer(), nullable=False),
        sa.Column('employee_count', sa.Integer(), nullable=False),
        sa.Column('recorded_transactions', sa.Boolean(), nullable=False),
        sa.Column('created_at', sa.DateTime(timezone=True), nullable=True),
        sa.ForeignKeyConstraint(['company_id'], ['companies.id'], ondelete='SET NULL'),
        sa.PrimaryKeyConstraint('id'),
    )
    op.create_table(
        'payroll_run_items',
        sa.Column('id', sa.String(), nullable=False),
        sa.Column('run_id', sa.String(), nullable=False),
        sa.Column('employee_id', sa.String(), nullable=True),
        sa.Column('employee_name', sa.String(), nullable=False),
        sa.Column('base_salary', sa.Integer(), nullable=False),
        sa.Column('loan_deduction', sa.Integer(), nullable=False),
        sa.Column('net_amount', sa.Integer(), nullable=False),
        sa.Column('has_override', sa.Boolean(), nullable=False),
        sa.Column('created_at', sa.DateTime(timezone=True), nullable=True),
        sa.ForeignKeyConstraint(['employee_id'], ['employees.id'], ondelete='SET NULL'),
        sa.ForeignKeyConstraint(['run_id'], ['payroll_runs.id'], ondelete='CASCADE'),
        sa.PrimaryKeyConstraint('id'),
    )


def downgrade() -> None:
    op.drop_table('payroll_run_items')
    op.drop_table('payroll_runs')