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 SubmitFBDPricingFeedAndFetchResults {

    /**
     * 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_FBD_PRICING_XML   = "/home/amazon/Fbdpricing"+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.getAllFbdListedItems();


        } 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 {
            createFBDPricingFeed();
            FileInputStream pricingfis = new FileInputStream(AMAZON_FBD_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(),"FbdPricing",milliseconds,itemIdsToUpdate);
                        if (feedProcessingResult == null){
                            checkResultCount++;
                            if (checkResultCount == 2){
                                String text = "";
                                String[] sendTo = { "eng@shop2020.in" };
                                String emailSubjectTxt = "Unable To Receive FBD 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 = "FBD Pricing Feed Success "+sdf.format(cal.getTime());
                    sendTo = new String[]{  "kshitij.sood@shop2020.in","anikendra.das@shop2020.in"};
                }
                else {
                    emailSubjectTxt = "FBD 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"+"FbdPricing"+"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 FBD 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 createFBDPricingFeed() throws CatalogServiceException, TException{


        File file = new File(AMAZON_FBD_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.getFbdPriceLastUpdatedOn() >= AItem.getFbdPriceLastUpdatedOnSc() && (!AItem.isSuppressFbdPriceUpdate())){
                if (AItem.getFbdPrice()==0){
                    continue;
                }
                AmazonEnvelope.Message message = new AmazonEnvelope.Message();
                Price price = new Price();
                price.setSKU("FBD"+String.valueOf(AItem.getItemid()));
                OverrideCurrencyAmount overridecurrencyamount = new OverrideCurrencyAmount();
                overridecurrencyamount.setCurrency(BaseCurrencyCodeWithDefault.DEFAULT);
                overridecurrencyamount.setValue(BigDecimal.valueOf(AItem.getFbdPrice()));
                price.setStandardPrice(overridecurrencyamount);
                Random rand = new Random();
                message.setMessageID(BigInteger.valueOf(Math.abs(rand.nextInt())));
                if ((AItem.getFbdPromoStartDate()<=getTimeInMilliseconds() && AItem.getFbdPromoEndDate()>=getTimeInMilliseconds()) && !AItem.isIsFbdPromotionActive()){
                    //safety check
                    continue;
                }
                if (AItem.isIsFbdPromotionActive() && (AItem.getFbdPromoStartDate()<=getTimeInMilliseconds() && AItem.getFbdPromoEndDate()>=getTimeInMilliseconds())){
                    if (AItem.getFbdPromoPrice()==0){
                        continue;
                    }
                    Sale sale = new Sale();
                    OverrideCurrencyAmount overridesaleamount = new OverrideCurrencyAmount();
                    overridesaleamount.setCurrency(BaseCurrencyCodeWithDefault.DEFAULT);
                    overridesaleamount.setValue(BigDecimal.valueOf(AItem.getFbdPromoPrice()));
                    sale.setSalePrice(overridesaleamount);
                    XMLGregorianCalendar startDate = null;
                    XMLGregorianCalendar endDate = null;
                    try {
                        GregorianCalendar calendar = new GregorianCalendar();
                        calendar.setTimeInMillis(AItem.getFbdPromoStartDate());
                        startDate = DatatypeFactory.newInstance().newXMLGregorianCalendar(
                                calendar);
                        calendar.setTimeInMillis(AItem.getFbdPromoEndDate());
                        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 FBD Prices");
            String text = "";
            String[] sendTo = { "kshitij.sood@shop2020.in"};
            String emailSubjectTxt = "No change in FBD 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;

    }





}