Subversion Repositories SmartDukaan

package com.amazonaws.mws.samples;

import in.shop2020.model.v1.order.AmazonFCWarehouseLocation;

import in.shop2020.model.v1.order.AmazonFbaOrderItem;

import in.shop2020.model.v1.order.AmazonHourlySaleSnapshot;

import in.shop2020.utils.GmailUtils;

import java.io.File;

import java.io.FileInputStream;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.FileReader;

import java.io.IOException;

import java.io.OutputStream;

import java.text.ParseException;

import java.text.SimpleDateFormat;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Calendar;

import java.util.Date;

import java.util.GregorianCalendar;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.TimeZone;

import java.util.Map.Entry;

import javax.xml.bind.JAXBContext;

import javax.xml.bind.JAXBException;

import javax.xml.bind.Unmarshaller;

import javax.xml.datatype.DatatypeConfigurationException;

import javax.xml.datatype.DatatypeFactory;

import javax.xml.datatype.XMLGregorianCalendar;

import org.apache.commons.io.IOUtils;

import org.apache.thrift.TException;

import org.json.JSONException;

import com.amazonaws.mws.MarketplaceWebService;

import com.amazonaws.mws.MarketplaceWebServiceClient;

import com.amazonaws.mws.MarketplaceWebServiceConfig;

import com.amazonaws.mws.MarketplaceWebServiceException;

import com.amazonaws.mws.model.AmazonEnvelope;

import com.amazonaws.mws.model.FulfillmentData;

import com.amazonaws.mws.model.GetReportListRequest;

import com.amazonaws.mws.model.GetReportRequest;

import com.amazonaws.mws.model.IdList;

import com.amazonaws.mws.model.Message;

import com.amazonaws.mws.model.Order;

import com.amazonaws.mws.model.RequestReportRequest;

import com.amazonaws.mws.samples.Consumer;

public class AmazonOrderData {

    public final static String accessKeyId = "AKIAII3SGRXBJDPCHSGQ";

    public final static String secretAccessKey = "B92xTbNBTYygbGs98w01nFQUhbec1pNCkCsKVfpg";

    public final static String appName = "Test";

    public final static String appVersion = "1.0";

    public final static String merchantId = "AF6E3O0VE0X4D";

    public final static String FILE_PATH = "/tmp/amazon-order-dump.xml";

    public final static SimpleDateFormat dateFormatter = new SimpleDateFormat("yyyy-MM-dd");

    public static List<AmazonFbaOrderItem> sales = new ArrayList<AmazonFbaOrderItem>();

    public static Date startDate;

    public static Date endDate;

    public static void terminate(String msg){

        String emailFromAddress = "build@shop2020.in";

        String password = "cafe@nes";

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

        String emailSubjectTxt = msg;

        try {

            GmailUtils mailer = new GmailUtils();

            mailer.sendSSLMessage(sendTo, emailSubjectTxt, "", emailFromAddress, password,"");

        }

        catch (Exception ex) {

            ex.printStackTrace();

        }

        System.exit(1);

    }

    public static String invokeServiceRequest() throws InterruptedException{

        /* Returns request id of the OrderReport */

        MarketplaceWebServiceConfig config = new MarketplaceWebServiceConfig();

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

        MarketplaceWebService service = new MarketplaceWebServiceClient(accessKeyId, secretAccessKey, appName, appVersion, config);

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

        RequestReportRequest orderreportrequest = new RequestReportRequest()

        .withMerchant(merchantId)

        .withMarketplaceIdList(marketplaces)

        .withReportType("_GET_XML_ALL_ORDERS_DATA_BY_ORDER_DATE_")

        .withReportOptions("ShowSalesChannel=true");

        DatatypeFactory df = null;

        try {

            df = DatatypeFactory.newInstance();

        } catch (DatatypeConfigurationException e) {

            e.printStackTrace();

            throw new RuntimeException(e);

        }

        try {

            startDate = dateFormatter.parse("2014-11-16");

            endDate = dateFormatter.parse("2014-11-20");

        } catch (ParseException e) {

            // TODO Auto-generated catch block

            e.printStackTrace();

        }

        GregorianCalendar ost = new GregorianCalendar();

        ost.setTimeInMillis(startDate.getTime());

        XMLGregorianCalendar  orderStartDate = df.newXMLGregorianCalendar(ost);

        ost = new GregorianCalendar();

        ost.setTimeInMillis(endDate.getTime());

        XMLGregorianCalendar orderEndDate = df.newXMLGregorianCalendar(ost);

        orderreportrequest.setStartDate(orderStartDate);

        orderreportrequest.setEndDate(orderEndDate);

        System.out.println(orderStartDate);

        System.out.println(orderEndDate);

        String orderreportrequestId = null;

        boolean retry=true;

        int retryCount =1;

        while(retry){

            if(retryCount==10){

                terminate("Amazon FBA Order Data  - Unable to get request id");

            }

            try {

                orderreportrequestId = RequestReportSample.invokeRequestReport(service, orderreportrequest);

                retry = false;

            } catch (MarketplaceWebServiceException e1) {

                e1.printStackTrace();

                Thread.sleep(60*1000);

                retryCount++;

            }

        }

        return orderreportrequestId;

    }

