api/router/va_lmd_hr.py

from fastapi import APIRouter
from uuid import uuid4
from datetime import datetime

from models.request_models import (
    LeaveRequest,
    PermitRequest,
    PresenceRequest
)

from data.dummy_data import (
    leave_db,
    permit_db,
    presence_db
)

router = APIRouter()


# =========================
# LEAVE
# =========================
@router.get("/leave")
def get_leave():
    return leave_db


@router.post("/leave")
def create_leave(data: LeaveRequest):
    item = {
        "id": str(uuid4()),
        **data.dict(),
        "created_at": datetime.now()
    }
    leave_db.append(item)
    return {
        "message": "Leave created successfully",
        "data": item
    }


# =========================
# PERMIT
# =========================
@router.get("/permit")
def get_permit():
    return permit_db


@router.post("/permit")
def create_permit(data: PermitRequest):
    item = {
        "id": str(uuid4()),
        **data.dict(),
        "created_at": datetime.now()
    }
    permit_db.append(item)
    return {
        "message": "Permit created successfully",
        "data": item
    }


# =========================
# PRESENCE
# =========================
@router.get("/presence")
def get_presence():
    return presence_db


@router.post("/presence")
def create_presence(data: PresenceRequest):
    item = {
        "id": str(uuid4()),
        **data.dict(),
        "created_at": datetime.now()
    }
    presence_db.append(item)
    return {
        "message": "Presence created successfully",
        "data": item
    }