ERP_ams/controllers/leave.go

package controllers

import (
	"ERP_ams/Nats/NatsServer"
	"ERP_ams/conf"
	"ERP_ams/models/Account"
	"ERP_ams/models/Attendance"
	userlibs "git.baozhida.cn/ERP_libs/User"
	"git.baozhida.cn/ERP_libs/lib"
	beego "github.com/beego/beego/v2/server/web"
	"math"
	"strconv"
	"time"
)

type LeaveController struct {
	beego.Controller
	User userlibs.User
}

func (c *LeaveController) Prepare() {
	c.User = *Account.User_r
}

// 管理员请假审批列表 只显示待审核
func (c *LeaveController) Leave_List() {
	var r_jsons lib.R_JSONS

	page, _ := c.GetInt("page")

	if page < 1 {
		page = 1
	}
	page_z, _ := c.GetInt("page_z")
	if page_z < 1 {
		page_z = conf.Page_size
	}
	// 年月 2023-01
	T_month := c.GetString("T_month")
	if len(T_month) > 0 {
		_, err := time.Parse("2006-01", T_month)
		if err != nil {
			c.Data["json"] = lib.JSONS{Code: 202, Msg: "日期格式错误!"}
			c.ServeJSON()
			return
		}
	}

	userList, _ := NatsServer.Read_User_List_All()
	Account.Read_User_All_Map(userList)
	r_jsons.Data, r_jsons.Num = Attendance.Read_Attendance_List("", c.User.T_uuid, T_month, 0, Attendance.AttendanceWaitAudit, page, page_z)
	r_jsons.Page = page
	r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
	c.ServeJSON()
	return
}

func (c *LeaveController) Leave_User_List() {
	var r_jsons lib.R_JSONS

	page, _ := c.GetInt("page")

	if page < 1 {
		page = 1
	}
	page_z, _ := c.GetInt("page_z")
	if page_z < 1 {
		page_z = conf.Page_size
	}
	T_uuid := c.GetString("T_uuid")
	if len(T_uuid) == 0 {
		T_uuid = c.User.T_uuid
	}
	// 年月 2023-01
	T_month := c.GetString("T_month")

	if len(T_month) > 0 {
		_, err := time.Parse("2006-01", T_month)
		if err != nil {
			c.Data["json"] = lib.JSONS{Code: 202, Msg: "日期格式错误!"}
			c.ServeJSON()
			return
		}
	}

	r_jsons.Data, r_jsons.Num = Attendance.Read_Attendance_List(T_uuid, "", T_month, 0, 0, page, page_z)
	r_jsons.Page = page
	r_jsons.Page_size = int(math.Ceil(float64(r_jsons.Num) / float64(page_z)))

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: r_jsons}
	c.ServeJSON()
	return
}

func (c *LeaveController) Leave_Type_List() {
	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Attendance.Read_AttendanceType_All()}
	c.ServeJSON()
	return
}

func (c *LeaveController) Leave_Add() {
	T_type, _ := c.GetInt("T_type")
	T_start_time := c.GetString("T_start_time")
	T_end_time := c.GetString("T_end_time")
	T_duration, _ := c.GetInt("T_duration")
	T_text := c.GetString("T_text")
	T_approver := c.GetString("T_approver")
	startTime, is := lib.TimeStrToTime(T_start_time)
	if !is {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式不正确"}
		c.ServeJSON()
		return
	}
	endTime, is := lib.TimeStrToTime(T_end_time)
	if !is {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式不正确"}
		c.ServeJSON()
		return
	}
	var_ := Attendance.Attendance{
		T_uid:        c.User.T_uuid,
		T_type:       T_type,
		T_start_time: startTime,
		T_end_time:   endTime,
		T_duration:   T_duration,
		T_text:       T_text,
		T_approver:   T_approver,
		T_State:      Attendance.AttendanceWaitAudit,
	}

	Id, err := Attendance.Add_Attendance(var_)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "添加失败"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "我的请假", "添加", var_)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Id}
	c.ServeJSON()
	return
}

