ERP_storage/models/Contract/Contract.go

package Contract

import (
	"ERP_storage/logs"
	"ERP_storage/models/Account"
	"git.baozhida.cn/ERP_libs/lib"
	orm2 "github.com/beego/beego/v2/client/orm"
	"strings"
	"time"

	_ "github.com/astaxie/beego/cache/redis"
	"github.com/beego/beego/v2/adapter/orm"
	_ "github.com/go-sql-driver/mysql"
)

// 合同
type Contract struct {
	Id            int     `orm:"column(ID);size(11);auto;pk"`
	T_number      string  `orm:"size(256);null"`         // 合同编号
	T_customer    string  `orm:"size(256);null"`         // 客户名称
	T_customer_id string  `orm:"size(256);null"`         // 客户id
	T_money       float32 `orm:"digits(12);decimals(2)"` // 合同金额
	T_discount    float32 `orm:"digits(12);decimals(2)"` // 合同优惠价
	T_project     string  `orm:"size(256);null"`         // 项目
	T_type        int     `orm:"size(2);default(1)"`     // 合同类型 1-销售合同 2-验证合同
	T_date        string  `orm:"size(256);null"`         // 签订时间
	T_out         int     `orm:"size(2);default(1)"`     // 出库状态  1-未出库 2-部分出库 3-已出库
	T_State       int     `orm:"size(2);default(1)"`     // 0 删除（伪删除）  1-已通过 2-未通过 3-待审核 审核状态
	T_pdf         string  `orm:"size(256);null"`         // 附件
	T_remark      string  `orm:"type(text);null"`        // 备注
	T_recoveries  string  `orm:"type(text);null"`        // 回款信息 回款时间,回款金额|回款时间,回款金额|
	T_invoice     string  `orm:"type(text);null"`        // 开票信息 开票时间,开票金额|开票时间,开票金额|
	T_start_date  string  `orm:"size(256);null"`         // 起始时间
	T_end_date    string  `orm:"size(256);null"`         // 终止时间

	T_submit   string `orm:"size(256);null"` // 合同提交人员
	T_approver string `orm:"size(256);null"` // 审批人

	CreateTime time.Time `orm:"column(create_time);type(timestamp);null;auto_now_add"` //auto_now_add 第一次保存时才设置时间
	UpdateTime time.Time `orm:"column(update_time);type(timestamp);null;auto_now"`     //auto_now 每次 model 保存时都会对时间自动更新
}

func (t *Contract) TableName() string {
	return "contract" // 数据库名称   // ************** 替换 FormulaList **************
}

type ContractDaoImpl struct {
	orm orm.Ormer
}

func NewContract(orm orm.Ormer) *ContractDaoImpl {
	return &ContractDaoImpl{orm: orm}
}

func init() {
	//注册模型
	orm.RegisterModel(new(Contract))
}

type Contract_R struct {
	Id              int
	T_number        string  // 合同编号
	T_customer      string  // 客户名称
	T_money         float32 // 合同金额
	T_discount      float32 // 合同优惠价
	T_type          int     // 合同类型 1-销售合同 2-验证合同
	T_date          string  // 签订时间
	T_out           int     // 出库状态  1-未出库 2-已部分出库 3-已全部出库
	T_submit        string  // 合同提交人员uuid
	T_submit_name   string  // 合同提交人员名称
	T_approver      string  // 合同审批人员uuid
	T_approver_name string  // 合同审批人员名称
	T_pdf           string  // 附件
	T_State         int     // 1-已通过 2-未通过 3-待审核 审核状态
}

type Contract_Detail struct {
	Id                    int
	T_number              string
	T_customer            string  // 客户名称
	T_money               float32 // 合同金额
	T_discount            float32 // 合同优惠价
	T_project             string  // 项目
	T_type                int     // 合同类型 1-销售合同 2-验证合同
	T_date                string  // 签订时间
	T_submit              string  // 合同提交人员uuid
	T_submit_name         string  // 合同提交人员名称
	T_approver            string  // 审批人uuid
	T_approver_name       string  // 审批人名称
	T_out                 int     // 出库状态  1-未出库 2-已部分出库 3-已全部出库
	T_State               int     // 1-已通过 2-未通过 3-待审核 审核状态
	T_remark              string  // 备注
	T_pdf                 string  // 附件
	T_start_date          string  // 起始时间
	T_end_date            string  // 终止时间
	T_Product             []ContractProduct_R
	T_recoveries          []ContractFinance_R // 回款信息 回款时间,回款金额|回款时间,回款金额|
	T_no_recoveries_money float32             // 未回款金额
	T_invoice             []ContractFinance_R // 开票信息 开票时间,开票金额|开票时间,开票金额|
	T_no_invoice_money    float32             // 未开票金额
}
type ContractFinance_R struct {
	T_date  string  // 时间
	T_money float32 // 金额）
}

