Subversion Repositories SmartDukaan

from xml.dom.minidom import parseString

import re

import urllib

import hashlib

import hmac

import base64

from time import strftime, gmtime

from requests import request

class MWS(object):

    URI = "/"

    VERSION = "2009-01-01"

    NS = ''

    def __init__(self, access_key, secret_key, merchant_id,

                 domain='https://mws-eu.amazonservices.com', uri="", version=""):

        self.access_key = access_key

        self.secret_key = secret_key

        self.merchant_id = merchant_id

        self.domain = domain

        self.uri = uri or self.URI

        self.version = version or self.VERSION

    def make_request(self, extra_data, action,method="GET", **kwargs):

        params = {

            'AWSAccessKeyId': self.access_key,

            'SignatureVersion': '2',

            'Timestamp': self.get_timestamp(),

            'Version': self.version,

            'SignatureMethod': 'HmacSHA256',

            'Action': action

        }

        if action=='GetLowestOfferListingsForSKU':

            params['ExcludeMe']='true'

        if action =='GetLowestOfferListingsForASIN':

            params['ItemCondition'] = 'New'

        params.update(extra_data)

        request_description = '&'.join(['%s=%s' % (k, urllib.quote(params[k], safe='-_.~').encode('utf-8')) for k in sorted(params)])

        signature = self.calc_signature(method, request_description)

        url = '%s%s?%s&Signature=%s' % (self.domain, self.uri, request_description, urllib.quote(signature))

        #headers = {'User-Agent': 'AmazonJavascriptScratchpad/1.0 (Language=Python)'}

        #headers.update(kwargs.get('extra_headers', {}))

        response = request(method, url)

        if action=='GetLowestOfferListingsForSKU':

            return self.parse_competitor_pricing_response(response)

        elif action=='GetMyPriceForSKU':

            return self.parse_my_pricing_response(response)

        elif action=='GetProductCategoriesForSKU':

            return self.parse_product_category_for_sku(response)

        elif action=='GetMatchingProductForId':

            return self.parse_product_attributes(response)

        elif action=='GetLowestOfferListingsForASIN':

            return self.parse_lowest_pricing_asins(response)

        elif action == 'GetMatchingProduct':

            return self.parse_get_matching_product(response)

        else:

            raise

    def parse_get_matching_product(self, response):

        spString = re.sub('<\?.*\?>','',response.text)

        spString = "<dom>" + spString + "</dom>"

        dom = parseString(spString)

        asinStatuses = dom.getElementsByTagName('GetMatchingProductResult')

        asinMap = {}

        for asinStatus in asinStatuses:

            #print asinStatus.attributes.items()

            status = asinStatus.attributes.items()[0][1]

            asin = asinStatus.attributes.items()[1][1]

            asinMap[asin] = status

        return asinMap

    def parse_product_category_for_sku(self,response):

        browseNodes = []

        #node = "" 

        spString = re.sub('<\?.*\?>','',response.text)

        spString = "<dom>" + spString + "</dom>"

        dom = parseString(spString)

        selfTag = dom.getElementsByTagName('Self')

        for element in selfTag:

            temp = []

            browsingNodes = element.getElementsByTagName('ProductCategoryName')

            for browsingNode in browsingNodes:

                temp.append(browsingNode.firstChild.nodeValue)

            browseNodes.append(temp)

        return browseNodes

    def parse_product_attributes(self,response):

        spString = re.sub('<\?.*\?>','',response.text)

        spString = "<dom>" + spString + "</dom>"

        dom = parseString(spString)

        sku = dom.getElementsByTagName("GetMatchingProductForIdResult")[0].attributes.items()[2][1]

        dimensions = dom.getElementsByTagName("ns2:PackageDimensions")

        for dimension in dimensions:

            height = dimension.getElementsByTagName("ns2:Height")[0].firstChild.nodeValue #Inch

            length = dimension.getElementsByTagName("ns2:Length")[0].firstChild.nodeValue #Inch

            width = dimension.getElementsByTagName("ns2:Width")[0].firstChild.nodeValue #Inch

            weight = dimension.getElementsByTagName("ns2:Weight")[0].firstChild.nodeValue #Pounds

            return {'sku':sku,'length':length,'width':width,'height':height,'weight':weight}

    def parse_lowest_pricing_asins(self,response):

        #GetLowestOfferListingsForASINResult

        spString = re.sub('<\?.*\?>','',response.text)

        spString = "<dom>" + spString + "</dom>"

        dom = parseString(spString)

        asinOffers = dom.getElementsByTagName('GetLowestOfferListingsForASINResult')

        asinMap = {}

        for asinOffer in asinOffers:

            asin = asinOffer.attributes.items()[1][1]

            asinMap[asin] = 0.0

            for offer in asinOffer.getElementsByTagName('LowestOfferListing'):

                price =  offer.getElementsByTagName('Amount')[0].firstChild.nodeValue

                asinMap[asin] = float(price) 

                break

        return asinMap

    def parse_competitor_pricing_response(self,response):

        spString = re.sub('<\?.*\?>','',response.text)

        spString = "<dom>" + spString + "</dom>"

        dom = parseString(spString)

        skuOffers = dom.getElementsByTagName('GetLowestOfferListingsForSKUResult')

        offerMap = {}

        otherInfo = {}

        for skuOffer in skuOffers:

            status = skuOffer.attributes.items()[0][1]

            sku = skuOffer.attributes.items()[1][1]

            info = []

            for offer in skuOffer.getElementsByTagName('LowestOfferListing'):

                temp = {}

                try:

                    if len(info)==3:

                        break

                    amount = offer.getElementsByTagName('Amount')[0].firstChild.nodeValue

                    temp['sellingPrice'] = float(amount)

                    temp['promoPrice'] = float(amount)

                    temp['fulfillmentChannel'] = offer.getElementsByTagName('FulfillmentChannel')[0].firstChild.nodeValue

                    try:

                        temp['shippingTime'] = (offer.getElementsByTagName('Max')[0].firstChild.nodeValue).replace('days','')

                    except:

                        temp['shippingTime'] = '0-0'

                    try:

                        temp['rating'] = (offer.getElementsByTagName('SellerPositiveFeedbackRating')[0].firstChild.nodeValue)

                        if temp['rating'] == 'Just Launched':

                            temp['rating'] = '100'

                        else:

                            str_rating = temp['rating'].replace('-',' ').replace('%','')

                            a =  str_rating.split()

                            l = []

                            for x in a:

                                if x.isdigit():

                                    l.append(x)

                            temp['rating'] = l[0]

                    except:

                        temp['rating'] = '0'

                except:

                    pass

                temp['notOurSku'] = True

                info.append(temp)

                offerMap[sku]=info

        for skuOffer in skuOffers:

            sku = skuOffer.attributes.items()[1][1]

            temp = {}

            temp['lowestMfnIgnored']=0.0

            temp['isLowestMfnIgnored']=False

            temp['lowestFba']=0.0

            temp['isLowestFba']=False

            temp['lowestMfn']=0.0

            temp['isLowestMfn']=False

            for offer in skuOffer.getElementsByTagName('LowestOfferListing'):

                try:

                    fulfillmentChannel = offer.getElementsByTagName('FulfillmentChannel')[0].firstChild.nodeValue

                    amount = offer.getElementsByTagName('Amount')[0].firstChild.nodeValue

                    try:

                        rating = (offer.getElementsByTagName('SellerPositiveFeedbackRating')[0].firstChild.nodeValue)

                        if rating == 'Just Launched':

                            rating = 100

                        else:

                            str_rating = rating.replace('-',' ').replace('%','')

                            a =  str_rating.split()

                            l = []

                            for x in a:

                                if x.isdigit():

                                    l.append(x)

                            rating = int(l[0])

                    except:

                        rating = 0

                    if fulfillmentChannel=='Merchant' and not temp['isLowestMfnIgnored'] and not temp['isLowestMfn'] and rating < 60:

                        temp['lowestMfnIgnored'] = float(amount)

                        temp['isLowestMfnIgnored']=True

                    elif fulfillmentChannel=='Merchant' and not temp['isLowestMfn'] and rating > 60:

                        temp['lowestMfn'] = float(amount)

                        temp['isLowestMfn']=True

                    elif fulfillmentChannel=='Amazon' and not temp['isLowestFba']:

                        temp['lowestFba'] = float(amount)

                        temp['isLowestFba']=True

                    else:

                        pass

                except:

                    pass

                otherInfo[sku]=temp

        return offerMap, otherInfo

    def parse_my_pricing_response(self,response):

        spString = re.sub('<\?.*\?>','',response.text)

        spString = "<dom>" + spString + "</dom>"

        dom = parseString(spString)

        skuOffers = dom.getElementsByTagName('GetMyPriceForSKUResult')

        skuMap = {}

        for skuOffer in skuOffers:

            try:

                status = skuOffer.attributes.items()[0][1]

                sku = skuOffer.attributes.items()[1][1]

                asin = skuOffer.getElementsByTagName('ASIN')[0].firstChild.nodeValue

                skuMap[sku]={}

            except:

                continue

            for offer in skuOffer.getElementsByTagName('Offer'):

                offerSku = offer.getElementsByTagName('SellerSKU')[0].firstChild.nodeValue

                fcChannel =  offer.getElementsByTagName('FulfillmentChannel')[0].firstChild.nodeValue

                if not (offerSku==sku and fcChannel=='AMAZON'):

                    continue

                temp = {}

                try:

                    promoPrice = offer.getElementsByTagName('LandedPrice')[0].getElementsByTagName('Amount')[0].firstChild.nodeValue

                    regularPrice = offer.getElementsByTagName('RegularPrice')[0].getElementsByTagName('Amount')[0].firstChild.nodeValue

                    temp['sellingPrice'] = float(regularPrice)

                    temp['promoPrice'] = float(promoPrice)

                    if promoPrice == regularPrice:

                        temp['promotion'] = False

                    else:

                        temp['promotion'] = True

                    temp['status'] = status

                    temp['asin'] = asin

                    temp['notOurSku'] = False

                    temp['fulfillmentChannel'] ='AMAZON'

                    temp['shippingTime'] =  '0-0'

                    temp['rating'] = '0'

                except:

                    pass

                skuMap[sku]=temp     

        return skuMap    

    def calc_signature(self, method, request_description):

        sig_data = method + '\n' + self.domain.replace('https://', '').lower() + '\n' + self.uri + '\n' + request_description

        return base64.b64encode(hmac.new(str(self.secret_key), sig_data, hashlib.sha256).digest())

    def get_timestamp(self):

        return strftime("%Y-%m-%dT%H:%M:%SZ", gmtime())