// 扣除
func (c *LeaveController) Leave_Deduct() {
	T_type, _ := c.GetInt("T_type")
	T_duration, _ := c.GetInt("T_duration")
	T_text := c.GetString("T_text")
	T_uuid := c.GetString("T_uuid")

	var_ := Attendance.Attendance{
		T_uid:      T_uuid,
		T_type:     T_type,
		T_duration: T_duration,
		T_text:     T_text,
		T_approver: c.User.T_uuid,
		T_State:    Attendance.AttendancePass,
	}

	Id, err := Attendance.Add_Attendance(var_)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "添加失败"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "我的请假", "扣除", var_)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: Id}
	c.ServeJSON()
	return
}

func (c *LeaveController) Leave_Update() {
	T_id, _ := c.GetInt("T_id")
	T_type, _ := c.GetInt("T_type")
	T_start_time := c.GetString("T_start_time")
	T_end_time := c.GetString("T_end_time")
	T_duration, _ := c.GetInt("T_duration")
	T_text := c.GetString("T_text")
	T_approver := c.GetString("T_approver")

	leave, err := Attendance.Read_Attendance_ById(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	if leave.T_State == Attendance.AttendancePass {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "无权修改!"}
		c.ServeJSON()
		return
	}

	if T_type > 0 {
		leave.T_type = T_type
	}

	if len(T_start_time) > 0 {
		startTime, is := lib.TimeStrToTime(T_start_time)
		if !is {
			c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式不正确"}
			c.ServeJSON()
			return
		}
		leave.T_start_time = startTime

	}
	if len(T_end_time) > 0 {
		endTime, is := lib.TimeStrToTime(T_end_time)
		if !is {
			c.Data["json"] = lib.JSONS{Code: 202, Msg: "时间格式不正确"}
			c.ServeJSON()
			return
		}
		leave.T_end_time = endTime
	}

	if T_duration > 0 {
		leave.T_duration = T_duration
	}

	if len(T_text) > 0 {
		leave.T_text = T_text
	}
	if len(T_approver) > 0 {
		leave.T_approver = T_approver
	}
	leave.T_State = Attendance.AttendanceWaitAudit

	_, err = Attendance.Update_Attendance(leave, "T_type", "T_start_time", "T_end_time", "T_duration", "T_text", "T_approver", "T_State")
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "添加失败"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "我的请假", "修改", leave)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: T_id}
	c.ServeJSON()
	return
}

func (c *LeaveController) Leave_Approval() {
	T_id, _ := c.GetInt("T_id")
	T_State, _ := c.GetInt("T_State")
	leave, err := Attendance.Read_Attendance_ById(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	if leave.T_approver != c.User.T_uuid {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "无权操作!"}
		c.ServeJSON()
		return
	}

	if T_State == 1 {
		leave.T_State = Attendance.AttendancePass
	}

	if T_State == 0 {
		leave.T_State = Attendance.AttendanceNotPass
	}

	id, err := Attendance.Update_Attendance(leave, "T_State")
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "添加失败"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "我的请假", "审批", leave)

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: id}
	c.ServeJSON()
	return
}

func (c *LeaveController) Leave_Del() {
	T_id, _ := c.GetInt("T_id")

	leave, err := Attendance.Read_Attendance_ById(T_id)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "T_id Err!"}
		c.ServeJSON()
		return
	}

	if leave.T_State == Attendance.AttendancePass {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "无权删除!"}
		c.ServeJSON()
		return
	}

	id, err := Attendance.Delete_Attendance(leave)
	if err != nil {
		c.Data["json"] = lib.JSONS{Code: 202, Msg: "删除失败!"}
		c.ServeJSON()
		return
	}

	NatsServer.AddUserLogs(c.User.T_uuid, "我的请假", "删除", strconv.Itoa(T_id))

	c.Data["json"] = lib.JSONS{Code: 200, Msg: "ok!", Data: id}
	c.ServeJSON()
	return
}