type Contract_Verify struct {
	Id             int
	T_number       string
	T_money        float32 // 合同金额
	T_discount     float32 // 合同优惠价
	T_date         string  // 签订时间
	T_start_date   string  // 起始时间
	T_end_date     string  // 终止时间
	T_verify_state int     // 1-未签约 2-已作废 3-已签约 4-即将到期
}

func ContractToContractFinance_R(T_detail string) (r []ContractFinance_R, money float32) {

	if len(T_detail) == 0 {
		return
	}
	detailList := strings.Split(strings.Trim(T_detail, "|"), "|")
	for _, detail := range detailList {
		T_date := strings.Split(detail, ",")[0]
		T_money := strings.Split(detail, ",")[1]
		temp := float32(lib.StringToFloat64TwoDecimal(T_money))
		money += temp
		r = append(r, ContractFinance_R{
			T_date:  T_date,
			T_money: temp,
		})

	}
	return r, money
}

func ContractToContract_R(t Contract) (r Contract_R) {
	// 1-未出库 2-已部分出库 3-已全部出库
	r.T_out = t.T_out
	// 1-已通过 2-未通过 3-待审核
	//if t.T_State == 1 {
	//	r.T_out = 2
	//	state := Read_ContractProduct_T_State_List(t.T_number)
	//	if len(state) == 1 {
	//		if state[0] == 1 {
	//			r.T_out = 1
	//		}
	//		if state[0] == 3 {
	//			r.T_out = 3
	//		}
	//	}
	//}
	r.Id = t.Id
	r.T_number = t.T_number
	r.T_customer = t.T_customer
	r.T_money = t.T_money
	r.T_type = t.T_type
	r.T_date = t.T_date
	r.T_State = t.T_State
	r.T_pdf = t.T_pdf
	r.T_submit = t.T_submit
	r.T_submit_name = Account.Read_User_T_name_Get(t.T_submit)
	r.T_approver = t.T_approver
	r.T_approver_name = Account.Read_User_T_name_Get(t.T_approver)
	return r
}

func ContractToContract_Detail(t Contract, productList []ContractProduct_R) (r Contract_Detail) {
	// 1-未出库 2-已部分出库 3-已全部出库
	r.T_out = t.T_out
	r.Id = t.Id
	r.T_number = t.T_number
	r.T_customer = t.T_customer
	r.T_money = t.T_money
	r.T_discount = t.T_discount
	r.T_project = t.T_project
	r.T_type = t.T_type
	r.T_date = t.T_date
	r.T_State = t.T_State
	r.T_remark = t.T_remark
	r.T_pdf = t.T_pdf
	r.T_submit = t.T_submit
	r.T_submit_name = Account.Read_User_T_name_Get(t.T_submit)
	r.T_approver = t.T_approver
	r.T_approver_name = Account.Read_User_T_name_Get(t.T_approver)
	r.T_Product = productList
	r.T_start_date = t.T_start_date
	r.T_end_date = t.T_end_date

	recoveriesList, recoveries := ContractToContractFinance_R(t.T_recoveries)
	r.T_recoveries = recoveriesList
	r.T_no_recoveries_money = t.T_money - recoveries

	invoiceList, invoice := ContractToContractFinance_R(t.T_invoice)
	r.T_invoice = invoiceList
	r.T_no_invoice_money = t.T_money - invoice

	return r
}

func ContractToContract_Verify(t Contract) (r Contract_Verify) {
	r.Id = t.Id
	r.T_number = t.T_number
	r.T_money = t.T_money
	r.T_discount = t.T_discount
	r.T_date = t.T_date
	r.T_start_date = t.T_start_date
	r.T_end_date = t.T_end_date
	nowTime := time.Now()
	now := time.Date(nowTime.Year(), nowTime.Month(), nowTime.Day(), 0, 0, 0, 0, time.Local)
	end, _ := lib.DateStrToTime(t.T_end_date)
	now2Month := now.AddDate(0, 2, 0)

	// 1-未签约 2-已作废 3-已签约 4-即将到期
	state := 0
	if len(t.T_end_date) == 0 {
		state = 1
	}
	// 结束时间<今天 已过期
	if end.Before(now) {
		state = 2
	} else {
		// 结束时间<今天+2月 即将到期 && 结束时间>今天
		if end.Before(now2Month) || end.Equal(now) {
			state = 4
		}
		// 结束时间>今天+2月 已签约
		if end.After(now2Month) || end.Equal(now2Month) {
			state = 3
		}
	}

	r.T_verify_state = state

	return r
}

// 添加
func (dao *ContractDaoImpl) Add_Contract(r Contract) (id int64, err error) {
	id, err = dao.orm.Insert(&r)
	if err != nil {
		logs.Error(lib.FuncName(), err)
	}
	return id, err
}

// 获取 ById
func (dao *ContractDaoImpl) Read_Contract_ByT_number(T_number string) (r Contract, err error) {
	qs := dao.orm.QueryTable(new(Contract))

	err = qs.Filter("T_number", T_number).Filter("T_State__gt", 0).One(&r)
	if err != nil {
		logs.Error(lib.FuncName(), err)
	}
	return
}

