Subversion Repositories SmartDukaan

from elixir import *

import pymongo

import MySQLdb

import codecs

from shop2020.config.client.ConfigClient import ConfigClient

from shop2020.model.v1.catalog.impl import DataService

from shop2020.model.v1.catalog.impl.DataService import Item, Category

from shop2020.thriftpy.model.v1.catalog.ttypes import status

config_client = ConfigClient()

host = config_client.get_property('catalog_service_db_hostname')

host_mongo = config_client.get_property('staging_hostname')

host_warehouse = config_client.get_property('warehouse_service_server_host')

DataService.initialize(db_hostname=host)

categoryMap ={}

itemMap = {} 

con = None

BASE_URL = "http://saholic.com/"

AFF_TRACK = "?afid=109&utm_source=Idealo&utm_medium=affiliate_CPC"

xstr = lambda s: s or ""

db = MySQLdb.connect(host_warehouse,"root","shop2020","warehouse" )

cursor = db.cursor()

class __ItemInfo:

    def __init__(self, itemId, price, url, imageUrl, ean, shortDesc, brand, modelName, color, productGroup, catalogItemId, modelNumber):

        self.itemId = itemId

        self.price = price

        self.url = url

        self.imageUrl = imageUrl

        self.ean = ean

        self.shortDesc = shortDesc

        self.modelName = modelName

        self.color = color

        self.productGroup = productGroup

        self.catalogItemId = catalogItemId

        self.modelNumber = modelNumber

        self.brand = brand

def get_mongo_connection(host='localhost', port=27017):

    global con

    if con is None:

        print "Establishing connection %s host and port %d" %(host,port)

        try:

            con = pymongo.MongoClient(host, port)

        except Exception, e:

            print e

            return None

    return con

def generateCategoryMap():

    global categoryMap

    result = session.query(Category).all()

    for cat in result:

        categoryMap[cat.id] = [cat.parent_category_id ,cat.display_name]    

def populateActiveProducts():

    generateCategoryMap()

    items = session.query(Item).filter(Item.status == status.ACTIVE).all()

    try:

        for item in items:

            info = __ItemInfo(item.id, item.sellingPrice, None, None, None, None, item.brand, item.model_name, item.color, None, item.catalog_item_id, item.model_number)

            if item.category == 10006:

                info.productGroup = "Mobile Phones > "+item.brand

            elif item.category == 10010:

                info.productGroup = "Tablets > "+item.brand

            else:

                info.productGroup = (categoryMap.get(categoryMap.get(item.category)[0])[1]+" > "+categoryMap.get(item.category)[1]+" > "+item.brand).replace(","," ")

            itemMap[item.id] = info

    finally:

        session.close()

def populateContent():

    global itemMap

    for v in itemMap.itervalues():

        try:

            print "MOngo ",v.catalogItemId

            col = list(get_mongo_connection(host_mongo).CONTENT.siteContent.find({'_id':v.catalogItemId}))

            if len(col) ==0:

                continue

            else:

                v.url = col[0]['url']

                v.imageUrl = col[0]['defaultImageUrl']

                v.shortDesc = col[0]['keySpecs']

        except Exception as e:

            print e

def getEanInformation():

    global itemMap

    for v in itemMap.itervalues():

        sql = '''select itemNumber  from inventoryItem where itemId=%d limit 1''' % \

            (v.itemId)

        print sql

        cursor.execute(sql)

        data = cursor.fetchone()

        print data

        if (data is None or data[0] is None or len(data[0])==0):

            v.ean = ""

        else:

            v.ean = data[0]

def writeFile():

    f = codecs.open('IdealoProductFeed-V1','w','utf-8')

    f.write('Article number\tEAN (European article number) / GTIN (Global Trade Item Number)\tManufacturers code / number\t \

    Manufacturer\tProduct Name\tDescription\tProduct Group\tPrice INR\tDelivery status\tProduct URL\tPicture URL\tDelivery Costs\n')

    for v in itemMap.itervalues():

        if v.url is None:

            continue

        try:

            f.write(str(v.itemId)+'\t'+str(v.ean)+'\t'+""'\t'+v.brand+'\t'+getProductName(v)+'\t'+getShortDesc(v.shortDesc)+'\t'+v.productGroup+'\t'+str(v.price)+'\t'+'In Stock'+'\t'+BASE_URL+v.url+AFF_TRACK+'\t'+v.imageUrl+'\t'+'0.00'+'\n')

        except Exception as e:

            print e

            print "Exception in writing ",v.itemId

    f.close()

def getShortDesc(descArr):

    description = "" 

    if descArr is None or len(descArr) == 0:

        return description

    else:

        for desc in descArr:

            description += desc.replace(',',' ')

        return description

def getProductName(item):

    return (xstr(item.brand)+" "+xstr(item.modelName)+" "+xstr(item.modelNumber)+" "+xstr(item.color)).replace(",","")

def main():

    populateActiveProducts()

    populateContent()

    try:

        getEanInformation()

    finally:

        db.close()

        print "Db connection closed"

    writeFile()

if __name__=='__main__':

    main()