更新研发

2022-01-17 04:47:39 +08:00
parent 7e2abb43e3
commit a95b2fccd7
7 changed files with 605 additions and 18 deletions
@@ -0,0 +1,315 @@
+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
+}
@@ -24,7 +24,7 @@
 	"db": {
 		"mysql": {
 			"host": "192.168.6.253",
-			"name": "myhs",
+			"name": "iedc_dev",
 			"password": "dasda8454456",
 			"port": "3306",
 			"prefix": "",
@@ -2,15 +2,25 @@ package main

 import (
 	"../../hotime"
+	"../common"
+	"io/ioutil"
+
+	//"../dri/aliyun"
+	"../dri/baidu"
 	"../dri/ddsms"
+	"./admin"
 	"./app"
 	"fmt"
+	"github.com/xuri/excelize"
+	"os"
 	"time"
 )

 func main() {
 	date, _ := time.Parse("2006-01-02 15:04", time.Now().Format("2006-01-02")+" 14:00")
 	fmt.Println(date, date.Unix())
+
+	baidu.DefaultBaiDuMap.Init("ZeT902EZvVgIoGVWEFK3osUm")
 	//fmt.Println("0123456"[1:7])
 	appIns := hotime.Init("config/config.json")
 	//RESTfull接口适配
@@ -18,6 +28,86 @@ func main() {
 		//支撑权限设置
 		return true
 	})
+	appIns.Router["admin"]["company_inout"] = admin.CompanyInOutCtr
+	appIns.Router["admin"]["test"] = hotime.Ctr{
+
+		"test": func(this *hotime.Context) {
+			data := excel()
+			this.Display(0, data)
+		},
+
+		"api": func(this *hotime.Context) {
+			//curl -i --get --include 'http://api.81api.com/getCompanyBaseInfo/小米科技有限责任公司/'  -H 'Authorization:APPCODE 你自己的AppCode'
+			//aliyun.DefaultCompany.Init("06c6a07e89dd45c88de040ee1489eef7")
+			//data,err:=aliyun.DefaultCompany.GetCompanyBaseInfo("小米科技有限责任公司")
+			//fmt.Println(data)
+			//if err!=nil{
+			//	this.Display(4,err)
+			//	return
+			//}
+
+			data1, _ := baidu.DefaultBaiDuMap.GetPosition("简阳市东溪镇奎星路28号")
+
+			this.Display(0, common.ObjToMap(data1))
+
+		},
+		"temp": func(this *hotime.Context) {
+			count := 0
+			for true {
+
+				data := this.Db.Get("area_copy", "id,parent_id,`index`,parent_code", common.Map{"index": nil, "ORDER": "`level` ASC"})
+
+				if data == nil {
+					break
+				}
+				count++
+				//if data.GetCeilInt("parent_id")==0{
+				//	data["parent_id"]=3750
+				//}
+				parentData := this.Db.Get("area_copy", "id,parent_id,`index`", common.Map{"area_code": data.GetString("parent_code")})
+				if parentData == nil {
+					this.Db.Update("area_copy", common.Map{"index": data.GetString("id") + ","}, common.Map{"id": data.GetString("id")})
+				} else {
+					this.Db.Update("area_copy", common.Map{"index": parentData.GetString("index") + data.GetString("id") + ",", "parent_id": parentData["id"]}, common.Map{"id": data.GetString("id")})
+				}
+
+			}
+			this.Display(0, count)
+
+		},
+		"tempctg": func(this *hotime.Context) {
+			dataBytes, e := ioutil.ReadFile("2017.json")
+			if e != nil {
+				this.Display(4, e)
+				return
+			}
+			data := common.ObjToSlice(string(dataBytes))
+
+			for k, _ := range data {
+				ins := data.GetMap(k)
+				ins["level"] = 1
+				id := this.Db.Insert("category", common.Map{"name": ins.GetString("name"),
+					"code":        ins.GetString("code"),
+					"level":       ins["level"],
+					"parent_id":   1,
+					"admin_id":    this.Session("admin_id").ToInt(),
+					"create_time": time.Now().Unix(),
+					"modify_time": time.Now().Unix(),
+				})
+
+				if id != 0 {
+					ins["id"] = id
+				}
+
+				ins["index"] = ",1," + common.ObjToStr(id) + ","
+				this.Db.Update("category", common.Map{"index": ins["index"]}, common.Map{"id": ins["id"]})
+
+				run(ins, this)
+
+			}
+
+		},
+	}

 	//makeCode := code.MakeCode{}
 	//fmt.Println(common.ObjToStr(makeCode.Db2JSON("admin","test",appIns.Db)))
@@ -29,3 +119,55 @@ func main() {
 		"app": app.Project,
 	})
 }
+
+func excel() common.Slice {
+	xlsx, err := excelize.OpenFile("tpt/piduqu.xlsx")
+	if err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+	list := xlsx.GetSheetList()
+	data := common.Slice{}
+	for _, v := range list {
+		rows, e := xlsx.GetRows(v)
+		fmt.Println(rows, e)
+		data = append(data, rows)
+
+		//for k1,v1:=range rows{
+		//
+		//}
+
+	}
+
+	return data
+}
+
+func run(ins common.Map, this *hotime.Context) {
+	cs := ins.GetSlice("children")
+	if cs == nil {
+		return
+	}
+	for ck, _ := range cs {
+
+		cins := cs.GetMap(ck)
+		cins["level"] = ins.GetCeilInt("level") + 1
+		id := this.Db.Insert("category", common.Map{"name": cins.GetString("name"),
+			"code":        cins.GetString("code"),
+			"level":       cins["level"],
+			"parent_id":   ins["id"],
+			"admin_id":    this.Session("admin_id").ToInt(),
+			"create_time": time.Now().Unix(),
+			"modify_time": time.Now().Unix(),
+		})
+
+		if id != 0 {
+			cins["id"] = id
+		}
+
+		cins["index"] = ins.GetString("index") + common.ObjToStr(id) + ","
+		this.Db.Update("category", common.Map{"index": cins["index"]}, common.Map{"id": cins["id"]})
+
+		run(cins, this)
+
+	}
+}