Subversion Repositories SmartDukaan

from walrus import *

from dtr.utils.utils import get_mongo_connection, REVERSE_SOURCE_MAP, CATEGORY_MAP, SUB_CATEGORY_MAP

import random

xstr = lambda s: s or ""

database = Database()

ac = database.autocomplete()

class Suggestion():

    def __init__(self, category_id, subCategoryId, suggestion, word):

        self.category_id = category_id

        self.subCategoryId = subCategoryId

        self.suggestion = suggestion

        self.word = word

def addData():

    items = get_mongo_connection().Catalog.MasterData.find({'source_id' : { "$in": REVERSE_SOURCE_MAP.keys() } },{'skuBundleId':1,'brand':1,'model_name':1,'brand':1,'category_id':1,'subCategoryId':1})

    catalogMap = {}

    for item in items:

        if not catalogMap.has_key(item['skuBundleId']):

            title = xstr(item['brand'])+" "+xstr(item['model_name'])

            catalogMap[item['skuBundleId']] = {'brand':item['brand'],'title':title,'category':item['category_id'],'subCategory':item['subCategoryId']}

    for skuBundleId, details in catalogMap.iteritems():

        if not ac.exists(obj_id=skuBundleId):

            print ac.store(obj_id=skuBundleId,title=details.get('title'),data={'skuBundleId':skuBundleId,'title':details.get('title'),'category_id':details.get('category'),'subCategoryId':details.get('subCategory'),'brand':details.get('brand')},obj_type='entry')

def autoComplete():

    print "++++++++++++++"

    c = 0

    for i in ac.search("iPhone Plus"):

        c = c+1

        print i

    print c

def filterSuggestions(categorySuggestion):

    returnObj = []

    for cat in ('Mobiles','Tablets'):

        v = categorySuggestion.get(CATEGORY_MAP.get(cat))

        if v is not None:

            returnObj.append(v[0].__dict__)

            select_range = 2 if len(v[1:]) > 2 else len(v[1:]) 

            random_list = random.sample(set(v[1:]), select_range)

            for random_suggestion in random_list:

                returnObj.append(random_suggestion.__dict__)

            categorySuggestion.pop(CATEGORY_MAP.get(cat))

    for v in categorySuggestion.itervalues():

        returnObj.append(v[0].__dict__)

        select_range = 2 if len(v[1:]) > 2 else len(v[1:]) 

        random_list = random.sample(set(v[1:]), select_range)

        for random_suggestion in random_list:

            returnObj.append(random_suggestion.__dict__)

    return returnObj

def getSuggestions(search_text):

    categorySuggestion = {}

    results = list(ac.search(search_text))

    if len(results) > 10:

        """Lets group data"""

        for i in results:

            if i.get('subCategoryId'):

                if not categorySuggestion.has_key(i.get('subCategoryId')):

                    suggestion_obj_primary = Suggestion(i.get('category_id'),i.get('subCategoryId'),search_text+" in "+SUB_CATEGORY_MAP.get(i.get('subCategoryId')), search_text)

                    suggestion_obj = Suggestion(i.get('category_id'),i.get('subCategoryId'),i.get('title'), i.get('title'))

                    categorySuggestion[i.get('subCategoryId')] = [suggestion_obj_primary,suggestion_obj]

                else:

                    suggestion_obj = Suggestion(i.get('category_id'),i.get('subCategoryId'),i.get('title'), i.get('title'))

                    categorySuggestion.get(i.get('subCategoryId')).append(suggestion_obj)

            else:

                if not categorySuggestion.has_key(i.get('category_id')):

                    suggestion_obj_primary = Suggestion(i.get('category_id'),i.get('subCategoryId'),search_text+" in "+CATEGORY_MAP.get(i.get('category_id')), search_text)

                    suggestion_obj = Suggestion(i.get('category_id'),i.get('subCategoryId'),i.get('title'), i.get('title'))

                    categorySuggestion[i.get('category_id')] = [suggestion_obj_primary, suggestion_obj]

                else:

                    suggestion_obj = Suggestion(i.get('category_id'),i.get('subCategoryId'),i.get('title'), i.get('title'))

                    categorySuggestion.get(i.get('category_id')).append(suggestion_obj)

        returnObj = filterSuggestions(categorySuggestion)    

    else:

        for i in results:

            returnObj.append({'suggestion':i.get('title'),'category_id':i.get('category_id'),'subCategoryId':i.get('subCategoryId')})

    return returnObj

if __name__ == "__main__":

    ac.flush()

    addData()

    for i in getSuggestions("apple"):

        print i