Subversion Repositories SmartDukaan

Rev

Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | RSS feed

/*******************************************************************************
 *  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 in.shop2020.logistics.LogisticsInfo;
import in.shop2020.logistics.DeliveryType;
import in.shop2020.logistics.LogisticsServiceException;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
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 com.amazonaws.mws.*;
import com.amazonaws.mws.model.*;
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.model.v1.catalog.CatalogService.Client;
import in.shop2020.model.v1.catalog.CatalogServiceException;
import in.shop2020.model.v1.catalog.Item;
import in.shop2020.model.v1.catalog.status;
import in.shop2020.model.v1.inventory.AmazonInventorySnapshot;
import in.shop2020.model.v1.inventory.InventoryServiceException;
import in.shop2020.model.v1.inventory.ItemInventory;
import in.shop2020.model.v1.inventory.Warehouse;
import in.shop2020.model.v1.inventory.WarehouseType;
import in.shop2020.thrift.clients.CatalogClient;
import in.shop2020.thrift.clients.InventoryClient;
import in.shop2020.thrift.clients.LogisticsClient;
import org.apache.thrift.TException;
import org.apache.thrift.transport.TTransportException;

/**
 * 
 * Submit Feed Samples
 * 
 * 
 */
public class SubmitNonFBAInventoryFeed {

    /**
     * Just add a few required parameters, and try the service Submit Feed
     * functionality
     * 
     * @param args
     *            unused
     *            
     */
    /**
     * @param args
     * @throws TException 
     * @throws CatalogServiceException 
     */
        
        final public static String AMAZON_NON_FBA_INVENTORY_XML = "/home/vikram/Nonfbainventory.xml";
        final public static boolean PURGE_OR_REPLACE   = false;
        public static  List<Amazonlisted> amazonNonFbaListedItems=null;
        public static List<in.shop2020.model.v1.catalog.Item> aliveItemsMap=null;
        public static Map<Long,AmazonItem> allItemsAmazonMap   = new HashMap<Long,AmazonItem>();
        
        static{

                CatalogClient catalogServiceClient = null;
                Map<Long,ItemInventory> availability= new HashMap<Long,ItemInventory>();
                try {
                        catalogServiceClient = new CatalogClient();
                } catch (TTransportException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                Client catalogClient = catalogServiceClient.getClient();
                try {
                        amazonNonFbaListedItems = catalogClient.getAllNonFbaListedItems();
                        aliveItemsMap = catalogClient.getAllAliveItems();
                        InventoryClient inventoryServiceClient = null;
                        try {
                                inventoryServiceClient = new InventoryClient();
                        } catch (TTransportException e) {
                                // TODO Auto-generated catch block
                                e.printStackTrace();
                        }
                        in.shop2020.model.v1.inventory.InventoryService.Client inventoryClient = inventoryServiceClient.getClient();
                        availability = inventoryClient.getInventorySnapshot(0);
                        List<Warehouse> warehouses = inventoryClient.getWarehouses(WarehouseType.OURS_THIRDPARTY,null,0,0,0);
                        List<Long> thirdpartywarehouseids = new ArrayList<Long>();
                        for(Warehouse warehouse:warehouses){
                                thirdpartywarehouseids.add(warehouse.getId());
                        }
                        //System.out.println("Availability is " + availability );
                        for(in.shop2020.model.v1.catalog.Item thriftItem:aliveItemsMap){
                                long available=0;
                                long reserve=0;
                                AmazonItem item;
                                if(availability.get(thriftItem.getId())!=null){ 
                                        ItemInventory iteminventory = availability.get(thriftItem.getId());
                                        Map<Long,Long> itemavailability = new HashMap<Long,Long>();
                                        itemavailability = iteminventory.getAvailability();
                                        for (Map.Entry<Long,Long> entry : itemavailability.entrySet()) {
                                                if(!thirdpartywarehouseids.contains(entry.getKey())){
                                                        available =  available + entry.getValue();      
                                                }
                                        }
                                        Map<Long,Long> itemreserve = new HashMap<Long,Long>();
                                        itemreserve = iteminventory.getReserved();
                                        for (Map.Entry<Long,Long> entry : itemreserve.entrySet()) {
                                                if(!thirdpartywarehouseids.contains(entry.getKey())){
                                                        reserve =  reserve + entry.getValue();
                                                }
                                        }
                                        item= new AmazonItem(thriftItem.getId(),available,reserve,thriftItem.getHoldInventory(),thriftItem.getDefaultInventory(),thriftItem.isRisky(),thriftItem.getItemStatus());
                                }
                                else{
                                        item = new AmazonItem(thriftItem.getId(),0,0,thriftItem.getHoldInventory(),thriftItem.getDefaultInventory(),thriftItem.isRisky(),thriftItem.getItemStatus());
                                }

                                //System.out.println(" Item details are " + thriftItem.getId() +" " + available + " " + reserve + " " + thriftItem.getHoldInventory() + " "+ thriftItem.getDefaultInventory() + " " + thriftItem.isRisky());
                                //System.out.println("+++++++++++++++++++++++");
                                allItemsAmazonMap.put(thriftItem.getId(),item);

                        }

                } 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 requestinventory = new SubmitFeedRequest();
        requestinventory.setMerchant(merchantId);
        requestinventory.setMarketplaceIdList(marketplaces);
        requestinventory.setFeedType("_POST_INVENTORY_AVAILABILITY_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 {
                
                createNonFbaInventoryFeed();
                FileInputStream inventoryfis = new FileInputStream(AMAZON_NON_FBA_INVENTORY_XML);
                requestinventory.setContentMD5(computeContentMD5HeaderValue(inventoryfis));
                        requestinventory.setFeedContent(inventoryfis);
                } catch (FileNotFoundException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                
                try {
                                invokeSubmitFeed(service,requestinventory);
                                Thread.sleep(30);
                        } 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
     */
    public static void invokeSubmitFeed(MarketplaceWebService service,
            SubmitFeedRequest request) {
        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();
            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 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());
            System.out.println();
            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());
        }
    }
    

