1

business/partner/purchase/tiktok_store/mtwm.go

@@ -47,44 +47,41 @@ func (c *PurchaseHandler) GetVendorID() int {
 }
 
 func (p *PurchaseHandler) GetVendorCategories(ctx *jxcontext.Context) (vendorCats []*model.SkuVendorCategory, err error) {
-	cats, err := api.MtwmAPI.RetailGetSpTagIds()
+	cats, err := getAPI("53802960", 0, "").GetShopCategory(0)
 	if err != nil {
 		return nil, err
 	}
-	vendorCatMapList := make([]map[string]*model.SkuVendorCategory, 3)
-	manID := 10000
-	for i := 0; i < 3; i++ {
-		vendorCatMapList[i] = make(map[string]*model.SkuVendorCategory)
-		for _, v := range cats {
-			if v.Level == 3 {
-				namePathList := strings.Split(strings.Trim(v.NamePath, ","), ",")
-				if len(namePathList) != 3 {
-					panic(fmt.Sprintf("%s没有三级结构", v.NamePath))
-				}
-				name := namePathList[i]
-				if _, ok := vendorCatMapList[i][name]; !ok {
-					cat := &model.SkuVendorCategory{
-						VendorID: model.VendorIDMTWM,
-						Name:     name,  //utils.Interface2String(v["name"]),
-						Level:    i + 1, //int(utils.MustInterface2Int64(v["level"])),
-					}
-					vendorCats = append(vendorCats, cat)
-					vendorCatMapList[i][name] = cat
-					if i == 2 {
-						cat.IsLeaf = 1
-						cat.VendorCategoryID = utils.Int64ToStr(v.ID)
-					} else {
-						cat.VendorCategoryID = utils.Int2Str(manID) // 非叶子结点编码没有实际使用
-						manID++
-					}
-					if i > 0 {
-						cat.ParentID = vendorCatMapList[i-1][namePathList[i-1]].VendorCategoryID
-					}
-				}
-			}
+
+	vendorCats = skuCategory(cats)
+	return vendorCats, nil
+}
+
+func skuCategory(param1 []*tiktok_api.RetailCategoryInfo) (vendorCats []*model.SkuVendorCategory) {
+	size := make([]*model.SkuVendorCategory, 0)
+	for _, v := range param1 {
+		param := &model.SkuVendorCategory{
+			VendorCategoryID: utils.Int64ToStr(v.Id),
+			VendorID:         model.VendorIDDD,
+			Name:             v.Name,
+			Level:            int(v.Level),
+			ParentID:         utils.Int64ToStr(v.ParentId),
+		}
+		if v.Enable == false {
+			param.IsLeaf = 0
+		} else {
+			param.IsLeaf = 1
+		}
+		if param.Name != "" {
+			size = append(size, param)
+		}
+
+		if len(v.Children) != 0 {
+			kk := skuCategory(v.Children)
+			vendorCats = append(vendorCats, kk...)
 		}
 	}
-	return vendorCats, nil
+	vendorCats = append(vendorCats, size...)
+	return
 }
 
 func rangeMtwm2JX(areaStr string) string {