Subversion Repositories SmartDukaan

from bson.json_util import dumps

from datetime import datetime

from dtr.config import PythonPropertyReader

from sqlalchemy.sql.expression import or_

import falcon

import traceback

global RETAILER_DETAIL_CALL_COUNTER

RETAILER_DETAIL_CALL_COUNTER = 0

DEALER_RETRY_FACTOR = int(PythonPropertyReader.getConfig('DEALER_RETRY_FACTOR'))

class RetailerDetail():

    global RETAILER_DETAIL_CALL_COUNTER

    def getRetryRetailer(self,failback=False):

        retailer = None

        status = 'fretry' if self.callType == 'followup' else 'retry'

        try:

            retailer = session.query(Retailers).filter_by(status=status).filter(Retailers.next_call_time<=datetime.now()).order_by(Retailers.call_priority).order_by(Retailers.next_call_time).with_lockmode("update").first()

            if retailer is not None:

                print "getRetryRetailer", retailer.id

            else:

                if failback:

                    retailer = self.getNewRetailer(False)

            if status=='fresh':

                retailer.status = 'assigned'

            else:

                retailer.status = 'fassigned'

            retailer.retry_count += 1

            session.commit()

        finally:

            session.close()

        return retailer

    def getNewRetailer(self,failback=True):

        retry = True

        retailer = None 

        try:

            while(retry):

                print "Calltype", self.callType

                status=self.callType

                query = session.query(Retailers).filter(Retailers.status==status)

                if status=='fresh':

                    query = query.filter_by(is_or=False, is_std=False).filter(or_(Retailers.agent_id==self.agentId, Retailers.agent_id==None)).order_by(-Retailers.agent_id)

                retailer = query.with_lockmode("update").first()

                if retailer is not None:

                    print "retailer", retailer.id

                    if status=="fresh":

                        userquery = session.query(Users)

                        if retailer.contact2 is not None:

                            userquery = userquery.filter(Users.mobile_number.in_([retailer.contact1,retailer.contact2]))

                        else:

                            userquery = userquery.filter_by(mobile_number=retailer.contact1)

                        user = userquery.first()

                        if user is not None:

                            retailer.status = 'alreadyuser'

                            print "retailer.status", retailer.status

                            session.commit()

                            continue

                        retailer.status = 'assigned'

                    else:

                        retailer.status = 'fassigned'

                    print "Found Retailer", retailer.id, "with status", status,"assigned to", self.agentId

                    retailer.retry_count = 0

                    session.commit()

                    print "-"*50

                    print "Found Retailer", retailer.id, "with status", status,"assigned to", self.agentId

                    retry=False

                else:

                    print "No fresh/followup retailers found"

                    if failback:

                        retailer = self.getRetryRetailer(False)

                    retry=False

        except:

            print traceback.print_exc()

        finally:

            session.close()

        return retailer

    def on_get(self, req, resp, agentId, callType):

        global RETAILER_DETAIL_CALL_COUNTER

        RETAILER_DETAIL_CALL_COUNTER += 1

        print "RETAILER_DETAIL_CALL_COUNTER", RETAILER_DETAIL_CALL_COUNTER

        print "I am here"

        self.agentId = int(agentId)

        self.callType = callType

        retryFlag = False 

        agentId = int(agentId)

        if RETAILER_DETAIL_CALL_COUNTER % DEALER_RETRY_FACTOR ==0:

            retryFlag=True

        if retryFlag:

            retailer = self.getRetryRetailer()

        else:

            retailer = self.getNewRetailer()

        if retailer is None:

            resp.body = "{\"result\":\"{}\"}"

        else:

            resp.body = json.dumps(todict(getRetailerObj(retailer)), encoding='utf-8')

    def on_post(self, req, resp, agentId, callType):

        self.agentId = int(agentId)

        self.callType = callType

        jsonReq = json.loads(req.stream.read(), encoding='utf-8')

        resp.body = json.dumps(jsonReq, encoding='utf-8')

def todict(obj, classkey=None):

    if isinstance(obj, dict):

        data = {}

        for (k, v) in obj.items():

            data[k] = todict(v, classkey)

        return data

    elif hasattr(obj, "_ast"):

        return todict(obj._ast())

    elif hasattr(obj, "__iter__"):

        return [todict(v, classkey) for v in obj]

    elif hasattr(obj, "__dict__"):

        data = dict([(key, todict(value, classkey)) 

            for key, value in obj.__dict__.iteritems() 

            if not callable(value) and not key.startswith('_')])

        if classkey is not None and hasattr(obj, "__class__"):

            data[classkey] = obj.__class__.__name__

        return data

    else:

        return obj

def getRetailerObj(retailer):

    obj = Mock()

    obj.title = retailer.title

    obj.contact1 = retailer.contact1

    obj.contact2 = retailer.contact2

    obj.address = retailer.address

    obj.id = retailer.id

    obj.scheduled = (retailer.call_priority == 'user_initiated')

    return obj

class Mock(object):

    pass

def main():

    ret = RetailerDetail()

    resp = Mock()

    ret.on_get(None, resp, 1, 'fresh')

if __name__ == '__main__':

        main()