class Products(MWS):

    URI = '/Products/2011-10-01'

    VERSION = '2011-10-01'

    NS = '{https://mws-eu.amazonservices.com/Products/2011-10-01}'

    def get_my_pricing_for_sku(self, marketplaceid, skus):

        data = dict(SellerId=self.merchant_id, MarketplaceId=marketplaceid)

        num=0

        for sku in skus:

            data['SellerSKUList.SellerSKU.%d' % (num + 1)] = sku

            num+=1

        return self.make_request(data,'GetMyPriceForSKU')

    def get_competitive_pricing_for_sku(self, marketplaceid, skus):

        data = dict(SellerId=self.merchant_id, MarketplaceId=marketplaceid)

        num=0

        for sku in skus:

            data['SellerSKUList.SellerSKU.%d' % (num + 1)] = sku

            num+=1

        return self.make_request(data,'GetLowestOfferListingsForSKU')

    def get_product_category_for_sku(self, marketplaceid, sku):

        data = dict(SellerId=self.merchant_id, MarketplaceId=marketplaceid)

        data['SellerSKU'] = sku

        return self.make_request(data,'GetProductCategoriesForSKU')

    def get_product_attributes_for_sku(self, marketplaceid, skus):

        data = dict(SellerId=self.merchant_id, MarketplaceId=marketplaceid)

        num=0

        for sku in skus:

            data['IdList.Id.%d' % (num + 1)] = sku

            num+=1

        data['IdType'] = 'ASIN'

        return self.make_request(data,'GetMatchingProductForId')

    def get_competitive_pricing_for_asin(self, marketplaceid, asins):

        data = dict(SellerId=self.merchant_id, MarketplaceId=marketplaceid)

        num=0

        for asin in asins:

            data['ASINList.ASIN.%d' % (num + 1)] = asin

            num+=1

        return self.make_request(data,'GetLowestOfferListingsForASIN')

    def get_matching_product(self, marketplaceid, asins):

        data = dict(SellerId=self.merchant_id, MarketplaceId=marketplaceid)

        num=0

        for asin in asins:

            data['ASINList.ASIN.%d' % (num + 1)] = asin

            num+=1

        return self.make_request(data,'GetMatchingProduct')

def main():

    p = Products("AKIAII3SGRXBJDPCHSGQ", "B92xTbNBTYygbGs98w01nFQUhbec1pNCkCsKVfpg", "AF6E3O0VE0X4D")

    print p.get_matching_product("A21TJRUUN4KGV", ["B016QBT87W","B0168L23VS"])

if __name__=='__main__':

    main()