A R R R Associates
111 lines
6.5 KiB
Python
111 lines
6.5 KiB
Python
"""Phase 6D employee leave management
|
|
|
|
Revision ID: 20260516_phase_6d_employee_leave
|
|
Revises: 20260515_phase_6c_employee_attendance
|
|
Create Date: 2026-05-10
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from alembic import op
|
|
import sqlalchemy as sa
|
|
|
|
revision = "20260516_phase_6d_employee_leave"
|
|
down_revision = "20260515_phase_6c_employee_attendance"
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade() -> None:
|
|
op.create_table(
|
|
"employee_leave_types",
|
|
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column("tenant_id", sa.Integer(), nullable=False),
|
|
sa.Column("branch_id", sa.Integer(), nullable=False),
|
|
sa.Column("code", sa.String(length=30), nullable=False),
|
|
sa.Column("name", sa.String(length=100), nullable=False),
|
|
sa.Column("description", sa.Text(), nullable=True),
|
|
sa.Column("annual_quota_days", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("carry_forward_allowed", sa.Boolean(), nullable=False, server_default=sa.false()),
|
|
sa.Column("allow_negative_balance", sa.Boolean(), nullable=False, server_default=sa.false()),
|
|
sa.Column("requires_approval", sa.Boolean(), nullable=False, server_default=sa.true()),
|
|
sa.Column("is_paid", sa.Boolean(), nullable=False, server_default=sa.true()),
|
|
sa.Column("is_active", sa.Boolean(), nullable=False, server_default=sa.true()),
|
|
sa.Column("created_by_user_id", sa.Integer(), nullable=True),
|
|
sa.Column("updated_by_user_id", sa.Integer(), nullable=True),
|
|
sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at_utc", sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(["branch_id"], ["branches.id"]),
|
|
sa.ForeignKeyConstraint(["created_by_user_id"], ["users.id"], ondelete="SET NULL"),
|
|
sa.ForeignKeyConstraint(["tenant_id"], ["tenants.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["updated_by_user_id"], ["users.id"], ondelete="SET NULL"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("tenant_id", "branch_id", "code", name="uq_employee_leave_types_tenant_branch_code"),
|
|
)
|
|
for col in ["tenant_id", "branch_id", "code", "is_active"]:
|
|
op.create_index(f"ix_employee_leave_types_{col}", "employee_leave_types", [col], unique=False)
|
|
|
|
op.create_table(
|
|
"employee_leave_balances",
|
|
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column("tenant_id", sa.Integer(), nullable=False),
|
|
sa.Column("branch_id", sa.Integer(), nullable=False),
|
|
sa.Column("employee_id", sa.Integer(), nullable=False),
|
|
sa.Column("leave_type_id", sa.Integer(), nullable=False),
|
|
sa.Column("opening_days", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("credited_days", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("availed_days", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("adjusted_days", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("balance_days", sa.Integer(), nullable=False, server_default="0"),
|
|
sa.Column("updated_by_user_id", sa.Integer(), nullable=True),
|
|
sa.Column("updated_at_utc", sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(["branch_id"], ["branches.id"]),
|
|
sa.ForeignKeyConstraint(["employee_id"], ["employees.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["leave_type_id"], ["employee_leave_types.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["tenant_id"], ["tenants.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["updated_by_user_id"], ["users.id"], ondelete="SET NULL"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
sa.UniqueConstraint("tenant_id", "employee_id", "leave_type_id", name="uq_employee_leave_balances_employee_type"),
|
|
)
|
|
for col in ["tenant_id", "branch_id", "employee_id", "leave_type_id"]:
|
|
op.create_index(f"ix_employee_leave_balances_{col}", "employee_leave_balances", [col], unique=False)
|
|
|
|
op.create_table(
|
|
"employee_leave_requests",
|
|
sa.Column("id", sa.Integer(), autoincrement=True, nullable=False),
|
|
sa.Column("tenant_id", sa.Integer(), nullable=False),
|
|
sa.Column("branch_id", sa.Integer(), nullable=False),
|
|
sa.Column("employee_id", sa.Integer(), nullable=False),
|
|
sa.Column("user_id", sa.Integer(), nullable=True),
|
|
sa.Column("leave_type_id", sa.Integer(), nullable=False),
|
|
sa.Column("from_date", sa.Date(), nullable=False),
|
|
sa.Column("to_date", sa.Date(), nullable=False),
|
|
sa.Column("days", sa.Integer(), nullable=False, server_default="1"),
|
|
sa.Column("reason", sa.Text(), nullable=True),
|
|
sa.Column("status", sa.String(length=30), nullable=False, server_default="pending"),
|
|
sa.Column("request_source", sa.String(length=30), nullable=False, server_default="employee_portal"),
|
|
sa.Column("reviewed_by_user_id", sa.Integer(), nullable=True),
|
|
sa.Column("reviewed_at_utc", sa.DateTime(timezone=True), nullable=True),
|
|
sa.Column("review_notes", sa.Text(), nullable=True),
|
|
sa.Column("created_by_user_id", sa.Integer(), nullable=True),
|
|
sa.Column("updated_by_user_id", sa.Integer(), nullable=True),
|
|
sa.Column("created_at_utc", sa.DateTime(timezone=True), nullable=False),
|
|
sa.Column("updated_at_utc", sa.DateTime(timezone=True), nullable=False),
|
|
sa.ForeignKeyConstraint(["branch_id"], ["branches.id"]),
|
|
sa.ForeignKeyConstraint(["created_by_user_id"], ["users.id"], ondelete="SET NULL"),
|
|
sa.ForeignKeyConstraint(["employee_id"], ["employees.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["leave_type_id"], ["employee_leave_types.id"], ondelete="RESTRICT"),
|
|
sa.ForeignKeyConstraint(["reviewed_by_user_id"], ["users.id"], ondelete="SET NULL"),
|
|
sa.ForeignKeyConstraint(["tenant_id"], ["tenants.id"], ondelete="CASCADE"),
|
|
sa.ForeignKeyConstraint(["updated_by_user_id"], ["users.id"], ondelete="SET NULL"),
|
|
sa.ForeignKeyConstraint(["user_id"], ["users.id"], ondelete="SET NULL"),
|
|
sa.PrimaryKeyConstraint("id"),
|
|
)
|
|
for col in ["tenant_id", "branch_id", "employee_id", "user_id", "leave_type_id", "from_date", "to_date", "status", "request_source"]:
|
|
op.create_index(f"ix_employee_leave_requests_{col}", "employee_leave_requests", [col], unique=False)
|
|
|
|
|
|
def downgrade() -> None:
|
|
op.drop_table("employee_leave_requests")
|
|
op.drop_table("employee_leave_balances")
|
|
op.drop_table("employee_leave_types")
|