    public static void writeDataToDisk(String requestId) throws FileNotFoundException, IOException, InterruptedException{

        /* Write OrderXml file to disk */

        int retryCount = 1;

        while(true){

            GetReportListRequest requestreportlist = new GetReportListRequest();

            requestreportlist.setMerchant( merchantId );

            final IdList requestIdList = new IdList(Arrays.asList(requestId));        

            requestreportlist.setReportRequestIdList(requestIdList);

            MarketplaceWebServiceConfig config = new MarketplaceWebServiceConfig();

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

            MarketplaceWebService service = new MarketplaceWebServiceClient(accessKeyId, secretAccessKey, appName, appVersion, config);

            Map<String, String> requestIdreportIdmap = GetReportListSample.invokeGetReportList(service, requestreportlist);

            if(requestIdreportIdmap.get(requestId)!=null){

                GetReportRequest requestorderreport = new GetReportRequest();

                requestorderreport.setMerchant( merchantId );

                requestorderreport.setReportId( requestIdreportIdmap.get(requestId));

                OutputStream orderreport=null;

                try {

                    orderreport = new FileOutputStream( FILE_PATH );

                } catch (FileNotFoundException e) {

                    e.printStackTrace();

                }

                requestorderreport.setReportOutputStream(orderreport);

                GetReportSample.invokeGetReport(service, requestorderreport);

                System.out.println("Order Report is ready please check");

                String toFind = "<AmazonEnvelope xmlns:xsi=\"http://www.w3.org/2001/XMLSchema-instance\" xsi:noNamespaceSchemaLocation=\"amzn-envelope.xsd\">";

                String toReplace = "<AmazonEnvelope xmlns=\"http://mws.amazonaws.com/doc/2009-01-01/\">";

                File orderReportFile = new File(FILE_PATH);

                String content = "";

                try {

                    content = IOUtils.toString(new FileInputStream(orderReportFile));

                } catch (FileNotFoundException e1) {

                    // TODO Auto-generated catch block

                    e1.printStackTrace();

                } catch (IOException e1) {

                    // TODO Auto-generated catch block

                    e1.printStackTrace();

                }

                content = content.replaceAll(toFind, toReplace);

                IOUtils.write(content, new FileOutputStream(orderReportFile));

                break;

            }

            else{

                if (retryCount == 5){

                    terminate("Unable to pull FBA Hourly sales report");

                }

                retryCount++;

                Thread.sleep(60*1000);

            }

        }

    }

