Subversion Repositories SmartDukaan

import pymongo
from datetime import datetime, timedelta, time, date
from shop2020.thriftpy.model.v1.order.ttypes import AmazonFCWarehouseLocation
from shop2020.utils.Utils import to_java_date, to_py_date
from shop2020.thriftpy.model.v1.order.ttypes import AmazonFbaSalesSnapshot, AmazonHourlySaleSnapshot, \
AmazonFbaOrderItem
import operator


con = None
prefix = {'FBA':0,'FBB':1,'FBG':2, 'FBD':3}

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 to_amazonFbaSaleSnapshot(saleSnapshot):
    amazonFbaSaleSnapshot = AmazonFbaSalesSnapshot(dateOfSale=0, item_id=0, totalOrderCount=0, amazonFbaInventory=0, isOutOfStock=False, salePrice=0, minFbaPrice=0, minMfnPrice=0, totalSale=0, promotionSale=0, promotionOrderCount=0, ourPrice=0, minFbaPriceSnapshotDate=0, minMfnPriceSnapshotDate=0, ourPriceSnapshotDate=0, salePriceSnapshotDate=0, fcLocation=None)
    if saleSnapshot is None:
        return amazonFbaSaleSnapshot
    amazonFbaSaleSnapshot.__dict__.update(saleSnapshot)
    return amazonFbaSaleSnapshot

def to_amazonFbaHourlySnapshot(saleSnapshot):
    amazonHourlySaleSnapshot = AmazonHourlySaleSnapshot(snapshotTime=0, item_id=0, totalOrderCount=0, amazonFbaInventory=0, isOutOfStock=False,  totalSale=0, promotionSale=0, promotionOrderCount=0,  fcLocation=None)
    if saleSnapshot is None:
        return amazonHourlySaleSnapshot
    amazonHourlySaleSnapshot.__dict__.update(saleSnapshot)
    return amazonHourlySaleSnapshot

def to_amazonFbaOrderItem(orderData):
    amazonFbaOrderItem = AmazonFbaOrderItem(amazonOrderId=None, purchaseDate=None, orderStatus=None, item_id=None, fcLocation=None, totalAmount=None, promotionDiscount=None, quantity=None)
    if orderData is None:
        return amazonFbaOrderItem
    amazonFbaOrderItem.__dict__.update(orderData)
    return amazonFbaOrderItem 

def getAmazonFbaSalesLatestSnapshotForItemLocationWise(item_id,fcLocation):
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonDailySaleSnapshot
    list = []
    list.append({"dateOfSale": to_java_date(datetime.combine((datetime.now()-timedelta(days=1)).date(), time(00,00,00)))})
    list.append({"item_id":item_id})
    list.append({"fcLocation":AmazonFCWarehouseLocation._VALUES_TO_NAMES.get(fcLocation)})
    r = collection.find({"$and":list}).sort([('dateOfSale',pymongo.DESCENDING)]).limit(1)
    if r.count() > 0:
        return to_amazonFbaSaleSnapshot(r.next())
    else:
        return to_amazonFbaSaleSnapshot(None)

def getInStockAmazonFbaSalesLatestSnapshotForItemLocationWise(item_id,fcLocation,outOfStockTime):
    fbasalessnapshot = []
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonDailySaleSnapshot
    toDate = datetime.now()-timedelta(days = 1)
    fromDate = outOfStockTime
    list = []
    list.append({"item_id":item_id})
    list.append({"fcLocation":AmazonFCWarehouseLocation._VALUES_TO_NAMES.get(fcLocation)})
    query = {}
    query['$gt'] = fromDate 
    query['$lte'] = to_java_date(datetime.combine(toDate.date(), time(00,00,00)))
    list.append({"dateOfSale":query})
    cursor = collection.find({"$and":list}).sort([('dateOfSale',pymongo.ASCENDING)])
    totalSale = 0
    noOfDays = 0
    for sale in cursor:
        totalSale = totalSale + sale.get('totalOrderCount')
        noOfDays = noOfDays + 1 
    return totalSale, noOfDays 
        
def getAmazonFbaSalesSnapshotForDays(interval):
    fbasalessnapshot = []
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonDailySaleSnapshot
    toDate = datetime.now()-timedelta(days = 1)
    fromDate = toDate-timedelta(days = interval)
    query = {}
    query['$gte'] = to_java_date(datetime.combine(fromDate.date(), time(00,00,00)))
    query['$lte'] = to_java_date(datetime.combine(toDate.date(), time(00,00,00)))
    cursor = collection.find({"dateOfSale":query}).sort([('dateOfSale',pymongo.ASCENDING)])
    for sale in cursor:
        fbasalessnapshot.append(to_amazonFbaSaleSnapshot(sale))
    return fbasalessnapshot

def getLastOutOfStock(item_id,fcLocation):
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonDailySaleSnapshot
    list = []
    list.append({"item_id":item_id})
    list.append({"fcLocation":AmazonFCWarehouseLocation._VALUES_TO_NAMES.get(fcLocation)})
    list.append({"isOutOfStock":True})
    r = collection.find({"$and":list}).sort([('dateOfSale',pymongo.DESCENDING)]).limit(1)
    if r.count() > 0:
        lastOutOfStock = to_amazonFbaSaleSnapshot(r.next())
    else:
        lastOutOfStock = to_amazonFbaSaleSnapshot(None)
    if lastOutOfStock.dateOfSale == 0:
        return lastOutOfStock, 0, 0
    else:
        totalSale, noOfDays =getInStockAmazonFbaSalesLatestSnapshotForItemLocationWise(item_id,fcLocation,lastOutOfStock.dateOfSale)
        return lastOutOfStock, totalSale, noOfDays 
    