    public static void createNonFbaInventoryFeed() throws CatalogServiceException, TException{


        File file = new File(AMAZON_NON_FBA_INVENTORY_XML);
        AmazonEnvelope amazonenvelope =  new AmazonEnvelope();
        Header header = new Header();
        header.setDocumentVersion("1.0");
        header.setMerchantIdentifier("AF6E3O0VE0X4D");
        header.setOverrideReleaseId(null);
        amazonenvelope.setHeader(header);
        amazonenvelope.setMessageType("Inventory");
        amazonenvelope.setMarketplaceName(null);
        //amazonenvelope.setPurgeAndReplace(PURGE_OR_REPLACE);
        amazonenvelope.setPurgeAndReplace(true);
        JAXBContext context;
        Marshaller m=null;
        try {
                context = JAXBContext.newInstance(AmazonEnvelope.class);
                m = context.createMarshaller();
        } catch (JAXBException e1) {
                // TODO Auto-generated catch block
                e1.printStackTrace();
        }

        XMLGregorianCalendar today = null;
        XMLGregorianCalendar restockdate = null;
        //Calendar localCalendar = Calendar.getInstance();
        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(currentYear,currentMonth,currentDayOfMonth));
                restockdate = DatatypeFactory.newInstance()
                .newXMLGregorianCalendar(
                                new GregorianCalendar(currentYear,currentMonth,currentDayOfMonth));
                restockdate.setTimezone(DatatypeConstants.FIELD_UNDEFINED);
                restockdate.setTime(DatatypeConstants.FIELD_UNDEFINED, DatatypeConstants.FIELD_UNDEFINED, DatatypeConstants.FIELD_UNDEFINED);  


        } catch (DatatypeConfigurationException e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
        }
        amazonenvelope.setEffectiveDate(today);
        LogisticsClient logisticsServiceClient;
                in.shop2020.logistics.LogisticsService.Client logisticsClient=null;
        try {
                logisticsServiceClient = new LogisticsClient("logistics_service_prod_server_host","logistics_service_prod_server_port");
                logisticsClient = logisticsServiceClient.getClient();
                
                } catch (TTransportException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
        
        List <AmazonEnvelope.Message> messages = amazonenvelope.getMessage();
        InventoryClient inventoryServiceClient = new InventoryClient();
        in.shop2020.model.v1.inventory.InventoryService.Client inventoryClient = inventoryServiceClient.getClient();
        long time = System.currentTimeMillis();
        for(Amazonlisted AItem:amazonNonFbaListedItems){
                if(!AItem.isIsInventoryOverride()){
                        AmazonEnvelope.Message message = new AmazonEnvelope.Message();  
                        Inventory inventory = new Inventory();
                        AmazonItem item = allItemsAmazonMap.get(AItem.getItemid());
                        if(item!=null && !item.getStatus().equals(status.PAUSED)){
                                long actualavailibility = (item.getAvailability() - item.getReserve());
                                long holdinventory = item.getHoldInventory();
                                long defaultinventory = item.getDefaultInventory();
                                if(item.getRisky()){
                                        if(actualavailibility > holdinventory){
                                                inventory.setQuantity(BigInteger.valueOf(actualavailibility - holdinventory));
                                        }
                                        else{
                                                inventory.setQuantity(BigInteger.valueOf(0));
                                        }
                                }               
                                else{
                                        if(actualavailibility > defaultinventory){
                                                inventory.setQuantity(BigInteger.valueOf(actualavailibility));
                                        }
                                        else{
                                                inventory.setQuantity(BigInteger.valueOf(defaultinventory));
                                        }
                                }
                        }
                        else{
                                inventory.setQuantity(BigInteger.valueOf(0));
                        }
                        AmazonInventorySnapshot amazonInventorySnapshot = new AmazonInventorySnapshot();
                            amazonInventorySnapshot.setAvailability(inventory.getQuantity().longValue());
                        amazonInventorySnapshot.setItem_id(AItem.getItemid());
                        inventoryClient.addOrUpdateAmazonInventoryForItem(amazonInventorySnapshot,time);
                        inventory.setSKU(Long.toString(AItem.getItemid()));
                        LogisticsInfo logisticinfo=new LogisticsInfo();
                        try {
                                     logisticinfo = logisticsClient.getLogisticsEstimation(AItem.getItemid(),"110001",DeliveryType.COD);
                                } catch (LogisticsServiceException e) {
                                        // TODO Auto-generated catch block
                                        e.printStackTrace();
                                }
                                if(AItem.isIsCustomTime()){
                                        inventory.setFulfillmentLatency(BigInteger.valueOf(AItem.getHandlingTime()));
                                }
                                        
                                else{
                                        inventory.setFulfillmentLatency(BigInteger.valueOf(logisticinfo.getShippingTime()));
                                }
                                if(inventory.getFulfillmentLatency().intValue() <= 0){
                                        inventory.setFulfillmentLatency(BigInteger.valueOf(1));
                                }
                        inventory.setSwitchFulfillmentTo("MFN");
                        Random rand = new Random();
                        message.setMessageID(BigInteger.valueOf(Math.abs(rand.nextInt())));
                        message.setInventory(inventory);
                        message.setOperationType("Update");
                        messages.add(message);
                }
        }
        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;
                
 }


    
    

}