func (dao *ContractDaoImpl) Read_Contract_ById(Id int) (r Contract, err error) {
	qs := dao.orm.QueryTable(new(Contract))
	err = qs.Filter("Id", Id).One(&r)
	if err != nil {
		logs.Error(lib.FuncName(), err)
	}
	return
}

// 修改
func (dao *ContractDaoImpl) Update_Contract(m Contract, cols ...string) error {
	_, err := dao.orm.Update(&m, cols...)
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return err
	}
	return nil
}

// 删除
func (dao *ContractDaoImpl) Delete_Contract(v Contract) error {
	v.T_State = 0
	_, err := dao.orm.Update(&v, "T_State")
	if err != nil {
		logs.Error(lib.FuncName(), err)
	}
	return err
}

// 获取列表
func (dao *ContractDaoImpl) Read_Contract_List(T_submit, T_name string, T_state, T_out, page, page_z int) (r_ []Contract_R, cnt int64) {
	// 也可以直接使用 Model 结构体作为表名
	qs := dao.orm.QueryTable(new(Contract))
	var offset int64
	if page <= 1 {
		offset = 0
	} else {
		offset = int64((page - 1) * page_z)
	}

	// 过滤
	cond := orm.NewCondition()
	cond1 := cond.And("T_State__gt", 0)
	if T_state > 0 {
		cond1 = cond1.And("T_State", T_state)
	}
	if T_out != 1 {
		cond1 = cond1.And("T_type", 1)

	}

	if len(T_name) > 0 {
		cond1 = cond1.AndCond(cond.Or("T_number__icontains", T_name).Or("T_customer__icontains", T_name))
	}

	if len(T_submit) > 0 {
		cond1 = cond1.And("T_submit", T_submit)
	}

	// T_out == 1 查询未出库或部分出库数据
	if T_out == 1 {
		cond1 = cond1.And("T_out__in", []int{1, 2})
	}

	// 查询
	var r []Contract
	_, err := qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-Id").All(&r)

	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}

	cnt, err = qs.SetCond((*orm2.Condition)(cond1)).Count()
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}

	for _, v := range r {
		r_ = append(r_, ContractToContract_R(v))
	}

	return r_, cnt
}

// 获取列表
func (dao *ContractDaoImpl) Read_VerifyContract_List(T_name, T_customer_id string, page, page_z int) (r_ []Contract_Verify, cnt int64) {
	// 也可以直接使用 Model 结构体作为表名
	qs := dao.orm.QueryTable(new(Contract))
	var offset int64
	if page <= 1 {
		offset = 0
	} else {
		offset = int64((page - 1) * page_z)
	}

	// 过滤
	cond := orm.NewCondition()
	cond1 := cond.And("T_State__gt", 0).And("T_type", 2).And("T_customer_id", T_customer_id)
	if len(T_name) > 0 {
		cond1 = cond1.AndCond(cond.Or("T_number__icontains", T_name).Or("T_customer__icontains", T_name))
	}

	// 查询
	var r []Contract
	var err error
	if page_z == 9999 {
		_, err = qs.SetCond((*orm2.Condition)(cond1)).OrderBy("-T_end_date").All(&r)
	} else {
		_, err = qs.Limit(page_z, offset).SetCond((*orm2.Condition)(cond1)).OrderBy("-T_end_date").All(&r)
	}

	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}

	cnt, err = qs.SetCond((*orm2.Condition)(cond1)).Count()
	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}

	for _, v := range r {
		r_ = append(r_, ContractToContract_Verify(v))
	}

	return r_, cnt
}

// 获取最新的验证合同
func (dao *ContractDaoImpl) Read_VerifyContract_Newest(T_customer_id string) (r_ Contract) {
	// 也可以直接使用 Model 结构体作为表名
	qs := dao.orm.QueryTable(new(Contract))

	// 过滤
	cond := orm.NewCondition()
	cond1 := cond.And("T_State__gt", 0).And("T_type", 2).And("T_customer_id", T_customer_id)

	// 查询
	var r []Contract
	_, err := qs.SetCond((*orm2.Condition)(cond1)).OrderBy("-T_end_date").All(&r)

	if err != nil {
		logs.Error(lib.FuncName(), err)
		return
	}

	if len(r) == 0 {
		return
	}

	return r[0]
}

// 获取验证合同
func (dao *ContractDaoImpl) Read_Contract_ByT_date(T_customer_id, T_start_date, T_end_date string) (r Contract, err error) {
	qs := dao.orm.QueryTable(new(Contract))

	err = qs.Filter("T_customer_id", T_customer_id).Filter("T_start_date", T_start_date).Filter("T_end_date", T_end_date).
		Filter("T_type", 2).Filter("T_State__gt", 0).One(&r)
	if err != nil {
		logs.Error(lib.FuncName(), err)
	}
	return
}