1. CategoryBrandList

  1. @logger.catch
  2. def CategoryBrandList(self, request: empty_pb2.Empty, context):
  3.     # 获取品牌分类列表
  4.     rsp = goods_pb2.CategoryBrandListResponse()
  5.     category_brands = GoodsCategoryBrand.select()
  7.     #分页
  8.     start = 0
  9.     per_page_nums = 10
  10.     if request.pagePerNums:
  11.         per_page_nums = request.PagePerNums
  12.     if request.pages:
  13.         start = per_page_nums * (request.pages - 1)
  15.     category_brands = category_brands.limit(per_page_nums).offset(start)
  17.     rsp.total = category_brands.count()
  18.     for category_brand in category_brands:
  19.         category_brand_rsp = goods_pb2.CategoryBrandResponse()
  21.         category_brand_rsp.id = category_brand.id
  22.         category_brand_rsp.brand.id = category_brand.brand.id
  23.         category_brand_rsp.brand.name = category_brand.brand.name
  24.         category_brand_rsp.brand.logo = category_brand.brand.logo
  26.         category_brand_rsp.category.id = category_brand.category.id
  27.         category_brand_rsp.category.name = category_brand.category.name
  28.         category_brand_rsp.category.parentCategory = category_brand.category.parent_category_id
  29.         category_brand_rsp.category.level = category_brand.category.level
  30.         category_brand_rsp.category.isTab = category_brand.category.is_tab
  32.         rsp.data.append(category_brand_rsp)
  33.     return rsp

2. GetCategoryBrandList

  1. @logger.catch
  2. def GetCategoryBrandList(self, request, context):
  3.     #获取某一个分类的所有品牌
  4.     rsp = goods_pb2.BrandListResponse()
  5.     try:
  6.         category = Category.get(Category.id == request.id)
  7.         category_brands = GoodsCategoryBrand.select().where(GoodsCategoryBrand.category == category)
  8.         rsp.total = category_brands.count()
  9.         for category_brand in category_brands:
  10.             brand_rsp = goods_pb2.BrandInfoResponse()
  11.             brand_rsp.id = category_brand.brand.id
  12.             brand_rsp.name = category_brand.brand.name
  13.             brand_rsp.logo = category_brand.brand.logo
  15.             rsp.data.append(brand_rsp)
  16.     except DoesNotExist as e:
  17.         context.set_code(grpc.StatusCode.NOT_FOUND)
  18.         context.set_details('记录不存在')
  19.         return rsp
  21.     return rsp

3. CreateCategoryBrand

  1. @logger.catch
  2. def CreateCategoryBrand(self, request: goods_pb2.CategoryBrandRequest, context):
  3.     category_brand = GoodsCategoryBrand()
  5.     try:
  6.         brand = Brands.get(request.brandId)
  7.         category_brand.brand = brand
  8.         category = Category.get(request.categoryId)
  9.         category_brand.category = category
  10.         category_brand.save()
  12.         rsp = goods_pb2.CategoryBrandResponse()
  13.         rsp.id = category_brand.id #是另外一种思路
  15.         return rsp
  16.     except DoesNotExist:
  17.         context.set_code(grpc.StatusCode.NOT_FOUND)
  18.         context.set_details('记录不存在')
  19.         return goods_pb2.CategoryBrandResponse()
  20.     except Exception as e:
  21.         context.set_code(grpc.StatusCode.INTERNAL)
  22.         context.set_details('内部错误')
  23.         return goods_pb2.CategoryBrandResponse()

4. DeleteCategoryBrand

5. UpdateCategoryBrand