更新一些脚本

script/group_power_data.go

@@ -3,6 +3,7 @@ package main
 import (
 	"encoding/json"
 	"fmt"
+	"git.hilo.cn/hilo-common/script/model"
 	"git.hilo.cn/hilo-common/script/mysql"
 	"github.com/tealeg/xlsx"
 	"io/ioutil"
@@ -17,14 +18,18 @@ type GroupPowerResponse struct {
 }
 
 type GroupPowerData struct {
+	Id                        uint64  `json:"id"`   // 家族id
 	Code                      string  `json:"code"` // 家族长ID
 	Area                      string  `json:"-"`
+	GroupPowerName            string  `json:"groupPowerName"`            // 家族昵称
 	GroupPowerExp             int64   `json:"groupPowerExp"`             // 经验值
 	GroupPowerConsume         int64   `json:"groupPowerConsume"`         // 家族房间流水
 	GroupPowerMemberNum       int     `json:"groupPowerMemberNum"`       // 成员数
 	GroupPowerMemberCharge    float64 `json:"groupPowerMemberCharge"`    // 成员充值$
 	GroupPowerNewMemberNum    int     `json:"groupPowerNewMemberNum"`    // 新成员数
 	GroupPowerNewMemberCharge float64 `json:"groupPowerNewMemberCharge"` // 新成员充值$
+	CodeSvip                  int     `json:"-"`                         // 家族长svip等级
+	Svip3UpCnt                int     `json:"-"`                         // svip3以上的成员数
 }
 
 func ats23(a interface{}) string {
@@ -43,12 +48,34 @@ func getAreaByCode(code string) string {
 	return "非阿语区"
 }
 
+func getUserSvipLevel(code string) int {
+	var user model.User
+	if err := mysql.ProdReadOnlyDB.Model(model.User{}).Where("code = ?", code).First(&user).Error; err != nil {
+		panic(err)
+	}
+	var level int
+	err := mysql.ProdReadOnlyDB.Table("user_svip").Where("user_id = ?", user.Id).Select("level").Scan(&level).Error
+	if err != nil {
+		panic(err)
+	}
+	return level
+}
+
+func getUserSvip3Cnt(groupPowerId uint64) int {
+	var total int
+	err := mysql.ProdReadOnlyDB.Raw("SELECT count(*) FROM `group_power_user` p,user_svip s where p.user_id = s.user_id AND s.`level` >= 3 AND p.group_power_id = ?", groupPowerId).Scan(&total).Error
+	if err != nil {
+		panic(err)
+	}
+	return total
+}
+
 func main() {
 	var data []GroupPowerData
 	pageIndex := 1
 	pageSize := 500
 	for {
-		url := fmt.Sprintf("https://apiv2.faceline.live/v1/groupPower/page?lang=zh-cn&ownerCode=&beginDate=2023-04-01&endDate=2023-04-12&pageIndex=%d&pageSize=%d&status=1", pageIndex, pageSize)
+		url := fmt.Sprintf("https://apiv2.faceline.live/v1/groupPower/page?lang=zh-cn&ownerCode=&beginDate=2023-06-01&endDate=2023-06-12&pageIndex=%d&pageSize=%d&status=1", pageIndex, pageSize)
 		method := "GET"
 
 		client := &http.Client{}
@@ -59,7 +86,7 @@ func main() {
 			return
 		}
 		req.Header.Add("nonce", "hilo")
-		req.Header.Add("token", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJVc2VySWQiOjI1MSwiRXh0ZXJuYWxJZCI6IiIsImV4cCI6MTY4Mzg3NTI2MH0.EH4OFsM8WIjBxvQjvq6vBuEPaHR0vOoCUdS53wmB-yo")
+		req.Header.Add("token", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJVc2VySWQiOjI1MSwiRXh0ZXJuYWxJZCI6IiIsImV4cCI6MTY4ODU0MDUxOX0.VWJAIv0iWtOgnefrfE5Wbeqc1MvmGjaxuRc47M1PvH8")
 
 		res, err := client.Do(req)
 		if err != nil {
@@ -84,20 +111,24 @@ func main() {
 		data = append(data, resp.Data.Data...)
 		pageIndex++
 	}
-	excelFileName := fmt.Sprintf("./家族4月数据.xlsx")
+	excelFileName := fmt.Sprintf("./家族6月数据.xlsx")
 	xlFile := xlsx.NewFile()
 	sheet, err := xlFile.AddSheet("slot")
 	if err != nil {
 		panic(err)
 	}
 	row := sheet.AddRow()
-	c1, c2, c3, c4, c5, c6, c7, c8 := row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell()
-	c1.Value, c2.Value, c3.Value, c4.Value, c5.Value, c6.Value, c7.Value, c8.Value = "家族长ID", "区域", "经验值", "家族房间流水", "成员数", "成员充值$", "新成员数", "新成员充值$"
+	c1, c2, c3, c4, c5, c6, c7, c8, c9, c10, c11 := row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell()
+	c1.Value, c2.Value, c3.Value, c4.Value, c5.Value, c6.Value, c7.Value, c8.Value, c9.Value, c10.Value, c11.Value = "家族长ID", "家族名称", "区域", "经验值", "家族房间流水", "成员数", "成员充值$", "新成员数", "新成员充值$", "家族长svip等级", "家族超过SVIP3的人数"
 	for _, d := range data {
+		d.Area = getAreaByCode(d.Code)
+		d.CodeSvip = getUserSvipLevel(d.Code)
+		d.Svip3UpCnt = getUserSvip3Cnt(d.Id)
 		row := sheet.AddRow()
-		c1, c2, c3, c4, c5, c6, c7, c8 := row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell()
-		c1.Value, c2.Value, c3.Value, c4.Value, c5.Value, c6.Value, c7.Value, c8.Value = d.Code, getAreaByCode(d.Code), ats23(d.GroupPowerExp), ats23(d.GroupPowerConsume), ats23(d.GroupPowerMemberNum),
-			ats23(d.GroupPowerMemberCharge), ats23(d.GroupPowerNewMemberNum), ats23(d.GroupPowerNewMemberCharge)
+		c1, c2, c3, c4, c5, c6, c7, c8, c9, c10, c11 := row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell()
+		c1.Value, c2.Value, c3.Value, c4.Value, c5.Value, c6.Value, c7.Value, c8.Value, c9.Value, c10.Value, c11.Value =
+			d.Code, d.GroupPowerName, d.Area, ats23(d.GroupPowerExp), ats23(d.GroupPowerConsume), ats23(d.GroupPowerMemberNum),
+			ats23(d.GroupPowerMemberCharge), ats23(d.GroupPowerNewMemberNum), ats23(d.GroupPowerNewMemberCharge), ats23(d.CodeSvip), ats23(d.Svip3UpCnt)
 	}
 	_ = xlFile.Save(excelFileName)
 }

script/slot_data.go

+package main
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/tealeg/xlsx"
+	"io/ioutil"
+	"net/http"
+)
+
+type SlotDataResponse struct {
+	Data struct {
+		Total int64      `json:"total"`
+		Data  []SlotData `json:"data"`
+	}
+}
+
+type SlotData struct {
+	Code        string  `json:"userCode"`
+	Country     string  `json:"country"`
+	Stake       int64   `json:"stake"`
+	Award       int64   `json:"award"`
+	ChargeMoney float64 `json:"chargeMoney"`
+	ProfitLoss  int64   `json:"profitLoss"`
+	Rank        int     `json:"rank"`
+}
+
+func ats24(a interface{}) string {
+	return fmt.Sprintf("%v", a)
+}
+
+func main() {
+	var data []SlotData
+	//pageIndex := 1
+	//pageSize := 50
+	for {
+		url := fmt.Sprintf("https://apiv2.faceline.live/v1/fruitMachine/rank/stake?lang=zh-cn&pageIndex=1&pageSize=50&beginDate=2023-05-01&endDate=2023-05-31&userCode=&timezone=1")
+		method := "GET"
+
+		client := &http.Client{}
+		req, err := http.NewRequest(method, url, nil)
+
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		req.Header.Add("nonce", "hilo")
+		req.Header.Add("token", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJVc2VySWQiOjI1MSwiRXh0ZXJuYWxJZCI6IiIsImV4cCI6MTY4ODAwNDgwMX0.x5vNFTFvemRq9Vk2ixOluc7Cynm9wSs5dvItjI5ZH-M")
+
+		res, err := client.Do(req)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+
+		body, err := ioutil.ReadAll(res.Body)
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		_ = res.Body.Close()
+		var resp SlotDataResponse
+		err = json.Unmarshal(body, &resp)
+		if err != nil {
+			panic(err)
+		}
+		if len(resp.Data.Data) <= 0 {
+			break
+		}
+		data = append(data, resp.Data.Data...)
+		break
+	}
+	excelFileName := fmt.Sprintf("./fruit数据.xlsx")
+	xlFile := xlsx.NewFile()
+	sheet, err := xlFile.AddSheet("fruit")
+	if err != nil {
+		panic(err)
+	}
+	row := sheet.AddRow()
+	c1, c2, c3, c4, c5, c6, c7 := row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell()
+	c1.Value, c2.Value, c3.Value, c4.Value, c5.Value, c6.Value, c7.Value = "用户ID", "国家", "投注钻石数", "中奖钻石数", "盈亏（中奖-投注）", "充值金额 $", "排名"
+	for _, d := range data {
+		row := sheet.AddRow()
+		c1, c2, c3, c4, c5, c6, c7 := row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell(), row.AddCell()
+		c1.Value, c2.Value, c3.Value, c4.Value, c5.Value, c6.Value, c7.Value = d.Code, d.Country, ats24(d.Stake), ats24(d.Award), ats24(d.ProfitLoss),
+			ats24(d.ChargeMoney), ats24(d.Rank)
+	}
+	_ = xlFile.Save(excelFileName)
+}