Files
arrr-erp/alembic/versions/20260520_phase_6k_geo_attendance.py
T
2026-06-20 15:01:44 +05:30

64 lines
3.9 KiB
Python

"""Phase 6K geo-fenced attendance and OD approval
Revision ID: 20260520_phase_6k_geo_attendance
Revises: 20260519_phase_6g_employee_payroll
Create Date: 2026-05-20
"""
from __future__ import annotations
from alembic import op
import sqlalchemy as sa
revision = "20260520_phase_6k_geo_attendance"
down_revision = "20260519_phase_6g_employee_payroll"
branch_labels = None
depends_on = None
def upgrade() -> None:
op.add_column("branch_settings", sa.Column("attendance_geo_enabled", sa.Boolean(), nullable=False, server_default=sa.false()))
op.add_column("branch_settings", sa.Column("attendance_geo_radius_meters", sa.Integer(), nullable=False, server_default="100"))
op.add_column("branch_settings", sa.Column("attendance_ip_enabled", sa.Boolean(), nullable=False, server_default=sa.false()))
op.add_column("branch_settings", sa.Column("attendance_allowed_ip_csv", sa.Text(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_in_latitude", sa.Float(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_in_longitude", sa.Float(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_in_accuracy_meters", sa.Float(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_in_distance_meters", sa.Float(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_in_ip", sa.String(length=80), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_in_geo_status", sa.String(length=40), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_in_ip_status", sa.String(length=40), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_out_latitude", sa.Float(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_out_longitude", sa.Float(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_out_accuracy_meters", sa.Float(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_out_distance_meters", sa.Float(), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_out_ip", sa.String(length=80), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_out_geo_status", sa.String(length=40), nullable=True))
op.add_column("employee_attendance", sa.Column("punch_out_ip_status", sa.String(length=40), nullable=True))
op.create_index("ix_employee_attendance_punch_in_geo_status", "employee_attendance", ["punch_in_geo_status"])
op.create_index("ix_employee_attendance_punch_in_ip_status", "employee_attendance", ["punch_in_ip_status"])
op.create_index("ix_employee_attendance_punch_out_geo_status", "employee_attendance", ["punch_out_geo_status"])
op.create_index("ix_employee_attendance_punch_out_ip_status", "employee_attendance", ["punch_out_ip_status"])
def downgrade() -> None:
op.drop_index("ix_employee_attendance_punch_out_ip_status", table_name="employee_attendance")
op.drop_index("ix_employee_attendance_punch_out_geo_status", table_name="employee_attendance")
op.drop_index("ix_employee_attendance_punch_in_ip_status", table_name="employee_attendance")
op.drop_index("ix_employee_attendance_punch_in_geo_status", table_name="employee_attendance")
for col in [
"punch_out_ip_status", "punch_out_geo_status", "punch_out_ip", "punch_out_distance_meters",
"punch_out_accuracy_meters", "punch_out_longitude", "punch_out_latitude",
"punch_in_ip_status", "punch_in_geo_status", "punch_in_ip", "punch_in_distance_meters",
"punch_in_accuracy_meters", "punch_in_longitude", "punch_in_latitude",
]:
op.drop_column("employee_attendance", col)
op.drop_column("branch_settings", "attendance_allowed_ip_csv")
op.drop_column("branch_settings", "attendance_ip_enabled")
op.drop_column("branch_settings", "attendance_geo_radius_meters")
op.drop_column("branch_settings", "attendance_geo_enabled")