新增客户导入

models/company.go

@@ -0,0 +1,16 @@
+package models
+
+import "rdluck_tools/orm"
+
+type Company struct {
+	CompanyId int
+	CompanyName string
+}
+
+func GetCompanyByName(companyName string)(item *Company,err error)  {
+	o:=orm.NewOrm()
+	sql:=`SELECT * FROM company WHERE company_name=? `
+	err=o.Raw(sql,companyName).QueryRow(&item)
+	return
+}
+

models/users.go

@@ -41,3 +41,11 @@ func GetWxUserByMobile(mobile string) (item *WxUser, err error) {
 	err = orm.NewOrm().Raw(sql, mobile).QueryRow(&item)
 	return
 }
+
+func AddWxUser(companyId int,mobile,realName string) (err error) {
+	sql:=`INSERT INTO wx_user(company_id,real_name,mobile,first_login,enabled,is_note,from_type,apply_method) VALUES (?,?,?,1,1,1,'report',0);`
+	orm.NewOrm().Raw(sql,companyId,realName,mobile).Exec()
+	return
+}
+
+

services/company.go

@@ -0,0 +1,88 @@
+package services
+
+import (
+	"fmt"
+	"github.com/tealeg/xlsx"
+	"hongze/hongze_task/models"
+	"hongze/hongze_task/utils"
+	"strings"
+)
+
+//客户
+func ImportCompany()  {
+	var err error
+	defer func() {
+		if err!=nil {
+			fmt.Println("err:",err.Error())
+		}
+	}()
+	fileName:="./company_0713.xlsx"
+	xlFile, err := xlsx.OpenFile(fileName)
+	if err != nil {
+		fmt.Printf("open failed: %s\n", err)
+	}
+	n:=0
+	for _,sheet:=range xlFile.Sheet{
+		rows:=sheet.MaxRow
+		for i:=0;i<rows;i++{
+			if i > 2 {
+				row,err:=sheet.Row(i)
+				if err!=nil {
+					fmt.Println("get row err:"+err.Error())
+					return
+				}
+
+				realName:=row.GetCell(1).String()
+				realName=strings.Trim(realName," ")
+				realName=strings.Replace(realName," ","",-1)
+
+				mobile:=row.GetCell(2).String()
+				mobile=strings.Trim(mobile," ")
+				mobile=strings.Replace(mobile," ","",-1)
+				fmt.Println(mobile)
+				companyName:=row.GetCell(6).String()
+				fmt.Println(companyName)
+				companyName=strings.Trim(companyName," ")
+				companyName=strings.Replace(companyName," ","",-1)
+				company,err:=models.GetCompanyByName(companyName)
+				if err!=nil  {
+					if utils.ErrNoRow() == err.Error(){
+						utils.FileLog.Info("客户不存在%s",companyName)
+						fmt.Println("GetCompanyByName Err:"+err.Error())
+						return
+					}else{
+						fmt.Println("GetCompanyByName Err:"+err.Error())
+						return
+					}
+				}
+				if company==nil {
+					fmt.Println("客户不存在,",companyName)
+					utils.FileLog.Info("客户不存在 %s",companyName)
+					return
+				}
+
+				wxUser,err:=models.GetWxUserByMobile(mobile)
+				if err!=nil && err.Error()!=utils.ErrNoRow()  {
+					fmt.Println("GetWxUserByMobile Err:"+err.Error())
+					return
+				}
+				if wxUser==nil {
+					//用户不存在，新增
+					fmt.Println("AddWxUser")
+					fmt.Println(company.CompanyId,mobile,realName)
+					err=models.AddWxUser(company.CompanyId,mobile,realName)
+					if err!=nil {
+						fmt.Println("AddWxUser Err:"+err.Error())
+						return
+					}
+				}else{
+					fmt.Println("用户存在：mobile:",mobile)
+					fmt.Println(company.CompanyId,mobile,realName)
+				}
+				fmt.Println(companyName,company.CompanyId)
+				n++
+			}
+		}
+	}
+	fmt.Println("len:",n)
+}

services/task.go

@@ -15,7 +15,7 @@ func Task() {
 	toolbox.AddTask("oneMinute", oneMinute)
 
 	toolbox.StartTask()
-
+	//ImportCompany()
 	fmt.Println("task end")
 }