def getLatestHourlySnapshot():
    fbaHourlySnapshot = []
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonHourlySaleSnapshot
    r = collection.find().sort([('snapshotTime',pymongo.DESCENDING)]).limit(1)
    lastestSnapshotTime = r.next().get('snapshotTime')
    cursor = collection.find({"snapshotTime":{"$in":[lastestSnapshotTime]}})
    for sale in cursor:
        fbaHourlySnapshot.append(to_amazonFbaHourlySnapshot(sale))
    return fbaHourlySnapshot

def getAmazonOrderData(startDate, endDate, offset, limit):
    fbaOrderData = []
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonSaleData
    list = []
    query = {}
    list.append({"orderStatus":"Shipped"})
    query['$gte'] = to_java_date(startDate)
    query['$lte'] = to_java_date(endDate)
    list.append({"purchaseDate":query})
    print list
    if offset is None and limit is None:
        cursor = collection.find({"$and":list}).sort([('purchaseDate',pymongo.ASCENDING)])
        print cursor.count()
    else:
        cursor = collection.find({"$and":list}).skip(offset).limit(limit).sort([('purchaseDate',pymongo.ASCENDING)])
    for orderData in cursor:
        fbaOrderData.append(to_amazonFbaOrderItem(orderData))
    return fbaOrderData

def getDistinctItemsFromOrderData(startDate, endDate):
    item_ids = []
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonSaleData
    list = []
    query = {}
    list.append({"orderStatus":"Shipped"})
    query['$gte'] = to_java_date(startDate)
    query['$lte'] = to_java_date(endDate)
    list.append({"purchaseDate":query})
    cursor = collection.find({"$and":list}).distinct('item_id')
    for item_id in cursor:
        item_ids.append(item_id)
    return item_ids

def getOrderByAmazonOrderIdAndSku(amazonOrderId, sku):
    list = []
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonSaleData
    list.append({"amazonOrderId": amazonOrderId})
    list.append({"item_id":int(sku[3:])})
    list.append({"fcLocation":AmazonFCWarehouseLocation._VALUES_TO_NAMES.get(prefix.get(sku[0:3]))})
    r = collection.find({"$and":list})
    if r.count() > 1:
        raise
    elif r.count()==0:
        raise
    else:
        return to_amazonFbaOrderItem(r.next())
def getAmazonOrderDataByItemId(startDate, endDate, offset, limit, item_id):
    fbaOrderData = []
    collection = get_mongo_connection().AmazonSaleSnapshot.AmazonSaleData
    list = []
    query = {}
    list.append({"item_id":item_id})
    query['$gte'] = to_java_date(startDate)
    query['$lte'] = to_java_date(endDate)
    list.append({"purchaseDate":query})
    print list
    if offset is None and limit is None:
        cursor = collection.find({"$and":list}).sort([('purchaseDate',pymongo.ASCENDING)])
        print cursor.count()
    else:
        cursor = collection.find({"$and":list}).skip(offset).limit(limit).sort([('purchaseDate',pymongo.ASCENDING)])
    for orderData in cursor:
        fbaOrderData.append(to_amazonFbaOrderItem(orderData))
    return fbaOrderData



def dict_nlargest(d,n):
    import heapq
    return heapq.nlargest(n ,d, key = lambda k: d[k])


def main():
    endDate = (datetime.now()).replace(hour=0, minute=0, second=0, microsecond=0)
    print endDate
    startDate = (datetime.now() -timedelta(days=30)).replace(hour=0, minute=0, second=0, microsecond=0)
    print startDate
    orderItemData = getAmazonOrderDataByItemId(startDate, endDate,None,None,16914)
    priceSku = {}
    print len(orderItemData)
    count = 0
    for x in orderItemData:
        if x.promotionDiscount > 0 or x.orderStatus == 'Cancelled':
            count +=1
            continue
        print x.totalAmount
        print x.quantity
        print "*************"
        price = x.totalAmount / x.quantity
        if priceSku.has_key(price):
            qty = priceSku.get(price)
            priceSku[price] = qty + x.quantity
        else:
            priceSku[price] = x.quantity
            
    print priceSku
    print "promotion order count ",count
    newdict = {}
    for key, value in sorted(priceSku.iteritems(), key=lambda (k,v): (v,k),reverse=True):
        print "%s: %s" % (key, value)
        rng = str(int(key)-25)+'-'+str(int(key))
        print rng
        total_qty = 0
        min = 0
        max = 0
        for x in range(int(key-25),int(key+1)):
            qty = priceSku.get(x)
            if qty is not None:
                if min==0:
                    min = x
                if max < x:
                    max = x 
                total_qty = qty + total_qty
        newdict[str(min)+'-'+str(max)] = total_qty 
    print newdict
    
    d = {}
    
    for key, value in sorted(priceSku.iteritems(), key=lambda (k,v): (v,k),reverse=True):
        print "%s: %s" % (key, value)
        rng = str(int(key))+'-'+str(int(key)+26)
        print rng
        total_qty = 0
        min = 0
        max = 0
        for x in range(int(key),int(key+26)):
            qty = priceSku.get(x)
            if qty is not None:
                if min==0:
                    min = x
                if max < x:
                    max = x 
                total_qty = qty + total_qty
        d[str(min)+'-'+str(max)] = total_qty 
    print d
        
        
        
        
        
        
        
        
        
        
    
    
            
    
if __name__ == "__main__":
    main()