Subversion Repositories SmartDukaan

'''

Created on 26-Aug-2011

@author: Varun Gupta

'''

import json, sys, os

cmd_folder = os.path.dirname(os.path.abspath(os.environ["HOME"] + "/code/trunk/PyProj/src/shop2020/"))

if cmd_folder not in sys.path:

    sys.path.insert(0, cmd_folder)

from shop2020.clients.CatalogClient import CatalogClient

def isValidRule(rule):

    try:

        if rule is None:

            return False

        elif rule['url'] is None:

            return False

        elif rule['source'] is None:

            return False

        else:

            return True

    except KeyError:

        return False

def extractBrandAndName(full_name):

#    brands = ('Micromax', 'BlackBerry', 'Blackberry', 'Motorola', 'Alcatel', 'Sony Ericsson', 'Apple', 'Spice', 'Nokia', 'HTC', 'Samsung', 'LG', 'Dell')

    try:

        client = CatalogClient().get_client()

        brands = client.getAllBrandsByCategory(10001)

        brands.append('Blackberry') #To resolve issue of 'BlackBerry' and 'Blackberry'

        print brands

        for brand in brands:

            if full_name.startswith(brand):  return (brand, full_name.replace(brand, '').strip())

    except Exception as e:

        print e

    return ("", full_name)

def getItemsWithTopScore(items):

    filterd_items = []

    top_score = -1.0

    for item in items:

        if item['score'] >= top_score:

            filterd_items.append(item)

            top_score = item['score'] 

        else:

            return filterd_items

    return filterd_items

def isPriceSame(items):

    for i in range(0, items.__len__() - 1):

        if items[i]['price'] != items[i + 1]['price']:    return False

    return True

def getProductClusters(products):

    '''

    Receives a list of products (returned from search results) &

    returns a clustered dictionary, where products are grouped by

    the 'source'

    '''

    clustered_results = {'flipkart': [], 'homeshop18': [], 'infibeam': [], 'letsbuy': []}

    for product in products:

        clustered_results[product['source']].append(product)

    return clustered_results

def getFilteredClustersWithTopScores(product_clusters):

    filtered_cluster = {}

    for source, products in product_clusters.iteritems():

        filtered_cluster[source] = getItemsWithTopScore(products)

    return filtered_cluster

def removePriceFormatting(price_string):

    return price_string.replace('Rs.', '').replace(',', '').strip()

def getDisplayInfo(filtered_cluster):

    display_info = {'flipkart': {}, 'homeshop18': {}, 'infibeam': {}, 'letsbuy': {}}

    for source, products in filtered_cluster.iteritems():

        if len(products) > 0:

            if isPriceSame(products):

                display_info[source]['price'] = products[0]['price']

                display_info[source]['data'] = None

                display_info[source]['url'] = products[0]['url']

                display_info[source]['text'] = removePriceFormatting(products[0]['price'])

            else:

                display_info[source]['price'] = None

                display_info[source]['data'] = json.dumps(products)

                display_info[source]['url'] = None

                display_info[source]['text'] = 'Conflict'

        else:

            display_info[source]['price'] = None

            display_info[source]['data'] = None

            display_info[source]['url'] = None

            display_info[source]['text'] = 'Not Found'

    return display_info