Subversion Repositories SmartDukaan

import in.shop2020.metamodel.core.Entity;

import in.shop2020.metamodel.util.CreationUtils;

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

import in.shop2020.model.v1.user.ItemCouponDiscount;

import in.shop2020.thrift.clients.CatalogClient;

import in.shop2020.thrift.clients.PromotionClient;

import java.util.Date;

	public static void main(String[] args) throws Exception {

		CatalogClient cl = new CatalogClient();

		in.shop2020.model.v1.catalog.InventoryService.Client client = cl.getClient();

		List<Item> items = cl.getClient().getAllItemsByStatus(status.ACTIVE);

		items.addAll(cl.getClient().getAllItemsByStatus(status.PAUSED));

		items.addAll(cl.getClient().getAllItemsByStatus(status.PAUSED_BY_RISK));

		Map<Long, List<Item>> entityIdItemMap1 = new HashMap<Long, List<Item>>();

		ProductListGenerator pl = new ProductListGenerator(entityIdItemMap1);

		pl.populateEntityIdItemMap(items);

		pl.generateProductXMLForDisplayAds();

	}

    private void populateEntityIdItemMap(List<Item> items){

        Date toDate = new Date();

        for(Item item: items){

            //TODO Can be removed as we are checking in calling function

            if(!(item.getItemStatus()==status.ACTIVE || item.getItemStatus()==status.CONTENT_COMPLETE || item.getItemStatus() == status.PAUSED)){

                continue;

            }

            if(toDate.getTime() < item.getStartDate() ||  item.getSellingPrice() == 0){

                continue;

            }

            List<Item> itemList = entityIdItemMap.get(item.getCatalogItemId());

            if(itemList == null){

                itemList = new ArrayList<Item>();

            }

            itemList.add(item);

            entityIdItemMap.put(item.getCatalogItemId(), itemList);

        }

        //Remove all items which have not been updated since last content generation.

        List<Long> removeEntities = new ArrayList<Long>();

        for(Long entityId:entityIdItemMap.keySet()){

            boolean isValidEntity = false;

            //If any one of the items has been updated before current timestamp, than we generate content for whole entity

            for(Item item: entityIdItemMap.get(entityId)){

                if(item.getUpdatedOn() > 0){

                    isValidEntity = true;

                }

            }

            if(!isValidEntity){

                removeEntities.add(entityId);

            }

        }

        for(Long entityId: removeEntities){

            entityIdItemMap.remove(entityId);

        }

    }

	/**

	 * Generate the xml list of all the active phones and store in a file

	 * @throws Exception

	 */

	public void generateProductXMLForDisplayAds() throws Exception {

		List<String> productXMLSnippets = new ArrayList<String>();

		productXMLSnippets.add("<saholic.com>");

		List<Long> itemIds = new ArrayList<Long>();

		for(Long entityId: entityIdItemMap.keySet()){

			List<Item> items = entityIdItemMap.get(entityId);

			Item firstItem = items.get(0);

			if(isMobile(firstItem)) {

				itemIds.add(firstItem.getId());

			}

		}

		PromotionClient promotionClient = new PromotionClient();

		in.shop2020.model.v1.user.PromotionService.Client promotionServiceClient = promotionClient.getClient();

		List<ItemCouponDiscount> itemsCouponsAndDiscounts = promotionServiceClient.getItemDiscountMap(itemIds);

		Map<Long, ItemCouponDiscount> couponsAndDiscounts = new HashMap<Long, ItemCouponDiscount>();

		for (ItemCouponDiscount itemCouponDiscount: itemsCouponsAndDiscounts) {

			couponsAndDiscounts.put(itemCouponDiscount.getItemId(), itemCouponDiscount);

		}

		for(Long entityId: entityIdItemMap.keySet()) {

			List<Item> items = entityIdItemMap.get(entityId);

			Item firstItem = items.get(0);

			if(!isMobile(firstItem)){

				continue;

			}

			productXMLSnippets.add(this.xmlIndentation[1] + "<products>");	

			productXMLSnippets.add(this.xmlIndentation[2] + "<ProductSKU>" + entityId + "</ProductSKU>");

			String title = getProductTitle(firstItem);

			productXMLSnippets.add(this.xmlIndentation[2] + "<ProductName>" + title + "</ProductName>");

			String url = getProductURL(entityId, firstItem);

			String imageUrl = "http://static" + ((entityId+1)%3) + ".saholic.com" + File.separator + "images" + File.separator + entityId + File.separator + "icon.jpg";    

			productXMLSnippets.add(this.xmlIndentation[2] + "<ProductURL>" + url + "</ProductURL>");

			productXMLSnippets.add(this.xmlIndentation[2] + "<ProductImageURL>" + imageUrl + "</ProductImageURL>");

			productXMLSnippets.add(this.xmlIndentation[2] + "<ProductMRP>" + firstItem.getMrp() + "</ProductMRP>");

			double minPrice = getMinPrice(items);

			productXMLSnippets.add(this.xmlIndentation[2] + "<ProductSellingPrice>" + (int)minPrice  + "</ProductSellingPrice>");

			productXMLSnippets.add(this.xmlIndentation[2] + "<ProductDiscount>" + (int)((firstItem.getMrp()-minPrice)/firstItem.getMrp()*100) + "</ProductDiscount>");

			long itemId = firstItem.getId();

			if(couponsAndDiscounts.containsKey(itemId))	{

				ItemCouponDiscount itemCouponDiscount = couponsAndDiscounts.get(itemId);

				productXMLSnippets.add(this.xmlIndentation[2] + "<ProductCoupons>");

				productXMLSnippets.add(this.xmlIndentation[3] + "<Coupon>");

				productXMLSnippets.add(this.xmlIndentation[4] + "<CouponCode>" + itemCouponDiscount.getCouponCode() + "</CouponCode>");

				productXMLSnippets.add(this.xmlIndentation[4] + "<PriceAfterCoupon>" + (int)(minPrice - itemCouponDiscount.getDiscount()) + "</PriceAfterCoupon>");

				productXMLSnippets.add(this.xmlIndentation[3] + "</Coupon>");

				productXMLSnippets.add(this.xmlIndentation[2] + "</ProductCoupons>");

			} else	{

				productXMLSnippets.add(this.xmlIndentation[2] + "<ProductCoupons />");

			}

			String description = "";

			Entity entity = CreationUtils.getEntity(entityId);

			if(entity!=null){

				if(entity.getSlide(130001) !=null){

					description = entity.getSlide(130001).getFreeformContent().getFreeformText();

				}

			}

			productXMLSnippets.add(this.xmlIndentation[2] + "<ProductDescription>" + description + "</ProductDescription>");

			productXMLSnippets.add(this.xmlIndentation[1] + "</products>");	

		}

		productXMLSnippets.add("</saholic.com>");

		String productDataXML = StringUtils.join(productXMLSnippets, "\n");

		Utils.info(productDataXML);

		// Write it to file

		String productXMLFilename = Utils.EXPORT_PARTNERS_CONTENT_PATH + "advertismentapi.xml";

		DBUtils.store(productDataXML, productXMLFilename);

	}