    public static void populateAndCommitSale() throws ParseException{

        JAXBContext jc = null;

        Unmarshaller unmarshaller = null;

        AmazonEnvelope amazonOrderData = null;

        try {

            jc = JAXBContext.newInstance(AmazonEnvelope.class);

            unmarshaller = jc.createUnmarshaller();

            amazonOrderData = (AmazonEnvelope)unmarshaller.unmarshal(new File(FILE_PATH));

        } catch (JAXBException e1) {

            e1.printStackTrace();

        }

        List<Message> orderMessageList = amazonOrderData.getMessage();

        System.out.println("Amazon Order List ... "+orderMessageList.size());

        for(Message orderMessage : orderMessageList){

            Order amazonOrder = orderMessage.getOrder();

            FulfillmentData orderFullfillmentData = amazonOrder.getFulfillmentData();

            if("Amazon.in".equalsIgnoreCase(amazonOrder.getSalesChannel()) && "Amazon".equalsIgnoreCase(orderFullfillmentData.getFulfillmentChannel())){

                for (int i= 0 ; i < amazonOrder.getOrderItem().size(); i++){

                    Long itemid;

                    AmazonFCWarehouseLocation fcLocation;

                    if(amazonOrder.getOrderItem().get(i).getSKU().startsWith("FBA")){

                        try{

                            itemid = Long.parseLong(amazonOrder.getOrderItem().get(i).getSKU().replaceAll("FBA",""));

                        }

                        catch(Exception ex){

                            continue;

                        }

                        fcLocation = AmazonFCWarehouseLocation.Mumbai;

                    }

                    else if(amazonOrder.getOrderItem().get(i).getSKU().startsWith("FBB")){

                        try{

                            itemid = Long.parseLong(amazonOrder.getOrderItem().get(i).getSKU().replaceAll("FBB",""));

                        }

                        catch(Exception ex){

                            continue;

                        }

                        fcLocation = AmazonFCWarehouseLocation.Bangalore;

                    }

                    else if(amazonOrder.getOrderItem().get(i).getSKU().startsWith("FBG")){

                        try{

                            itemid = Long.parseLong(amazonOrder.getOrderItem().get(i).getSKU().replaceAll("FBG",""));

                        }

                        catch(Exception ex){

                            continue;

                        }

                        fcLocation = AmazonFCWarehouseLocation.Gurgaon;

                    }

                    else{

                        continue;

                    }

                    Integer qty=0;

                    if(amazonOrder.getOrderItem().get(i).getQuantity()!=0){

                        qty = new Integer(amazonOrder.getOrderItem().get(i).getQuantity());

                    }

                    Float itemSale = null;

                    if(amazonOrder.getOrderItem().get(i).getItemPrice()!=null && amazonOrder.getOrderItem().get(i).getItemPrice().getComponent()!=null && amazonOrder.getOrderItem().get(i).getItemPrice().getComponent().size()>0){

                        if(amazonOrder.getOrderItem().get(i).getItemPrice().getComponent().get(0).getAmount().getValue()!=0){

                            itemSale = new Float(amazonOrder.getOrderItem().get(i).getItemPrice().getComponent().get(0).getAmount().getValue());

                        }

                        else{

                            itemSale = (float) 0;

                        }

                    }

                    else{

                        itemSale = (float) 0;

                    }

                    Float itemDiscount; 

                    if(amazonOrder.getOrderItem().get(i).getPromotion()!=null){

                        if(amazonOrder.getOrderItem().get(i).getPromotion().getItemPromotionDiscount()!=null && amazonOrder.getOrderItem().get(i).getPromotion().getItemPromotionDiscount().floatValue()!=0.0f){

                            itemDiscount = amazonOrder.getOrderItem().get(i).getPromotion().getItemPromotionDiscount();

                        }

                        else{

                            itemDiscount = new Float(0);

                        }

                    }

                    else{

                        itemDiscount = new Float(0);

                    }

                    AmazonFbaOrderItem amazonFbaOrderItem = new AmazonFbaOrderItem();

                    amazonFbaOrderItem.setItem_id(itemid);

                    amazonFbaOrderItem.setAmazonOrderId(amazonOrder.getAmazonOrderID());

                    amazonFbaOrderItem.setOrderStatus(amazonOrder.getOrderItem().get(i).getItemStatus());

                    amazonFbaOrderItem.setFcLocation(fcLocation);

                    amazonFbaOrderItem.setTotalAmount(itemSale);

                    amazonFbaOrderItem.setPromotionDiscount(itemDiscount);

                    amazonFbaOrderItem.setQuantity(qty);

                    amazonFbaOrderItem.setPurchaseDate(getPurchaseDate(amazonOrder.getPurchaseDate().toString()));

                    sales.add(amazonFbaOrderItem);

                }

            }

        }

        boolean result=false;

        try {

            result = Consumer.addOrUpdateFbaSales(sales, startDate.getTime(), endDate.getTime());

        } catch (JSONException e) {

            terminate("Unable to commit hourly FBA sales data to mongo");

        }

        System.out.println("Data Committed ? " + result);

        if (!result){

            terminate("Unable to commit hourly FBA sales data to mongo");

        }

    }

    public static long getPurchaseDate(String purchaseDateString) throws ParseException{

        SimpleDateFormat istFormatter = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ss");

        istFormatter.setLenient(false);

        TimeZone zone= TimeZone.getTimeZone("GMT");

        istFormatter.setTimeZone(zone);

        Date date = istFormatter.parse(purchaseDateString);

        return date.getTime();

    }

    public static void main (String[] args) throws InterruptedException, FileNotFoundException, IOException, TException, ParseException{

        String requestId= invokeServiceRequest();

        writeDataToDisk(requestId);

        //populateAndCommitSale();

    }

}