Subversion Repositories SmartDukaan

/*******************************************************************************

 *  Copyright 2009 Amazon Services.

 *  Licensed under the Apache License, Version 2.0 (the "License"); 

 *  

 *  You may not use this file except in compliance with the License. 

 *  You may obtain a copy of the License at: http://aws.amazon.com/apache2.0

 *  This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 

 *  CONDITIONS OF ANY KIND, either express or implied. See the License for the 

 *  specific language governing permissions and limitations under the License.

 * ***************************************************************************** 

 *

 *  Marketplace Web Service Java Library

 *  API Version: 2009-01-01

 *  Generated: Wed Feb 18 13:28:48 PST 2009 

 * 

 */

package com.amazonaws.mws.samples;

import java.io.ByteArrayInputStream;

import in.shop2020.model.v1.catalog.Amazonlisted;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.IOException;

import java.io.PrintWriter;

import java.io.StringWriter;

import java.math.BigDecimal;

import java.math.BigInteger;

import java.nio.channels.FileChannel;

import java.security.DigestInputStream;

import java.security.MessageDigest;

import java.security.NoSuchAlgorithmException;

import java.util.Arrays;

import java.util.Calendar;

import java.util.Date;

import java.util.GregorianCalendar;

import java.util.HashMap;

import java.util.Iterator;

import java.util.List;

import java.util.ArrayList;

import java.util.Map;

import java.util.Random;

import java.util.TimeZone;

import javax.xml.bind.JAXBContext;

import javax.xml.bind.JAXBException;

import javax.xml.bind.Marshaller;

import javax.xml.datatype.DatatypeConfigurationException;

import javax.xml.datatype.DatatypeConstants;

import javax.xml.datatype.DatatypeFactory;

import javax.xml.datatype.XMLGregorianCalendar;

import javax.xml.parsers.ParserConfigurationException;

import com.amazonaws.mws.*;

import com.amazonaws.mws.model.*;

import com.amazonaws.mws.samples.ProcessFeed;

import com.amazonaws.mws.samples.CheckFeedSubmission;

import com.amazonaws.mws.mock.MarketplaceWebServiceMock;

import org.apache.commons.codec.binary.*;

import in.shop2020.feeds.products.AmazonEnvelope;

import in.shop2020.feeds.products.BaseCurrencyCodeWithDefault;

import in.shop2020.feeds.products.CE;

import in.shop2020.feeds.products.HandheldOrPDA;

import in.shop2020.feeds.products.Header;

import in.shop2020.feeds.products.Inventory;

import in.shop2020.feeds.products.Listings;

import in.shop2020.feeds.products.OverrideCurrencyAmount;

import in.shop2020.feeds.products.Price;

import in.shop2020.feeds.products.Product;

import in.shop2020.feeds.products.StandardProductID;

import in.shop2020.feeds.products.Price.Sale;

import in.shop2020.model.v1.catalog.CatalogService.Client;

import in.shop2020.model.v1.catalog.CatalogServiceException;

import in.shop2020.model.v1.catalog.Item;

import in.shop2020.model.v1.inventory.InventoryServiceException;

import in.shop2020.model.v1.inventory.ItemInventory;

import in.shop2020.thrift.clients.CatalogClient;

import in.shop2020.thrift.clients.InventoryClient;

import in.shop2020.thrift.clients.WarehouseClient;

import in.shop2020.utils.GmailUtils;

import org.apache.thrift.TException;

import org.apache.thrift.transport.TTransportException;

import org.xml.sax.SAXException;

/**

 * 

 * Submit Feed Samples

 * 

 * 

 */

public class SubmitFBGPricingFeedAndFetchResults {

    /**

     * Just add a few required parameters, and try the service Submit Feed

     * functionality

     * 

     * @param args

     *            unused

     *            

     */

    /**

     * @param args

     * @throws TException 

     * @throws CatalogServiceException 

     */

    static java.text.SimpleDateFormat sdf = new java.text.SimpleDateFormat("yyyy-MM-dd-HH:mm:ss");

    static Calendar cal=GregorianCalendar.getInstance();

    final public static String AMAZON_FBG_PRICING_XML   = "/home/amazon/Fbgpricing"+sdf.format(cal.getTime())+".xml";

    final public static boolean PURGE_OR_REPLACE   = false;

    public static  List<Amazonlisted> aItems=null;

    public static List<Long> itemIdsToUpdate=null;

    static long milliseconds;

