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 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))

    returnObj = []

    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'),'word':i.get('title')})

    return returnObj

def main():

    pass

if __name__ == "__main__":

    main()