Subversion Repositories SmartDukaan

def getBrandSubCategoryInfo(category_id, id_list, type):

    if mc.get("brandSubCategoryInfo") is None:

        collection =  get_mongo_connection().Catalog.Deals

        result = collection.aggregate( 

                [

                    {'$match':{'category_id':category_id,'showDeal':1}},

                    {"$group": { "_id": { "subCategoryId": "$subCategoryId", "brand_id": "$brand_id", "brand":"$brand","subCategory":"$subCategory" }, "count": {"$sum": 1} } }

                ]

            );

        from pprint import pprint

        pprint(result)

        subCategoryMap = {}

        brandMap = {}

        for data in result['result']:

            result_set = data['_id']

            count = data['count']

            subCategoryId = result_set['subCategoryId']

            brand_id = result_set['brand_id']

            subCategory = result_set['subCategory']

            brand = result_set['brand']

            if subCategoryMap.has_key(subCategoryId):

                subCategoryMap.get(subCategoryId).append({'brand_id':brand_id, 'count':count, 'subCategory':subCategory, 'brand':brand})

            else:

                subCategoryMap[subCategoryId] = [{'brand_id':brand_id, 'count':count, 'subCategory':subCategory, 'brand':brand}]

            if brandMap.has_key(brand_id):

                brandMap.get(brand_id).append({'subCategoryId':subCategoryId, 'count':count, 'subCategory':subCategory, 'brand':brand})

            else:

                brandMap[brand_id] = [{'subCategoryId':subCategoryId, 'count':count, 'subCategory':subCategory, 'brand':brand}]

        mc.set("brandSubCategoryInfo",{'subCategoryMap':subCategoryMap, 'brandMap': brandMap}, 600)

    returnMap = {}    

    brandSubCategoryInfo = mc.get("brandSubCategoryInfo")

    if brandSubCategoryInfo is None:

        return []

    if type == "subCategoryInfo":

        if brandSubCategoryInfo.get('subCategoryMap') is None:

            return []

        subCategories = [int(subCategoryId) for subCategoryId in id_list.split('^')]

        subCategoryMap = brandSubCategoryInfo.get('subCategoryMap')

        for subCategory in subCategories:

            tempList = subCategoryMap.get(subCategory)

            print tempList

            if tempList is None:

                continue

            for v in tempList:

                if returnMap.has_key(v['brand_id']):

                    existing = returnMap.get(v['brand_id'])

                    existing['count'] = existing['count'] + v['count'] 

                else:

                    returnMap[v['brand_id']] = {'brand_id':v['brand_id'],'brand':v['brand'],'count':v['count']}

    elif type == "brandInfo":

        if brandSubCategoryInfo.get('brandMap') is None:

            return []

        brands = [int(brand_id) for brand_id in id_list.split('^')]

        brandMap = brandSubCategoryInfo.get('brandMap')

        for brand_id in brands:

            tempList = brandMap.get(brand_id)

            print tempList

            if tempList is None:

                continue

            for v in tempList:

                if returnMap.has_key(v['subCategoryId']):

                    existing = returnMap.get(v['subCategoryId'])

                    existing['count'] = existing['count'] + v['count'] 

                else:

                    returnMap[v['brand_id']] = {'subCategoryId':v['subCategoryId'],'subCategory':v['subCategory'],'count':v['count']}

    return returnMap.values()

    x = getBrandSubCategoryInfo(6,"15^17","subCategoryInfo")

    print "==============="

    print x