1. BannerList

  1. @logger.catch
  2. def BannerList(self, request: empty_pb2.Empty, context):
  3.     # 获取分类列表
  4.     rsp = goods_pb2.BannerListResponse()
  5.     banners = Banner.select()
  7.     rsp.total = banners.count()
  8.     for banner in banners:
  9.         banner_rsp = goods_pb2.BannerResponse()
  11.         banner_rsp.id = banner.id
  12.         banner_rsp.image = banner.image
  13.         banner_rsp.index = banner.index
  14.         banner_rsp.url = banner.url
  16.         rsp.data.append(banner_rsp)
  18.         return rsp

2. CreateBanner

  1. @logger.catch
  2. def CreateBanner(self, request: goods_pb2.BannerRequest, context):
  3.     banner = Banner()
  5.     banner.image = request.image
  6.     banner.index = request.index
  7.     banner.url = request.url
  8.     banner.save()
  10.     banner_rsp = goods_pb2.BannerResponse()
  11.     banner_rsp.id = banner.id
  12.     banner_rsp.image = banner.image
  13.     banner_rsp.url = banner.url
  15.     return banner_rsp

3. DeleteBanner

  1. @logger.catch
  2. def DeleteBanner(self, request: goods_pb2.BannerRequest, context):
  3.     try:
  4.         banner = Banner.get(request.id)
  5.         banner.delete_instance()
  7.         return empty_pb2.Empty()
  8.     except DoesNotExist:
  9.         context.set_code(grpc.StatusCode.NOT_FOUND)
  10.         context.set_details('记录不存在')
  11.         return empty_pb2.Empty()

4. UpdateBanner

  1. @logger.catch
  2. def UpdateBanner(self, request: goods_pb2.BannerRequest, context):
  3.     try:
  4.         banner = Banner.get(request.id)
  5.         if request.image:
  6.             banner.image = request.image
  7.         if request.index:
  8.             banner.index = request.index
  9.         if request.url:
  10.             banner.url = request.url
  12.         banner.save()
  14.         return empty_pb2.Empty()
  15.     except DoesNotExist:
  16.         context.set_code(grpc.StatusCode.NOT_FOUND)
  17.         context.set_details('记录不存在')
  18.         return empty_pb2.Empty()