Subversion Repositories SmartDukaan

'''

Created on Feb 12, 2015

@author: amit

'''

from dtr import main

from dtr.storage import Mongo, Mysql

import falcon

import json

import urlparse

import re

class StoreOrder():

    def on_post(self, req, resp):

        try:

            string1 = req.stream.read()

            req_json = json.loads(string1, encoding='utf-8')

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')

        store = main.getStore(int(req_json['sourceId']))

        result = store.parseOrderRawHtml(int(req_json['orderId']), req_json['subTagId'], int(req_json['userId']), req_json['rawHtml'], req_json['orderSuccessUrl'])

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

    def on_get(self, req, resp, userId):

        page = req.get_param_as_int("page")

        window = req.get_param_as_int("window")

        string1 = req.get_param("searchMap")

        searchMap={}

        if string1 is None or string1 == "":

            pass

        else:

            try:

                searchMap = json.loads(string1, encoding='utf-8')

            except:

                raise falcon.HTTPError(falcon.HTTP_400,

                    'Malformed JSON',

                    'Could not decode the request body. The '

                    'JSON was incorrect.')

        result = Mongo.getMerchantOrdersByUser(int(userId), page, window, searchMap)

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

class Orders():

    def on_get(self, req, resp):

        page = req.get_param_as_int("page")

        window = req.get_param_as_int("window")

        string1 = req.get_param("searchMap")

        searchMap={}

        if string1 is None or string1 == "":

            pass

        else:

            try:

                searchMap = json.loads(string1, encoding='utf-8')

            except:

                raise falcon.HTTPError(falcon.HTTP_400,

                    'Malformed JSON',

                    'Could not decode the request body. The '

                    'JSON was incorrect.')

        result = Mongo.getMerchantOrdersByUser(None, page, window, searchMap)

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

class Track():

    def on_post(self, req, resp, userId):

        try:

            string1 = req.stream.read()

            req_obj = urlparse.parse_qs(string1)

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')

        try:

            store = main.getStore(req.get_param_as_int("storeId"))

            if req.get_param_as_int("storeId") == 1:

                result = store.trackOrdersForUser(int(userId),req_obj['url'][0],req_obj['html'][0])

                resp.body = json.dumps({'result':result}, encoding='utf-8')

            else:

                resp.body = json.dumps({'result':'NOT_IMPLEMENTED'}, encoding='utf-8')

        except:

            resp.body = json.dumps({'result':'PARSE_ERROR'}, encoding='utf-8')

        print "Returned from track POST", resp.body

        resp.body = json.dumps({'result':'PARSE_ERROR'}, encoding='utf-8')

    def on_get(self, req, resp, userId):

        try:

            storeId = req.get_param_as_int("storeId")

            result = main.getStore(storeId).getTrackingUrls(int(userId))

            resp.body = json.dumps({'result':result}, encoding='utf-8')

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')

class Refunds():

    def on_get(self, req, resp, userId):

        try:

            page = req.get_param_as_int("page")

            window = req.get_param_as_int("window")

            result = Mongo.getRefunds(int(userId),page,window)

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

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')

class Transactions():

    def on_get(self, req, resp):

        try:

            page = req.get_param_as_int("page")

            window = req.get_param_as_int("window")

            user = req.get_param("user")

            status = req.get_param("status")

            if user == "" :

                user = None

            if status == "" :

                status = None

            result = Mysql.getOrders(page, window,status,user)

            resp.body = result

            resp.content_type = 'text/html'

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')

class SnapShot():

    def on_get(self, req, resp):

        try:

            resp.content_type = 'text/html'

            user = req.get_param_as_int("user")

            file = req.get_param("file") 

            if  file is not None:

                #fetch the page and return html

                with open ("/AmazonTrack/User%d/%s"%(user, file), "r") as myfile:

                    resp.body=myfile.read() 

            else:

                #return user specific urls

                resp.body=getUserUrls(user)

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')

def getUserUrls(user):

    pass

class RawHtml():

    def on_get(self, req, resp, orderId):

        try:

            result = Mysql.getOrderHtml(orderId)

            result = re.subn(r'(src|href|style)=\s?(\'|").*?\2(?s)', '', re.subn(r'<(script|style).*?</\1>(?s)', '', result)[0])[0]

            resp.body = result

            resp.content_type = 'text/html'

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')

class PendingRefunds():

    def on_get(self, req, resp,userId):

        try:

            result = Mongo.getPendingRefunds(int(userId))

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

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')

class PendingCashBacks():

    def on_get(self, req, resp,userId):

        try:

            result = Mongo.getPendingCashbacks(int(userId))

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

        except ValueError:

            raise falcon.HTTPError(falcon.HTTP_400,

                'Malformed JSON',

                'Could not decode the request body. The '

                'JSON was incorrect.')