package data_manage

import (
	"eta/eta_api/global"
	"eta/eta_api/utils"
	"time"
)

// EdbPythonCode python指标运算代码
type EdbPythonCode struct {
	EdbPythonCodeId int    `orm:"column(edb_python_code_id);pk" gorm:"primaryKey"`
	EdbInfoId       int    `description:"指标id"`
	EdbCode         string `description:"指标编码"`
	PythonCode      string `description:"python代码"`
	ModifyTime      time.Time
	CreateTime      time.Time
}

// Update 更新EdbPythonCode信息
func (edbPythonCode *EdbPythonCode) Update(cols []string) (err error) {
	o := global.DbMap[utils.DbNameIndex]
	err = o.Model(edbPythonCode).Select(cols).Updates(edbPythonCode).Error
	return
}

// AddEdbPythonCode python指标运算代码
func AddEdbPythonCode(item *EdbPythonCode) (lastId int64, err error) {
	o := global.DbMap[utils.DbNameIndex]
	err = o.Create(item).Error
	if err != nil {
		return
	}
	lastId = int64(item.EdbPythonCodeId)
	return
}

// GetEdbPythonCodeById 根据指标id获取python代码
func GetEdbPythonCodeById(edbInfoId int) (item *EdbPythonCode, err error) {
	o := global.DbMap[utils.DbNameIndex]
	sql := ` SELECT * FROM edb_python_code WHERE edb_info_id=? `
	err = o.Raw(sql, edbInfoId).First(&item).Error
	return
}