brand.py 8.88 KB
import json
from utils.base import APIView, get_offset_count
from utils.logger import error_logger


class BrandListView(APIView):

    def get(self, request):
        offset, count = get_offset_count(request)
        cn_name = request.GET.get('cn_name', None)
        id_ = request.GET.get('id', None)
        has_area = request.GET.get('has_area', None)
        is_online = request.GET.get('is_online', None)
        has_icon = request.GET.get('has_icon', None)
        classify_id = request.GET.get('classify_id', None)

        data = self.rpc['neptune/commodity/brand/list'](
            offset=offset, count=count, cn_name=cn_name, id_=id_, has_area=has_area, 
            is_online=is_online, has_icon=has_icon, classify_id=classify_id).unwrap()
        brand_ids = [obj.get('id') for obj in data]

        product_count = self.rpc['neptune/commodity/brand/product_count'](ids=brand_ids).unwrap()

        category_ids = self.rpc['neptune/commodity/brand/brands_category_ids'](ids=brand_ids).unwrap()                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                     
        
        category_infos = self.rpc['neptune/commodity/category/infos'](ids=category_ids).unwrap()
        
        category_dict = {str(obj.get('id')): obj for obj in category_infos} if category_infos else {}

        for obj in data:
            brand_id = obj.get('id')
            obj['product_num'] = product_count.get(str(brand_id)) or 0
            obj['category_infos'] = [{'id': id_, 'cn_name': category_dict.get(str(id_)).get('cn_name')}
                                     for id_ in category_ids.get(str(brand_id))] if category_dict.get(str(id_)) else []
            obj['grade'] = 0
            obj['classify_info'] = []

            obj.pop('platform')

        total = self.rpc['neptune/commodity/brand/count'](
            cn_name=cn_name, id_=id_, has_area=has_area, is_online=is_online, has_icon=has_icon,
            classify_id=classify_id).unwrap()
        result = {
            'list': data,
            'total': total,
        }

        return result


class BrandInfoView(APIView):
    
    def get(self, request):
        id_ = request.GET.get('id')

        data = self.rpc['neptune/commodity/brand/info'](id_=id_).unwrap()
        classify_infos = self.rpc['neptune/commodity/classify/infos'](brand_id=id_).unwrap()
        data['classify_infos'] = [{'id': obj.get('id'), 'cn_name': obj.get('cn_name')} for obj in classify_infos]
        category_ids = self.rpc['neptune/commodity/brand/category_ids'](id_=id_).unwrap()
        category_infos = self.rpc['neptune/commodity/category/infos'](ids=category_ids).unwrap()
        data['category_infos'] = [{'id': obj.get('id'), 'cn_name': obj.get('cn_name')} for obj in category_infos]
        return data


class BrandCreateView(APIView):

    def post(self, request):
        cn_name = request.POST.get('cn_name')
        icon = request.POST.get('icon', '')
        en_name = request.POST.get('en_name', None)
        alias = request.POST.get('alias', None)
        area = request.POST.get('area', None)
        classify_id = request.POST.get('classify_id', None)
        category_ids = json.loads(request.POST.get('category_ids', '[]'))
        description = request.POST.get('description', None)
        is_online = request.POST.get('is_online', False)

        if not cn_name or not classify_id:
            return r'缺少参数'

        data = self.rpc['neptune/commodity/brand/create'](
            cn_name=cn_name, icon=icon, en_name=en_name, alias=alias, area=area, 
            classify_id=classify_id, description=description,
            is_online=is_online).unwrap()
        brand_id = data.get('id')

        self.rpc['neptune/commodity/brand/batch_add_categorys'](
            id_=brand_id, category_ids=category_ids).unwrap()

        return data

    
class BrandUpdateView(APIView):

    def post(self, request):
        id_ = request.POST.get('id', None)
        cn_name = request.POST.get('cn_name', None)
        en_name = request.POST.get('en_name', None)
        alias = request.POST.get('alias', None)
        area = request.POST.get('area', None)
        icon = request.POST.get('icon', None)
        description = request.POST.get('description', None)
        is_online = request.POST.get('is_online', None)
        classify_id = request.POST.get('classify_id', None)
        category_ids = json.loads(request.POST.get('category_ids', '[]'))

        if not id_:
            return r'缺少参数'

        data = self.rpc['neptune/commodity/brand/update'](
            id_=id_, cn_name=cn_name, en_name=en_name, alias=alias, area=area, icon=icon, 
            description=description, is_online=is_online, classify_id=classify_id).unwrap()

        self.rpc['neptune/commodity/brand/batch_add_categorys'](
            id_=id_, category_ids=category_ids).unwrap()

        return data