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, category_name):

        self.category_id = category_id

        self.subCategoryId = subCategoryId

        self.suggestion = suggestion

        self.category_name = category_name

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 = 5 if len(v[1:]) > 5 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 = 3 if len(v[1:]) > 3 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,SUB_CATEGORY_MAP.get(i.get('subCategoryId')))

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

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

                else:

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

                    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, CATEGORY_MAP.get(i.get('category_id')))

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

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

                else:

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

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

        returnObj = filterSuggestions(categorySuggestion)    

    else:

        for i in results:

            category_name = None

            if i.get('category_id') == 6:

                category_name = SUB_CATEGORY_MAP.get(i.get('subCategoryId'))

            elif i.get('category_id') in (3,5):

                category_name = CATEGORY_MAP.get(i.get('category_id'))

            else:

                pass

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

    return returnObj

def main():

    print datetime.datetime.now()

    print getSuggestions("iph")

    print datetime.datetime.now()

if __name__ == "__main__":

    main()