1. GetSubCategory

  1. @logger.catch
  2. def GetSubCategory(self, request: goods_pb2.CategoryListRequest, context):
  3.     category_list_rsp = goods_pb2.SubCategoryListResponse()
  5.     try:
  6.         category_info = Category.get(Category.id == request.id)
  7.         category_list_rsp.info.id = category_info.id
  8.         category_list_rsp.info.name = category_info.name
  9.         category_list_rsp.info.level = category_info.level
  10.         category_list_rsp.info.isTab = category_info.is_tab
  11.         if category_info.parent_category:
  12.             category_list_rsp.info.parentCategory = category_info.parent_category_id
  13.             except DoesNotExist:
  14.                 context.set_code(grpc.StatusCode.NOT_FOUND)
  15.                 context.set_details('记录不存在')
  16.                 return empty_pb2.Empty()
  18.             categorys = Category.select().where(Category.parent_category == request.id)
  19.             category_list_rsp.total = categorys.count()
  20.             for category in categorys:
  21.                 category_rsp = goods_pb2.CategoryInfoResponse()
  22.                 category_rsp.id = category.id
  23.                 category_rsp.name = category.name
  24.                 if category_info.parent_category:
  25.                     category_rsp.parentCategory = category_info.parent_category_id
  26.                     category_rsp.level = category.level
  27.                     category_rsp.isTab = category.is_tab
  29.                     category_list_rsp.subCategorys.append(category_rsp)
  31.                     return category_list_rsp

2. CreateCategory

  1. @logger.catch
  2. def CreateCategory(self, request: goods_pb2.CategoryInfoRequest, context):
  3.     try:
  4.         category = Category()
  5.         category.name = request.name
  6.         if request.level != 1:
  7.             category.parent_category = request.parentCategory
  8.             category.level = request.level
  9.             category.is_tab = request.isTab
  10.             category.save()
  12.             category_rsp = goods_pb2.CategoryInfoResponse()
  13.             category_rsp.id = category.id
  14.             category_rsp.name = category.name
  15.             if category.parent_category:
  16.                 category_rsp.parentCategory = category.parent_category.id
  17.                 category_rsp.level = category.level
  18.                 category_rsp.isTab = category.is_tab
  19.                 except Exception as e:
  20.                     context.set_code(grpc.StatusCode.INTERNAL)
  21.                     context.set_details('插入数据失败')
  22.                     return empty_pb2.Empty()
  24.                 return category_rsp

3. DeleteCategory

  1. @logger.catch
  2. def DeleteCategory(self, request: goods_pb2.DeleteCategoryRequest, context):
  3.     try:
  4.         category = Category.get(request.id)
  5.         category.delete_instance()
  7.         # TODO 删除响应的category下的商品
  8.         return empty_pb2.Empty()
  9.     except DoesNotExist:
  10.         context.set_code(grpc.StatusCode.NOT_FOUND)
  11.         context.set_details('记录不存在')
  12.         return empty_pb2.Empty()

4. UpdateCategory

  1. @logger.catch
  2. def UpdateCategory(self, request: goods_pb2.CategoryInfoRequest, context):
  3.     try:
  4.         category = Category.get(request.id)
  5.         if request.name:
  6.             category.name = request.name
  7.         if request.parentCategory:
  8.             category.parent_category = request.parentCategory
  9.         if request.level:
  10.             category.level = request.level
  11.         if request.isTab:
  12.             category.is_tab = request.isTab
  13.         category.save()
  15.         return empty_pb2.Empty()
  16.     except DoesNotExist:
  17.         context.set_code(grpc.StatusCode.NOT_FOUND)
  18.         context.set_details('记录不存在')
  19.         return empty_pb2.Empty()