forked from golang/hotime
316 lines
7.5 KiB
Go
316 lines
7.5 KiB
Go
|
package admin
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
. "../../../hotime"
|
|||
|
|
. "../../common"
|
|||
|
|
"../../dri/baidu"
|
|||
|
|
"errors"
|
|||
|
|
"fmt"
|
|||
|
|
"github.com/chain-zhang/pinyin"
|
|||
|
|
"github.com/xuri/excelize"
|
|||
|
|
"io"
|
|||
|
|
"os"
|
|||
|
|
"strings"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var CompanyInOutCtr = Ctr{
|
|||
|
|
|
|||
|
|
"upload": func(this *Context) {
|
|||
|
|
|
|||
|
|
//读取网络文件
|
|||
|
|
fi, fheader, err := this.Req.FormFile("file")
|
|||
|
|
if err != nil {
|
|||
|
|
this.Display(3, err)
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
filePath := this.Config.GetString("filePath")
|
|||
|
|
if filePath == "" {
|
|||
|
|
filePath = "excel/2006/01/02/"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
path := time.Now().Format(filePath)
|
|||
|
|
e := os.MkdirAll(this.Config.GetString("tpt")+"/"+path, os.ModeDir)
|
|||
|
|
|
|||
|
|
if e != nil {
|
|||
|
|
this.Display(3, e)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
filePath = path + Md5(ObjToStr(RandX(100000, 9999999))) + fheader.Filename[strings.LastIndex(fheader.Filename, "."):]
|
|||
|
|
newFile, e := os.Create(this.Config.GetString("tpt") + "/" + filePath)
|
|||
|
|
|
|||
|
|
if e != nil {
|
|||
|
|
this.Display(3, e)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
_, e = io.Copy(newFile, fi)
|
|||
|
|
|
|||
|
|
if e != nil {
|
|||
|
|
this.Display(3, e)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//this.Display(0, filePath)
|
|||
|
|
|
|||
|
|
//读取excel
|
|||
|
|
data := excel(this.Config.GetString("tpt") + "/" + filePath)
|
|||
|
|
if len(data) != 1 {
|
|||
|
|
this.Display(3, "表格不标准，请重新提交")
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
err = decodeData2Sql(data[0], this)
|
|||
|
|
if err != nil {
|
|||
|
|
this.Display(4, err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
this.Display(0, "上传成功")
|
|||
|
|
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func decodeData2Sql(table [][]string, this *Context) error {
|
|||
|
|
tags := []Map{} //报错所有的tag记录
|
|||
|
|
tagCtgs := []Map{}
|
|||
|
|
for k, v := range table {
|
|||
|
|
|
|||
|
|
//第一排是指标分类,去除掉第一个参数，其存在合并所以，如果是合并的则删除
|
|||
|
|
if k == 0 {
|
|||
|
|
|
|||
|
|
for ck, cv := range v {
|
|||
|
|
cv = strings.Replace(cv, " ", "", -1)
|
|||
|
|
cv = strings.Replace(cv, "\r\t", "", -1)
|
|||
|
|
cv = strings.Replace(cv, "\r", "", -1)
|
|||
|
|
v[ck] = cv
|
|||
|
|
|
|||
|
|
if ck == 0 {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if cv == "" {
|
|||
|
|
v[ck] = v[ck-1]
|
|||
|
|
cv = v[ck]
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tagCtg := this.Db.Get("tag_ctg", "*", Map{"name": cv})
|
|||
|
|
if tagCtg != nil {
|
|||
|
|
tagCtgs = append(tagCtgs, tagCtg)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
tagCtg = Map{
|
|||
|
|
"name": cv,
|
|||
|
|
"admin_id": this.Session("admin_id").ToInt(),
|
|||
|
|
"create_time": time.Now().Unix(),
|
|||
|
|
"modify_time": time.Now().Unix(),
|
|||
|
|
}
|
|||
|
|
id := this.Db.Insert("tag_ctg", tagCtg)
|
|||
|
|
|
|||
|
|
if id != 0 {
|
|||
|
|
tagCtg["id"] = id
|
|||
|
|
this.Db.Update("tag_ctg", Map{
|
|||
|
|
"index": "," + ObjToStr(id) + ",",
|
|||
|
|
}, Map{"id": id})
|
|||
|
|
|
|||
|
|
tagCtgs = append(tagCtgs, tagCtg)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//第二排是指标项,去除掉第一个参数,
|
|||
|
|
//最重要的是参数tag生成，通过中文转拼音获取首字母实现，遇到同名的则最后加一位数字
|
|||
|
|
if k == 1 {
|
|||
|
|
|
|||
|
|
for ck, cv := range v {
|
|||
|
|
cv = strings.Replace(cv, " ", "", -1)
|
|||
|
|
cv = strings.Replace(cv, "\r\t", "", -1)
|
|||
|
|
cv = strings.Replace(cv, "\r", "", -1)
|
|||
|
|
v[ck] = cv
|
|||
|
|
if ck == 0 {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
//补充分类
|
|||
|
|
if ck >= len(table[0]) {
|
|||
|
|
tagCtgs = append(tagCtgs, tagCtgs[len(tagCtgs)-1])
|
|||
|
|
}
|
|||
|
|
//补充说明
|
|||
|
|
if ck >= len(table[3]) {
|
|||
|
|
table[3] = append(table[3], table[3][len(table[3])-1])
|
|||
|
|
}
|
|||
|
|
//补充单位
|
|||
|
|
if ck >= len(table[2]) {
|
|||
|
|
table[2] = append(table[2], "")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tag := this.Db.Get("tag", "*", Map{"name": cv})
|
|||
|
|
|
|||
|
|
if tag != nil {
|
|||
|
|
tags = append(tags, tag)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
sn, err := pinyin.New(cv).Split(" ").Mode(pinyin.InitialsInCapitals).Convert()
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
sn = cv
|
|||
|
|
} else {
|
|||
|
|
|
|||
|
|
sns := strings.Split(sn, " ")
|
|||
|
|
|
|||
|
|
sn = "IEDC"
|
|||
|
|
for _, v1 := range sns {
|
|||
|
|
if v1 != "" {
|
|||
|
|
sn = sn + string([]rune(v1)[:1])
|
|||
|
|
fmt.Println(sn, v1)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
//sn=strings.Replace(sn,"\ufffd","",-1)
|
|||
|
|
|
|||
|
|
tagsn := this.Db.Get("tag", "id", Map{"sn": sn})
|
|||
|
|
if tagsn != nil {
|
|||
|
|
for i := 0; i < 100; i++ {
|
|||
|
|
tagsn = this.Db.Get("tag", "id", Map{"sn": sn + ObjToStr(i)})
|
|||
|
|
if tagsn == nil {
|
|||
|
|
sn = sn + ObjToStr(i)
|
|||
|
|
break
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
tag = Map{
|
|||
|
|
"name": cv,
|
|||
|
|
"description": table[3][ck], //第三行对应位置为描述
|
|||
|
|
"unit": table[2][ck],
|
|||
|
|
"admin_id": this.Session("admin_id").ToInt(),
|
|||
|
|
"type": 1,
|
|||
|
|
"sn": sn,
|
|||
|
|
"tag_ctg_id": tagCtgs[ck-1].GetCeilInt("id"),
|
|||
|
|
"create_time": time.Now().Unix(),
|
|||
|
|
"modify_time": time.Now().Unix(),
|
|||
|
|
}
|
|||
|
|
id := this.Db.Insert("tag", tag)
|
|||
|
|
tag["id"] = id
|
|||
|
|
tags = append(tags, tag)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//第四排以后是数据项,去除掉第一个参数,
|
|||
|
|
//根据统一社会信用代码识别是否存在，同时将校验部分数字
|
|||
|
|
if k > 4 {
|
|||
|
|
|
|||
|
|
//整行数据
|
|||
|
|
rowData := Map{}
|
|||
|
|
for ck, cv := range v {
|
|||
|
|
if ck == 0 {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
cv = strings.Replace(cv, " ", "", -1)
|
|||
|
|
cv = strings.Replace(cv, "\r\t", "", -1)
|
|||
|
|
cv = strings.Replace(cv, "\r", "", -1)
|
|||
|
|
v[ck] = cv
|
|||
|
|
if cv == "" {
|
|||
|
|
return errors.New("第" + ObjToStr(k+1) + "行，" + ObjToStr(ck+1) + "列，数据为空请注意检查")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
rowData[tags[ck-1].GetString("sn")] = cv
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
companyData := Map{
|
|||
|
|
"name": rowData["IEDCQYMC"],
|
|||
|
|
"sn": rowData["IEDCTYSHXYDM"],
|
|||
|
|
"address": rowData["IEDCQYDZ"],
|
|||
|
|
"unit": rowData["IEDCSDMC"],
|
|||
|
|
"zdmj": rowData["IEDCYDMJ"],
|
|||
|
|
"yysr": rowData["IEDC2NYYSR"],
|
|||
|
|
"lrze": rowData["IEDC2NLRZE"],
|
|||
|
|
"yjsj": rowData["IEDCYNSJ"],
|
|||
|
|
"yfjf": rowData["IEDCNYFFY"],
|
|||
|
|
"zgrs": rowData["IEDCSBRS"],
|
|||
|
|
"zywrwpfdl": rowData["IEDCZYWRWPFDL"],
|
|||
|
|
"zhnh": rowData["IEDCZHNH"],
|
|||
|
|
"admin_id": this.Session("admin_id").ToInt(),
|
|||
|
|
"upload_data": rowData.ToJsonString(),
|
|||
|
|
//"create_time": time.Now().Unix(),
|
|||
|
|
"modify_time": time.Now().Unix(),
|
|||
|
|
//"lat": lat,
|
|||
|
|
//"lng": lng,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
baiduDataStr, _ := baidu.DefaultBaiDuMap.GetPosition(companyData.GetString("address"))
|
|||
|
|
baiduData := ObjToMap(baiduDataStr)
|
|||
|
|
|
|||
|
|
city := Map{}
|
|||
|
|
if baiduData != nil && baiduData.GetCeilInt("status") == 0 {
|
|||
|
|
address := baiduData.GetSlice("result")
|
|||
|
|
if len(address) != 0 {
|
|||
|
|
companyData["lat"] = address.GetMap(0).GetMap("location").GetFloat64("lat")
|
|||
|
|
companyData["lng"] = address.GetMap(0).GetMap("location").GetFloat64("lng")
|
|||
|
|
|
|||
|
|
city = this.Db.Get("city", "id", Map{"name[~]": address.GetMap(0).GetString("district"), "ORDER": "`level` DESC"})
|
|||
|
|
if city == nil {
|
|||
|
|
city = this.Db.Get("city", "id", Map{"name[~]": address.GetMap(0).GetString("city"), "ORDER": "`level` DESC"})
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if city != nil {
|
|||
|
|
companyData["city_id"] = city.GetCeilInt("id")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//行业ID
|
|||
|
|
category := this.Db.Get("category", "id", Map{"code": rowData["IEDCSSXYXLDM"]})
|
|||
|
|
if category != nil {
|
|||
|
|
companyData["category_id"] = category.GetCeilInt("id")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
//此处需要优化
|
|||
|
|
companyData["org_id"] = 3
|
|||
|
|
|
|||
|
|
company := this.Db.Get("company", "id", Map{"sn": companyData["sn"]})
|
|||
|
|
//没有则创建
|
|||
|
|
if company == nil {
|
|||
|
|
//更新两张表
|
|||
|
|
companyData["create_time"] = time.Now().Unix()
|
|||
|
|
id := this.Db.Insert("company", companyData)
|
|||
|
|
companyData["company_id"] = id
|
|||
|
|
} else {
|
|||
|
|
//有则更新
|
|||
|
|
this.Db.Update("company", companyData, Map{"sn": companyData["sn"]})
|
|||
|
|
companyData["company_id"] = company.GetCeilInt("id")
|
|||
|
|
companyData["create_time"] = time.Now().Unix()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
this.Db.Insert("company_history", companyData)
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
return nil
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func excel(filePath string) [][][]string {
|
|||
|
|
xlsx, err := excelize.OpenFile(filePath)
|
|||
|
|
if err != nil {
|
|||
|
|
fmt.Println(err)
|
|||
|
|
os.Exit(1)
|
|||
|
|
}
|
|||
|
|
list := xlsx.GetSheetList()
|
|||
|
|
var data [][][]string
|
|||
|
|
for _, v := range list {
|
|||
|
|
rows, e := xlsx.GetRows(v)
|
|||
|
|
if e != nil {
|
|||
|
|
fmt.Println(e)
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
data = append(data, rows)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return data
|
|||
|
|
}
|