Subversion Repositories SmartDukaan

from shop2020.utils.Utils import to_py_date, to_java_date

from datetime import datetime

from elixir import *

from dtr.storage import DataService

from dtr.storage.DataService import Orders, Users, CallHistory

from sqlalchemy.sql.expression import func

con = None

dateWiseOrderMap = {}

weekWiseOrderMap = {}

monthWiseOrderMap = {}

orderIds = []

cutOff = 1425839400

#cutOff = 1425234600

oneDay = 86400

monthCutOff = 1425148200

weekCutOff = 1425839400

#weekCutOff = 1425234600

DataService.initialize()

    global z

    worksheet.write(row, 6, 'DSO', boldStyle)   

    worksheet.write(row, 7, 'DTV', boldStyle)

    row=0

    z=0    

    for x in sorted(dateWiseOrderMap):

        row += 1

        column = 6

        worksheet.write(row,column,dateWiseOrderMap.get(x).count)

        column += 1

    row =0

    for x in sorted(dateWiseOrderMap):

        row += 1

        column = 7

        worksheet.write(row,column,dateWiseOrderMap.get(x).value)

        column += 1    

    #sendmail(["manas.kapoor@shop2020.in"], "", TMP_FILE, SUBJECT)

    worksheet.write(row, 6, 'WTS', boldStyle)

    worksheet.write(row, 7, 'WTV', boldStyle)

    row = 0        

    for x in sorted(weekWiseOrderMap):

        row += 1

        column = 6

        worksheet.write(row,column,weekWiseOrderMap.get(x).count)

        column+=1

    row = 0

    for x in sorted(weekWiseOrderMap):

        row += 1

        column = 7

        worksheet.write(row,column,weekWiseOrderMap.get(x).value)

        column+=1    

    #MAILTO = ['manas.kapoor@saholic.com']

class __Order:

    def __init__(self, count, value):

        self.count = count

        self.value = value

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 populateOrderMap():

    global dateWiseOrderMap

    allOrders = get_mongo_connection().Dtr.merchantOrder.find({'createdOnInt':{"$gte":cutOff}}).sort([('createdOnInt',pymongo.ASCENDING)])

    for orders in allOrders:

        if orders.get('orderId') not in order_ids:

            continue

        cdate = ((to_py_date(orders.get('createdOnInt') * 1000)))

        millisec = to_java_date(datetime(cdate.year, cdate.month, cdate.day))

        if dateWiseOrderMap.has_key(millisec):

            orderObj = dateWiseOrderMap.get(millisec)

            q, c = getSubOrderQuantity(orders.get('subOrders'))

            orderObj.count += q

            orderObj.value += c

        else:

            orderObj = __Order(None, None)

            q, c = getSubOrderQuantity(orders.get('subOrders'))

            orderObj.count = q

            orderObj.value = c 

            dateWiseOrderMap[millisec] = orderObj

def getSubOrderQuantity(subOrders):

    q = 0

    c = 0

    if subOrders is None:

        return q, c

    for subOrder in subOrders:

        q = q + int(subOrder.get('quantity'))

        try:

            c = c + float(subOrder.get('amountPaid').encode('utf-8'))

        except AttributeError:

            c = c + float(subOrder.get('amountPaid'))

    return q, c

def populateWeekWiseMap():

    global weekCutOff

    while(True):

        quantity, amount = 0 , 0

        for i in xrange(weekCutOff , (weekCutOff * 7)+1):

            orderObj = dateWiseOrderMap.get(i*1000)

            quantity += orderObj.count

            amount += orderObj.value

            i = i + oneDay

        weekWiseOrderMap[weekCutOff * 1000] = __Order(quantity, amount)

        weekCutOff = weekCutOff * 1000 * 7 + oneDay

        if weekCutOff >= to_py_date(datetime.now()):

            break

def populateWeekWiseMap1():

    global weekCutOff

    while(True):

        #print weekCutOff *1000

        #print to_java_date(datetime.now())

        #print "**********************"

        if weekCutOff *1000 >= to_java_date(datetime.now()):

            #print "Breaking outer while"

            break

        init = weekCutOff

        breakPoint = weekCutOff + (6 * oneDay)

        quantity, amount = 0 , 0

        while(True):

            #print to_py_date(weekCutOff*1000)

            #print "weekCutOff ",weekCutOff

            #print "breakPoint ",breakPoint

            orderObj = dateWiseOrderMap.get(weekCutOff * 1000)

            if orderObj is not None:

                quantity += orderObj.count

                amount += orderObj.value

            weekCutOff = weekCutOff + oneDay

            if weekCutOff > breakPoint:

                weekWiseOrderMap[init * 1000] = __Order(quantity, amount)

                #print "Breaking inner while"

                break 

def populateMonthWiseMap():

    global monthCutOff

    while(True):

        quantity, amount = 0 , 0

        for i in xrange(monthCutOff , (monthCutOff * 30)+1):

            orderObj = dateWiseOrderMap.get(i*1000)

            quantity += orderObj.count

            amount += orderObj.value

            i = i + oneDay

        weekWiseOrderMap[monthCutOff * 1000] = __Order(quantity, amount)

        monthCutOff = monthCutOff * 1000 * 30 + oneDay

        if monthCutOff >= to_py_date(datetime.now()):

            break

def populateMonthWiseMap1():

    global weekCutOff

    while(True):

        print weekCutOff *1000

        print to_java_date(datetime.now())

        print "**********************"

        if weekCutOff *1000 >= to_java_date(datetime.now()):

            print "Breaking outer while"

            break

        init = weekCutOff

        breakPoint = weekCutOff + (6 * oneDay)

        quantity, amount = 0 , 0

        while(True):

            print to_py_date(weekCutOff*1000)

            print "weekCutOff ",weekCutOff

            print "breakPoint ",breakPoint

            orderObj = dateWiseOrderMap.get(weekCutOff * 1000)

            if orderObj is None:

                print "None for ", to_py_date(weekCutOff * 1000)

            if orderObj is not None:

                quantity += orderObj.count

                amount += orderObj.value

            weekCutOff = weekCutOff + oneDay

            if weekCutOff > breakPoint:

                weekWiseOrderMap[init * 1000] = __Order(quantity, amount)

                print "Breaking inner while"

                break 

def populateValidOrders():

    global order_ids

    allOrders = session.query(Orders.id).join((Users,Orders.user_id==Users.id)).filter(~(func.lower(Users.referrer)).like('emp%')).filter(Orders.status=='ORDER_CREATED').all()

    order_ids = list(zip(*allOrders)[0])

    populateValidOrders()

    populateOrderMap()

    populateWeekWiseMap1()