From d1a8878d20a529a37159f71a2af9af8ae4b30d81 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=82=B9=E5=AE=97=E6=A5=A0?= <zouzongnan1995@icloud.com>
Date: Sun, 9 Oct 2022 22:01:51 +0800
Subject: [PATCH] 1

---
 .../brand_list/request/brand_list_request.go  |  2 ++
 .../response/brand_list_response.go           |  2 ++
 platformapi/tiktok_shop/tiktok_api/sku.go     | 20 +++++++++++++++++++
 .../tiktok_shop/tiktok_api/sku_test.go        |  2 +-
 4 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/platformapi/tiktok_shop/sdk-golang/api/brand_list/request/brand_list_request.go b/platformapi/tiktok_shop/sdk-golang/api/brand_list/request/brand_list_request.go
index 4d793783..a586ce5e 100644
--- a/platformapi/tiktok_shop/sdk-golang/api/brand_list/request/brand_list_request.go
+++ b/platformapi/tiktok_shop/sdk-golang/api/brand_list/request/brand_list_request.go
@@ -2,6 +2,8 @@ package brand_list_request
 
 import (
 	"encoding/json"
+	brand_list_response "git.rosy.net.cn/baseapi/platformapi/tiktok_shop/sdk-golang/api/brand_list/response"
+	doudian_sdk "git.rosy.net.cn/baseapi/platformapi/tiktok_shop/sdk-golang/core"
 )
 
 type BrandListRequest struct {
diff --git a/platformapi/tiktok_shop/sdk-golang/api/brand_list/response/brand_list_response.go b/platformapi/tiktok_shop/sdk-golang/api/brand_list/response/brand_list_response.go
index 83b36cac..985594fa 100644
--- a/platformapi/tiktok_shop/sdk-golang/api/brand_list/response/brand_list_response.go
+++ b/platformapi/tiktok_shop/sdk-golang/api/brand_list/response/brand_list_response.go
@@ -1,5 +1,7 @@
 package brand_list_response
 
+import doudian_sdk "git.rosy.net.cn/baseapi/platformapi/tiktok_shop/sdk-golang/core"
+
 type BrandListResponse struct {
 	doudian_sdk.BaseDoudianOpApiResponse
 	Data *BrandListData `json:"data"`
diff --git a/platformapi/tiktok_shop/tiktok_api/sku.go b/platformapi/tiktok_shop/tiktok_api/sku.go
index e9f6f2b0..8bab0e39 100644
--- a/platformapi/tiktok_shop/tiktok_api/sku.go
+++ b/platformapi/tiktok_shop/tiktok_api/sku.go
@@ -2,6 +2,7 @@ package tiktok_api
 
 import (
 	"errors"
+	brand_list_request "git.rosy.net.cn/baseapi/platformapi/tiktok_shop/sdk-golang/api/brand_list/request"
 	product_addV2_request "git.rosy.net.cn/baseapi/platformapi/tiktok_shop/sdk-golang/api/product_addV2/request"
 	product_addV2_response "git.rosy.net.cn/baseapi/platformapi/tiktok_shop/sdk-golang/api/product_addV2/response"
 	product_del_request "git.rosy.net.cn/baseapi/platformapi/tiktok_shop/sdk-golang/api/product_del/request"
@@ -231,3 +232,22 @@ func (a *API) EditPrice(skuPrice *sku_editPrice_request.SkuEditPriceParam) error
 	}
 	return nil
 }
+
+// GetSkuBrand 获取商品所属品牌
+func (a *API) GetSkuBrand(categoryId int64) (int64, error) {
+	request := brand_list_request.New()
+	request.Param.CategoryId = categoryId
+	result, err := request.Execute(a.accessTokenObj)
+	if err != nil {
+		return 0, err
+	}
+	if result.Code != RequestSuccessCode {
+		return 0, errors.New(result.SubMsg)
+	}
+
+	if result.Data == nil || len(result.Data.AuthBrandList) == 0 {
+		return 596120136, nil // 无品牌默认值
+	}
+
+	return result.Data.AuthBrandList[0].BrandId, nil
+}
diff --git a/platformapi/tiktok_shop/tiktok_api/sku_test.go b/platformapi/tiktok_shop/tiktok_api/sku_test.go
index d70daf60..3e993f6c 100644
--- a/platformapi/tiktok_shop/tiktok_api/sku_test.go
+++ b/platformapi/tiktok_shop/tiktok_api/sku_test.go
@@ -104,7 +104,7 @@ func TestCreateStoreCommodity(t *testing.T) {
 		//SubstituteGoodsUrl:    "",
 		//SaleChannelType:       "",
 		//RecruitInfo:           nil,
-		StoreId: 0,
+		//StoreId:               0,
 		//MainProductId:         0,
 		//SaleLimitId:           0,
 		//NamePrefix:            "",