    static{

        CatalogClient catalogServiceClient = null;

        try {

            catalogServiceClient = new CatalogClient();

        } catch (TTransportException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        Client catalogClient = catalogServiceClient.getClient();

        try {

            Calendar cal=GregorianCalendar.getInstance();

            milliseconds = cal.getTimeInMillis();

            aItems = catalogClient.getAllFbgListedItems();

        } catch (TException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

    public static void main(String... args) throws CatalogServiceException, TException {

        /************************************************** **********************

         * Access Key ID and Secret Access Key ID, obtained from:

         * http://aws.amazon.com

         ***********************************************************************/

        //public static final String marketplaceId = "A21TJRUUN4KGV";

        final String accessKeyId = "AKIAII3SGRXBJDPCHSGQ";

        final String secretAccessKey = "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";

        final String appName = "Test";

        final String appVersion = "1.0";

        MarketplaceWebServiceConfig config = new MarketplaceWebServiceConfig();

        /************************************************************************

         * Uncomment to set the appropriate MWS endpoint.

         ************************************************************************/

        // US

        // config.setServiceURL("https://mws.amazonservices.com");

        // UK

        // config.setServiceURL("https://mws.amazonservices.co.uk");

        // Germany

        // config.setServiceURL("https://mws.amazonservices.de");

        // France

        // config.setServiceURL("https://mws.amazonservices.fr");

        // Italy

        // config.setServiceURL("https://mws.amazonservices.it");

        // Japan

        // config.setServiceURL("https://mws.amazonservices.jp");

        // China

        // config.setServiceURL("https://mws.amazonservices.com.cn");

        // Canada

        // config.setServiceURL("https://mws.amazonservices.ca");

        // India

        config.setServiceURL("https://mws.amazonservices.in");

        /************************************************************************

         * You can also try advanced configuration options. Available options are:

         *

         *  - Signature Version

         *  - Proxy Host and Proxy Port

         *  - User Agent String to be sent to Marketplace Web Service

         *

         ***********************************************************************/

        /************************************************************************

         * Instantiate Http Client Implementation of Marketplace Web Service        

         ***********************************************************************/

        MarketplaceWebService service = new MarketplaceWebServiceClient(

                accessKeyId, secretAccessKey, appName, appVersion, config);

        /************************************************************************

         * Setup request parameters and uncomment invoke to try out sample for

         * Submit Feed

         ***********************************************************************/

        /************************************************************************

         * Marketplace and Merchant IDs are required parameters for all

         * Marketplace Web Service calls.

         ***********************************************************************/

        //final String merchantId = "AF6E300VE0X4D";

        final String merchantId = "AF6E3O0VE0X4D";  

        // marketplaces to which this feed will be submitted; look at the

        // API reference document on the MWS website to see which marketplaces are

        // included if you do not specify the list yourself

        final IdList marketplaces = new IdList(Arrays.asList("A21TJRUUN4KGV"));

        SubmitFeedRequest requestpricing = new SubmitFeedRequest();

        requestpricing.setMerchant(merchantId);

        requestpricing.setMarketplaceIdList(marketplaces);

        requestpricing.setFeedType("_POST_PRODUCT_PRICING_DATA_");

        // MWS exclusively offers a streaming interface for uploading your

        // feeds. This is because

        // feed sizes can grow to the 1GB+ range - and as your business grows

        // you could otherwise

        // silently reach the feed size where your in-memory solution will no

        // longer work, leaving you

        // puzzled as to why a solution that worked for a long time suddenly

        // stopped working though

        // you made no changes. For the same reason, we strongly encourage you

        // to generate your feeds to

        // local disk then upload them directly from disk to MWS via Java -

        // without buffering them in JavaFileNotFoundException

        // memory in their entirety.

        // Note: MarketplaceWebServiceClient will not retry a submit feed request

        // because there is no way to reset the InputStream from our client. 

        // To enable retry, recreate the InputStream and resubmit the feed

        // with the new InputStream. 

        //

        try {

            createFBGPricingFeed();

            FileInputStream pricingfis = new FileInputStream(AMAZON_FBG_PRICING_XML);

            requestpricing.setContentMD5(computeContentMD5HeaderValue(pricingfis));

            requestpricing.setFeedContent(pricingfis);

        } catch (FileNotFoundException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        try {

            invokeSubmitFeed(service,requestpricing);

            Thread.sleep(180);

        } catch (InterruptedException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

    /**

     * Submit Feed request sample Uploads a file for processing together with

     * the necessary metadata to process the file, such as which type of feed it

     * is. PurgeAndReplace if true means that your existing e.g. inventory is

     * wiped out and replace with the contents ocf this feed - use with caution

     * (the default is false).

     * 

     * @param service

     *            instance of MarketplaceWebService service

     * @param request

     *            Action to invoke

     * @throws InterruptedException 

     */

    public static void invokeSubmitFeed(MarketplaceWebService service,

            SubmitFeedRequest request) throws InterruptedException {

        boolean retry = false;

        int retryCount = 0;

        while (!retry) {

            try {

                SubmitFeedResponse response = service.submitFeed(request);

                System.out.println("SubmitFeed Action Response");

                System.out

                .println("=============================================================================");

                System.out.println();

                System.out.print("    SubmitFeedResponse");

                System.out.println();

                FeedSubmissionInfo feedSubmissionInfo=null;

                if (response.isSetSubmitFeedResult()) {

                    System.out.print("        SubmitFeedResult");

                    System.out.println();

                    SubmitFeedResult submitFeedResult = response

                    .getSubmitFeedResult();

                    if (submitFeedResult.isSetFeedSubmissionInfo()) {

                        System.out.print("            FeedSubmissionInfo");

                        System.out.println();

                        feedSubmissionInfo = submitFeedResult

                        .getFeedSubmissionInfo();

                        if (feedSubmissionInfo.isSetFeedSubmissionId()) {

                            System.out.print("                FeedSubmissionId");

                            System.out.println();

                            System.out.print("                    "

                                    + feedSubmissionInfo.getFeedSubmissionId());

                            System.out.println();

                        }

                        if (feedSubmissionInfo.isSetFeedType()) {

                            System.out.print("                FeedType");

                            System.out.println();

                            System.out.print("                    "

                                    + feedSubmissionInfo.getFeedType());

                            System.out.println();

                        }

                        if (feedSubmissionInfo.isSetSubmittedDate()) {

                            System.out.print("                SubmittedDate");

                            System.out.println();

                            System.out.print("                    "

                                    + feedSubmissionInfo.getSubmittedDate());

                            System.out.println();

                        }

                        if (feedSubmissionInfo.isSetFeedProcessingStatus()) {

                            System.out

                            .print("                FeedProcessingStatus");

                            System.out.println();

                            System.out.print("                    "

                                    + feedSubmissionInfo.getFeedProcessingStatus());

                            System.out.println();

                        }

                        if (feedSubmissionInfo.isSetStartedProcessingDate()) {

                            System.out

                            .print("                StartedProcessingDate");

                            System.out.println();

                            System.out

                            .print("                    "

                                    + feedSubmissionInfo

                                    .getStartedProcessingDate());

                            System.out.println();

                        }

                        if (feedSubmissionInfo.isSetCompletedProcessingDate()) {

                            System.out

                            .print("                CompletedProcessingDate");

                            System.out.println();

                            System.out.print("                    "

                                    + feedSubmissionInfo

                                    .getCompletedProcessingDate());

                            System.out.println();

                        }

                    }

                }

                if (response.isSetResponseMetadata()) {

                    System.out.print("        ResponseMetadata");

                    System.out.println();

                    ResponseMetadata responseMetadata = response

                    .getResponseMetadata();

                    if (responseMetadata.isSetRequestId()) {

                        System.out.print("            RequestId");

                        System.out.println();

                        System.out.print("                "

                                + responseMetadata.getRequestId());

                        System.out.println();

                    }

                }

                System.out.println(response.getResponseHeaderMetadata());

                retry =true;

                System.out.println("============Sleeping For 5 Minutes=============");

                Thread.sleep(300000);

                CheckFeedSubmission checkResult = new CheckFeedSubmission();

                ProcessFeed feedProcessingResult=null;

                int checkResultCount=0;

                try {

                    while (feedProcessingResult == null){

                        feedProcessingResult = checkResult.getResult(feedSubmissionInfo.getFeedSubmissionId(),"FbgPricing",milliseconds,itemIdsToUpdate);

                        if (feedProcessingResult == null){

                            checkResultCount++;

                            if (checkResultCount == 2){

                                String text = "";

                                String[] sendTo = { "eng@shop2020.in" };

                                String emailSubjectTxt = "Unable To Receive FBG Pricing Feed Result.Submission Id "+feedSubmissionInfo.getFeedSubmissionId();

                                String emailFromAddress = "build@shop2020.in";

                                String password = "cafe@nes";

                                GmailUtils mailer = new GmailUtils();

                                try {

                                    mailer.sendSSLMessage(sendTo, emailSubjectTxt, text, emailFromAddress, password, new ArrayList<File>());

                                }

                                catch (Exception e) {

                                    e.printStackTrace();

                                }

                                System.exit(1);

                            }

                            System.out.println("=====Feed Result Not Ready Retry Again After 5 Minutes=============");

                            Thread.sleep(300000);

                        }

                    }

                } catch (ParserConfigurationException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                } catch (SAXException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                } catch (IOException e) {

                    // TODO Auto-generated catch block

                    e.printStackTrace();

                }

                String emailSubjectTxt="";

                String[] sendTo;

                if (Boolean.valueOf(feedProcessingResult.getResult())){

                    emailSubjectTxt = "FBG Pricing Feed Success "+sdf.format(cal.getTime());

                    sendTo = new String[]{ "vikram.raghav@shop2020.in", "kshitij.sood@shop2020.in","anikendra.das@shop2020.in"};

                }

                else {

                    emailSubjectTxt = "FBG Pricing Feed Failure "+sdf.format(cal.getTime());

                    sendTo = new String[]{ "eng@shop2020.in"};

                }

                String text = "MessagesProcessed "+feedProcessingResult.getProcessed()+"\n"+"MessagesSuccessful "+feedProcessingResult.getSuccess()+"\n"+"" +

                "MessagesWithError "+feedProcessingResult.getError()+"\n"+"MessagesWithWarning "+feedProcessingResult.getWarn(); 

                String emailFromAddress = "build@shop2020.in";

                String password = "cafe@nes";

                GmailUtils mailer = new GmailUtils();

                try {

                    mailer.sendSSLMessage(sendTo, emailSubjectTxt, text, emailFromAddress, password, "/home/amazon/feedSubmission"+"FbgPricing"+"Result.xml");

                }

                catch (Exception ex) {

                    ex.printStackTrace();

                }

                System.out.println();

            } catch (MarketplaceWebServiceException ex) {

                System.out.println("Caught Exception: " + ex.getMessage());

                System.out.println("Response Status Code: " + ex.getStatusCode());

                System.out.println("Error Code: " + ex.getErrorCode());

                System.out.println("Error Type: " + ex.getErrorType());

                System.out.println("Request ID: " + ex.getRequestId());

                System.out.print("XML: " + ex.getXML());

                System.out.println("ResponseHeaderMetadata: " + ex.getResponseHeaderMetadata());

                if (retryCount == 2) {

                    StringWriter errors = new StringWriter();

                    ex.printStackTrace(new PrintWriter(errors));

                    String text = errors.toString(); 

                    String[] sendTo = { "eng@shop2020.in" };

                    String emailSubjectTxt = "Unable To Send FBG Pricing Feed ";

                    String emailFromAddress = "build@shop2020.in";

                    String password = "cafe@nes";

                    GmailUtils mailer = new GmailUtils();

                    try {

                        mailer.sendSSLMessage(sendTo, emailSubjectTxt, text, emailFromAddress, password, new ArrayList<File>());

                    }

                    catch (Exception e) {

                        e.printStackTrace();

                    }

                    System.exit(1);

                }

                retryCount++;

                Thread.sleep(60000);

            }

        }

    }

    public static long getTimeInMilliseconds(){

        Calendar cal=GregorianCalendar.getInstance();

        return Long.valueOf(cal.getTimeInMillis());

    }

    public static void createFBGPricingFeed() throws CatalogServiceException, TException{

        File file = new File(AMAZON_FBG_PRICING_XML);

        AmazonEnvelope amazonenvelope =  new AmazonEnvelope();

        Header header = new Header();

        header.setDocumentVersion("1.0");

        header.setMerchantIdentifier("AF6E3O0VE0X4D");

        header.setOverrideReleaseId(null);

        amazonenvelope.setHeader(header);

        amazonenvelope.setMessageType("Price");

        amazonenvelope.setMarketplaceName(null);

        amazonenvelope.setPurgeAndReplace(PURGE_OR_REPLACE);

        JAXBContext context;

        Marshaller m=null;

        try {

            context = JAXBContext.newInstance(AmazonEnvelope.class);

            m = context.createMarshaller();

        } catch (JAXBException e1) {

            e1.printStackTrace();

        }

        XMLGregorianCalendar today = null;

        Calendar localCalendar = Calendar.getInstance(TimeZone.getDefault());

        int currentMonth = localCalendar.get(Calendar.MONTH);

        int currentYear = localCalendar.get(Calendar.YEAR);

        int currentDayOfMonth = localCalendar.get(Calendar.DAY_OF_MONTH);

        try {

            today = DatatypeFactory.newInstance()

            .newXMLGregorianCalendar(

                    new GregorianCalendar());

            localCalendar = Calendar.getInstance(TimeZone.getDefault());

        } catch (DatatypeConfigurationException e) {

            // TODO Auto-generated catch block

            e.printStackTrace(); 

        }

        amazonenvelope.setEffectiveDate(today);

        List <AmazonEnvelope.Message> messages = amazonenvelope.getMessage();

        int count=0;

        System.out.println("Inside price listing creation");

        itemIdsToUpdate = new ArrayList<Long>();

        for(Amazonlisted AItem:aItems){

            if ( AItem.getFbgPriceLastUpdatedOn() >= AItem.getFbgPriceLastUpdatedOnSc() && (!AItem.isSuppressFbgPriceUpdate())){

                AmazonEnvelope.Message message = new AmazonEnvelope.Message();

                Price price = new Price();

                price.setSKU("FBG"+String.valueOf(AItem.getItemid()));

                OverrideCurrencyAmount overridecurrencyamount = new OverrideCurrencyAmount();

                overridecurrencyamount.setCurrency(BaseCurrencyCodeWithDefault.DEFAULT);

                overridecurrencyamount.setValue(BigDecimal.valueOf(AItem.getFbgPrice()));

                price.setStandardPrice(overridecurrencyamount);

                Random rand = new Random();

                message.setMessageID(BigInteger.valueOf(Math.abs(rand.nextInt())));

                if ((AItem.getFbgPromoStartDate()<=getTimeInMilliseconds() && AItem.getFbgPromoEndDate()>=getTimeInMilliseconds()) && !AItem.isIsFbgPromotionActive()){

                    //safety check

                    continue;

                }

                if (AItem.isIsFbgPromotionActive() && (AItem.getFbgPromoStartDate()<=getTimeInMilliseconds() && AItem.getFbgPromoEndDate()>=getTimeInMilliseconds())){

                    Sale sale = new Sale();

                    OverrideCurrencyAmount overridesaleamount = new OverrideCurrencyAmount();

                    overridesaleamount.setCurrency(BaseCurrencyCodeWithDefault.DEFAULT);

                    overridesaleamount.setValue(BigDecimal.valueOf(AItem.getFbgPromoPrice()));

                    sale.setSalePrice(overridesaleamount);

                    XMLGregorianCalendar startDate = null;

                    XMLGregorianCalendar endDate = null;

                    try {

                        GregorianCalendar calendar = new GregorianCalendar();

                        calendar.setTimeInMillis(AItem.getFbgPromoStartDate());

                        startDate = DatatypeFactory.newInstance().newXMLGregorianCalendar(

                                calendar);

                        calendar.setTimeInMillis(AItem.getFbgPromoEndDate());

                        endDate = DatatypeFactory.newInstance().newXMLGregorianCalendar(

                                calendar);

                    } catch (DatatypeConfigurationException e) {

                        e.printStackTrace();

                        continue;

                    }

                    sale.setStartDate(startDate);

                    sale.setEndDate(endDate);

                    price.setSale(sale);

                }

                message.setPrice(price);

                message.setOperationType("Update");

                messages.add(message);

                itemIdsToUpdate.add(AItem.getItemid());

                count++;

            }

        }

        if ( messages.size() == 0 ){

            System.out.println("No change in FBG Prices");

            String text = "";

            String[] sendTo = {"vikram.raghav@shop2020.in", "kshitij.sood@shop2020.in"};

            String emailSubjectTxt = "No change in FBG Prices";

            String emailFromAddress = "build@shop2020.in";

            String password = "cafe@nes";

            GmailUtils mailer = new GmailUtils();

            try {

                mailer.sendSSLMessage(sendTo, emailSubjectTxt, text, emailFromAddress, password, new ArrayList<File>());

            }

            catch (Exception e) {

                e.printStackTrace();

            }

            System.exit(1);

        }

        try {

            m.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);

            m.marshal(amazonenvelope, file);

            m.marshal(amazonenvelope, System.out);

        } catch (Exception e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

    }

    public static String computeContentMD5HeaderValue( FileInputStream fis) {

        try {

            DigestInputStream dis = new DigestInputStream( fis,

                    MessageDigest.getInstance( "MD5" ));

            byte[] buffer = new byte[8192];

            while( dis.read( buffer ) > 0 );

            String md5Content = new String(Base64.encodeBase64(dis.getMessageDigest().digest())

            );

            // Effectively resets the stream to be beginning of the file via a

            fis.getChannel().position( 0 );

            return md5Content;

        } catch (NoSuchAlgorithmException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        } catch (IOException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        return null;

    }

}