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')})
    
    return returnObj


def main():
    pass

if __name__ == "__main__":
    main()