1

platformapi/tiktok_shop/tiktok_api/sku.go

@@ -440,9 +440,9 @@ func (a *API) GetProductAuditList(page, pageSize, status int64) ([]product_audit
 }
 
 // GetRecommendCategory 根据图片预测商品类目,优先四级>三级>二级>1级
-func (a *API) GetRecommendCategory(picParams []string) int64 {
+func (a *API) GetRecommendCategory(picParams []string) (int64, error) {
 	if len(picParams) == 0 {
-		return 0
+		return 0, errors.New("抖音根据图片预测分类,图片不能为空")
 	}
 	request := product_GetRecommendCategory_request.New()
 	request.Param.Scene = "smart_publish" // 根据图片获取分类属性
@@ -454,10 +454,14 @@ func (a *API) GetRecommendCategory(picParams []string) int64 {
 
 	result, err := request.Execute(a.accessTokenObj)
 	if err != nil {
-		return 0
+		return 0, err
 	}
 	if result.Code != RequestSuccessCode {
-		return 0
+		return 0, errors.New(result.SubMsg)
+	}
+
+	if len(result.Data.CategoryDetails) == 0 {
+		return 0, errors.New("抖音根据图片获取分类,返回值为0")
 	}
 
 	var firstCid int64
@@ -465,33 +469,31 @@ func (a *API) GetRecommendCategory(picParams []string) int64 {
 	var thirdCid int64
 	var fourthCid int64
 
-	for _, v := range result.Data.CategoryDetails {
-		if v.CategoryDetail.FirstCid != 0 {
-			firstCid = v.CategoryDetail.FirstCid
-		}
-		if v.CategoryDetail.SecondCid != 0 {
-			secondCid = v.CategoryDetail.SecondCid
-		}
-		if v.CategoryDetail.ThirdCid != 0 {
-			thirdCid = v.CategoryDetail.ThirdCid
-		}
-		if v.CategoryDetail.FourthCid != 0 {
-			fourthCid = v.CategoryDetail.FourthCid
-		}
+	if result.Data.CategoryDetails[0].CategoryDetail.FirstCid != 0 {
+		firstCid = result.Data.CategoryDetails[0].CategoryDetail.FirstCid
+	}
+	if result.Data.CategoryDetails[0].CategoryDetail.SecondCid != 0 {
+		secondCid = result.Data.CategoryDetails[0].CategoryDetail.SecondCid
+	}
+	if result.Data.CategoryDetails[0].CategoryDetail.ThirdCid != 0 {
+		thirdCid = result.Data.CategoryDetails[0].CategoryDetail.ThirdCid
+	}
+	if result.Data.CategoryDetails[0].CategoryDetail.FourthCid != 0 {
+		fourthCid = result.Data.CategoryDetails[0].CategoryDetail.FourthCid
 	}
 	if fourthCid != 0 {
-		return fourthCid
+		return fourthCid, nil
 	}
 	if thirdCid != 0 {
-		return thirdCid
+		return thirdCid, nil
 	}
 	if secondCid != 0 {
-		return secondCid
+		return secondCid, nil
 	}
 	if firstCid != 0 {
-		return firstCid
+		return firstCid, nil
 	}
-	return 0
+	return 0, errors.New("抖音获取推荐分类异常")
 }
 
 /*--------------------------------方案二 将住上铺平铺到所有门店店中,不在使用子商品------------------------------------------------------------------------------------*/