380 lines
12 KiB
Go
380 lines
12 KiB
Go
package test_data
|
||
|
||
import (
|
||
"bytes"
|
||
"encoding/csv"
|
||
"encoding/json"
|
||
"errors"
|
||
"github.com/extrame/xls"
|
||
"github.com/xuri/excelize/v2"
|
||
"golang.org/x/text/encoding/simplifiedchinese"
|
||
"golang.org/x/text/transform"
|
||
"gorm.io/gorm"
|
||
"io/ioutil"
|
||
"log"
|
||
"os"
|
||
"path"
|
||
"path/filepath"
|
||
"strings"
|
||
"testData/global"
|
||
"testData/model"
|
||
"testData/utils"
|
||
)
|
||
|
||
func HandleSTS8200Excel(fileText *model.FileText) {
|
||
rows := make([][]string, 0)
|
||
if strings.ToLower(path.Ext(fileText.Name)) == ".csv" {
|
||
csvBytes, err := ioutil.ReadFile(fileText.Path)
|
||
if err != nil {
|
||
log.Println("open err:", err)
|
||
return
|
||
}
|
||
//解决读取csv中文乱码的问题
|
||
reader := csv.NewReader(transform.NewReader(bytes.NewReader(csvBytes), simplifiedchinese.GBK.NewDecoder()))
|
||
//reader := csv.NewReader(file)
|
||
reader.FieldsPerRecord = -1
|
||
csvRows, err := reader.ReadAll()
|
||
if err != nil {
|
||
log.Println("open err:", err)
|
||
return
|
||
}
|
||
rows = csvRows
|
||
} else if strings.ToLower(path.Ext(fileText.Name)) == ".xls" {
|
||
if xlFile, err := xls.Open(fileText.Path, "utf-8"); err == nil {
|
||
//第一个sheet
|
||
sheet := xlFile.GetSheet(0)
|
||
if sheet.MaxRow != 0 {
|
||
temp := make([][]string, 0)
|
||
for i := 0; i < int(sheet.MaxRow); i++ {
|
||
row := sheet.Row(i)
|
||
if row == nil {
|
||
continue
|
||
}
|
||
data := make([]string, 0)
|
||
if row.LastCol() > 0 {
|
||
for j := 0; j < row.LastCol(); j++ {
|
||
col := row.Col(j)
|
||
data = append(data, col)
|
||
}
|
||
temp = append(temp, data)
|
||
}
|
||
}
|
||
rows = append(rows, temp...)
|
||
} else {
|
||
log.Println("open err:", err)
|
||
return
|
||
}
|
||
}
|
||
} else if strings.ToLower(path.Ext(fileText.Name)) == ".xlsx" {
|
||
xlsxFile, err := excelize.OpenFile(fileText.Path)
|
||
if err != nil {
|
||
log.Println("open err:", err)
|
||
return
|
||
}
|
||
sheetName := xlsxFile.GetSheetName(0)
|
||
rows, err = xlsxFile.GetRows(sheetName)
|
||
if err != nil {
|
||
log.Println("xlsx读取数据失败:", err)
|
||
return
|
||
}
|
||
}
|
||
|
||
details := make(map[string]string)
|
||
newExcel := make([][]string, 0)
|
||
var title []string
|
||
titleInfoMap := make(map[string]model.DataInfo)
|
||
//sBinMap := make(map[string]model.BinInfo)
|
||
sBinMap := make(map[string]string)
|
||
paramsMap := make(map[string]string)
|
||
sbinStartIndex, sbinEndIndex := -1, -1
|
||
titleIndex, unitIndex, limitLIndex, limitUIndex, dataIndex := -1, -1, -1, -1, -1
|
||
for i := 1; i < len(rows); i++ {
|
||
if rows[i][0] == "" {
|
||
continue
|
||
}
|
||
if i == len(rows)-1 && unitIndex == -1 {
|
||
log.Println("特殊文件格式,文件路径为:", fileText.Path)
|
||
return
|
||
}
|
||
if sbinEndIndex < 0 {
|
||
if strings.Contains(rows[i][0], "SBin") {
|
||
if sbinStartIndex < 0 {
|
||
sbinStartIndex = i
|
||
}
|
||
s := rows[i][0]
|
||
if strings.Index(s, "]") != -1 {
|
||
sbinNumStartIndex := strings.Index(s, "[")
|
||
sbinNumEndIndex := strings.Index(s, "]")
|
||
var startIndex, endIndex int
|
||
for index := sbinNumEndIndex + 1; index < len(s); index++ {
|
||
if startIndex == 0 {
|
||
if s[index] != ' ' {
|
||
startIndex = index
|
||
}
|
||
} else {
|
||
if s[index] == ' ' || s[index] == '_' {
|
||
endIndex = index
|
||
break
|
||
}
|
||
}
|
||
}
|
||
//sBinMap[s[sbinNumStartIndex+1:sbinNumEndIndex]] = model.BinInfo{
|
||
// Name: s[startIndex:endIndex],
|
||
// HBin: s[strings.LastIndex(s, " ")+1:],
|
||
//}
|
||
sBinMap[s[sbinNumStartIndex+1:sbinNumEndIndex]] = s[strings.LastIndex(s, " ")+1:]
|
||
paramsMap[s[startIndex:endIndex]] = s[sbinNumStartIndex+1 : sbinNumEndIndex]
|
||
}
|
||
} else {
|
||
if sbinStartIndex > 0 {
|
||
sbinEndIndex = i
|
||
continue
|
||
}
|
||
}
|
||
}
|
||
//if len(sBinMap) == 0 {
|
||
rows[i][0] = strings.ReplaceAll(rows[i][0], "：", ":")
|
||
splitIndex := strings.Index(rows[i][0], ":")
|
||
if splitIndex != -1 {
|
||
details[rows[i][0][:splitIndex]] = rows[i][0][splitIndex+1:]
|
||
}
|
||
//}
|
||
if titleIndex < 0 {
|
||
if strings.Contains(rows[i][0], "SITE_NUM") {
|
||
titleIndex = i
|
||
title = rows[i]
|
||
continue
|
||
}
|
||
} else {
|
||
if unitIndex < 0 {
|
||
if strings.Contains(rows[i][0], "Unit") {
|
||
rows[i] = append(rows[i], utils.FillData(len(rows[titleIndex])-len(rows[i]))...)
|
||
unitIndex = i
|
||
continue
|
||
}
|
||
}
|
||
if limitLIndex < 0 {
|
||
if strings.Contains(rows[i][0], "LimitL") {
|
||
rows[i] = append(rows[i], utils.FillData(len(rows[titleIndex])-len(rows[i]))...)
|
||
limitLIndex = i
|
||
continue
|
||
}
|
||
}
|
||
if limitUIndex < 0 {
|
||
if strings.Contains(rows[i][0], "LimitU") {
|
||
rows[i] = append(rows[i], utils.FillData(len(rows[titleIndex])-len(rows[i]))...)
|
||
limitUIndex = i
|
||
continue
|
||
}
|
||
}
|
||
if dataIndex < 0 && limitUIndex > 0 {
|
||
if rows[i][0] != rows[i][2] {
|
||
dataIndex = i
|
||
break
|
||
}
|
||
}
|
||
}
|
||
}
|
||
if titleIndex != -1 {
|
||
for k, v := range rows[titleIndex] {
|
||
titleInfoMap[v] = model.DataInfo{
|
||
Unit: rows[unitIndex][k],
|
||
LimitL: rows[limitLIndex][k],
|
||
LimitU: rows[limitUIndex][k],
|
||
}
|
||
}
|
||
for i := dataIndex; i < len(rows); i++ {
|
||
if len(rows[i]) < len(rows[titleIndex]) {
|
||
rows[i] = append(rows[i], utils.FillData(len(rows[titleIndex])-len(rows[i]))...)
|
||
}
|
||
newExcel = append(newExcel, rows[i])
|
||
}
|
||
} else {
|
||
log.Println("特殊文件格式,文件路径为:", fileText.Path)
|
||
return
|
||
}
|
||
|
||
//if details["Test_Code"] == "" {
|
||
// details["Test_Code"] = step
|
||
//} else {
|
||
// details["Test_Code"] = details["Test_Code"][:2]
|
||
//}
|
||
//strReader := transform.NewReader(bytes.NewReader([]byte(details["Device_Name"])), simplifiedchinese.GBK.NewDecoder())
|
||
//product, _ := ioutil.ReadAll(strReader)
|
||
//if string(product) == "" {
|
||
// product = []byte(fileText.ProductName)
|
||
//}
|
||
testProgram := details["Program"][strings.LastIndex(details["Program"], "\\")+1:]
|
||
sbinHbin, _ := json.Marshal(&sBinMap)
|
||
titleInfo, _ := json.Marshal(&titleInfoMap)
|
||
paramsSbin, _ := json.Marshal(¶msMap)
|
||
var fileName string
|
||
if fileText.Procedure == "CP" {
|
||
fileName = fileText.ProductName + "_" + fileText.Lot + "_" + fileText.ChipNum + "_" + fileText.Procedure + ".csv"
|
||
} else {
|
||
fileName = fileText.ProductName + "_" + fileText.Lot + "_" + fileText.SubBatch + "_" + fileText.Procedure + ".csv"
|
||
}
|
||
//fileName := string(product) + "_" + fileText.Lot + "_" + details["Test_Code"] + ".csv"
|
||
//var waferID string
|
||
//if fileText.Factory == "saimeike" {
|
||
// if step == "CP1" || step == "CP2" {
|
||
// length := len(fileText.Name)
|
||
// waferID = fileText.Name[length-10 : length-8]
|
||
// fileName = string(product) + "_" + fileText.Lot + "_" + waferID + "_" + details["Test_Code"] + ".csv"
|
||
// } else {
|
||
// if strings.Index(fileText.SubBatch, "-") != strings.LastIndex(fileText.SubBatch, "-") {
|
||
// fileText.SubBatch = fileText.SubBatch[:strings.LastIndex(fileText.SubBatch, "-")]
|
||
// }
|
||
// fileName = string(product) + "_" + fileText.Lot + "_" + fileText.SubBatch + "_" + details["Test_Code"] + ".csv"
|
||
// }
|
||
//} else if fileText.Factory == "xinde" {
|
||
// s := strings.Split(fileText.Name, "_")
|
||
// if len(s) < 5 {
|
||
// log.Println("文件名格式有误,文件路径为:", fileText.Path)
|
||
// return
|
||
// }
|
||
// waferIDStr := s[len(s)-3]
|
||
// if len(waferIDStr) < 5 {
|
||
// log.Println("文件名格式有误,文件路径为:", fileText.Path)
|
||
// return
|
||
// }
|
||
// splitIndex := strings.Index(waferIDStr, "-")
|
||
// if splitIndex == -1 {
|
||
// log.Println("文件名格式有误,文件路径为:", fileText.Path)
|
||
// return
|
||
// }
|
||
// waferID = waferIDStr[splitIndex+1 : splitIndex+3]
|
||
// if _, ok := details["WAFER_ID"]; ok {
|
||
// details["Test_Code"] = "CP"
|
||
// if len(waferIDStr[:splitIndex]) < 8 {
|
||
// fileText.Lot = strings.ToUpper(waferIDStr[:splitIndex])
|
||
// }
|
||
// fileText.SubBatch = ""
|
||
// fileName = string(product) + "_" + fileText.Lot + "_" + waferID + "_" + details["Test_Code"] + ".csv"
|
||
// }
|
||
//} else if fileText.Factory == "qipai" {
|
||
// //details["Test_Code"]
|
||
//} else if fileText.Factory == "changdian" {
|
||
// if details["Device_Name"] == "" {
|
||
// log.Println("文件缺少产品型号信息,文件路径为:", fileText.Path)
|
||
// return
|
||
// }
|
||
// product = []byte(details["Device_Name"])
|
||
// fileName = details["Device_Name"] + "_" + fileText.Lot + "_" + fileText.SubBatch + "_" + details["Test_Code"] + ".csv"
|
||
//}
|
||
//strings.ReplaceAll(fileText.Lot, ".", "-")
|
||
dirPath := filepath.Join("/testData/test/" + fileText.Factory)
|
||
utils.MakeDir(dirPath)
|
||
filePath := filepath.Join(dirPath, fileName)
|
||
var fileHandled *model.FileHandled
|
||
if _, err := os.Stat(filePath); err != nil {
|
||
if os.IsNotExist(err) {
|
||
newExcel = append([][]string{title}, newExcel...)
|
||
_, err = os.Create(filePath)
|
||
if err != nil {
|
||
log.Println("创建文件失败:", err)
|
||
return
|
||
}
|
||
//if fileText.ChipNum != "" {
|
||
// waferID = fileText.ChipNum
|
||
//}
|
||
if errors.Is(global.PostGreSQL.Where("name = ?", fileName).First(&fileHandled).Error, gorm.ErrRecordNotFound) {
|
||
//global.PostGreSQL.Create(&model.FileHandled{
|
||
// Name: fileName,
|
||
// Path: filePath,
|
||
// Size: "",
|
||
// Product: string(product),
|
||
// PBI: fileText.PBI,
|
||
// Factory: fileText.Factory,
|
||
// Step: details["Test_Code"],
|
||
// Lot: fileText.Lot,
|
||
// SubBatch: fileText.SubBatch,
|
||
// TestMachineModel: "STS8200",
|
||
// TestMachine: details["Tester ID"],
|
||
// TestProgram: testProgram,
|
||
// BeginningTime: details["Beginning Time"],
|
||
// EndingTime: details["Ending Time"],
|
||
// SbinHbin: string(sbinHbin),
|
||
// ParamsSbin: string(paramsSbin),
|
||
// TitleInfo: string(titleInfo),
|
||
// WaferID: waferID,
|
||
//})
|
||
global.PostGreSQL.Create(&model.FileHandled{
|
||
Name: fileName,
|
||
Path: filePath,
|
||
Size: "",
|
||
Product: fileText.ProductName,
|
||
PBI: fileText.PBI,
|
||
Factory: fileText.Factory,
|
||
Step: fileText.Procedure,
|
||
Lot: fileText.Lot,
|
||
SubBatch: fileText.SubBatch,
|
||
TestMachineModel: "STS8200",
|
||
TestMachine: details["Tester ID"],
|
||
TestProgram: testProgram,
|
||
BeginningTime: details["Beginning Time"],
|
||
EndingTime: details["Ending Time"],
|
||
SbinHbin: string(sbinHbin),
|
||
ParamsSbin: string(paramsSbin),
|
||
TitleInfo: string(titleInfo),
|
||
WaferID: fileText.ChipNum,
|
||
})
|
||
}
|
||
}
|
||
}
|
||
newCsv, err := os.OpenFile(filePath, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
|
||
if err != nil {
|
||
log.Println("打开新Excel错误:", fileText.Path)
|
||
return
|
||
}
|
||
defer newCsv.Close()
|
||
writer := csv.NewWriter(newCsv)
|
||
defer writer.Flush()
|
||
err = writer.WriteAll(newExcel)
|
||
if err != nil {
|
||
log.Println("写入新Excel错误:", fileText.Path)
|
||
return
|
||
}
|
||
newFile, err := os.Stat(filePath)
|
||
if err != nil {
|
||
log.Println("获取新Excel信息:", fileText.Path)
|
||
return
|
||
}
|
||
global.PostGreSQL.Model(&fileHandled).Where("name = ?", fileName).Updates(map[string]interface{}{
|
||
"product": fileText.ProductName,
|
||
"step": fileText.Procedure,
|
||
//"product": string(product),
|
||
//"step": details["Test_Code"],
|
||
"test_machine_model": "STS8200",
|
||
"test_machine": details["Tester ID"],
|
||
"test_program": testProgram,
|
||
"beginning_time": details["Beginning Time"],
|
||
"ending_time": details["Ending Time"],
|
||
"sbin_hbin": string(sbinHbin),
|
||
"params_sbin": string(paramsSbin),
|
||
"title_info": string(titleInfo),
|
||
"size": utils.FormatFileSize(float64(newFile.Size())),
|
||
})
|
||
var report *model.Report
|
||
if errors.Is(global.PostGreSQL.Where("product = ? AND pbi = ? AND factory = ? AND lot = ? AND step = ? AND sub_batch = ? AND wafer_id = ?",
|
||
fileText.ProductName, fileText.PBI, fileText.Factory, fileText.Lot, fileText.Procedure, fileText.SubBatch, fileText.ChipNum).First(&report).Error, gorm.ErrRecordNotFound) {
|
||
report = &model.Report{
|
||
Product: fileText.ProductName,
|
||
PBI: fileText.PBI,
|
||
Factory: fileText.Factory,
|
||
Step: fileText.Procedure,
|
||
Lot: fileText.Lot,
|
||
SubBatch: fileText.SubBatch,
|
||
TestMachine: details["Tester ID"],
|
||
TestProgram: testProgram,
|
||
WaferID: fileText.ChipNum,
|
||
}
|
||
global.PostGreSQL.Create(&report)
|
||
}
|
||
if fileText.Procedure == "CP" {
|
||
SaveCP(report)
|
||
} else {
|
||
SaveFT(report)
|
||
}
|
||
}
|