WebSVN – SmartDukaan – Diff – /trunk/profitmandi-cron/src/main/java/com/smartdukaan/cron/migrations/RunOnceTasks.java

 import java.util.Collections;
 import java.util.HashMap;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
+import java.util.concurrent.TimeUnit;
 import java.util.stream.Collectors;
 import org.apache.commons.io.output.ByteArrayOutputStream;
 import org.apache.commons.lang.StringUtils;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
+import org.apache.solr.common.util.Hash;
+import org.openqa.selenium.OutputType;
+import org.openqa.selenium.TakesScreenshot;
+import org.openqa.selenium.WebDriver;
+import org.openqa.selenium.chrome.ChromeDriver;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.core.io.ByteArrayResource;
 import org.springframework.mail.javamail.JavaMailSender;
 import org.springframework.stereotype.Component;
 import org.springframework.transaction.annotation.Transactional;
 import com.smartdukaan.cron.scheduled.ScheduledTasks;
 import com.spice.profitmandi.common.util.FileUtil;
+import com.spice.profitmandi.common.util.FormattingUtils;
 import com.spice.profitmandi.common.util.Utils;
 import com.spice.profitmandi.dao.entity.catalog.Scheme;
 import com.spice.profitmandi.dao.entity.fofo.CurrentInventorySnapshot;
 import com.spice.profitmandi.dao.entity.fofo.FofoOrder;
 import com.spice.profitmandi.dao.entity.fofo.FofoOrderItem;
 import com.spice.profitmandi.dao.enumuration.catalog.AmountType;
 import com.spice.profitmandi.dao.enumuration.catalog.SchemeType;
 import com.spice.profitmandi.dao.repository.GenericRepository;
 import com.spice.profitmandi.dao.repository.catalog.ItemRepository;
 import com.spice.profitmandi.dao.repository.catalog.SchemeRepository;
+import com.spice.profitmandi.dao.repository.cs.CsService;
 import com.spice.profitmandi.dao.repository.dtr.FofoStoreRepository;
 import com.spice.profitmandi.dao.repository.fofo.CurrentInventorySnapshotRepository;
 import com.spice.profitmandi.dao.repository.fofo.DebitNoteRepository;
 import com.spice.profitmandi.dao.repository.fofo.FofoOrderItemRepository;
 import com.spice.profitmandi.dao.repository.fofo.FofoOrderRepository;
 	@Autowired
 	private LineItemRepository lineItemRepository;
 	@Autowired
 	private SaholicInventorySnapshotRepository saholicInventorySnapshotRepository;
+	@Autowired
+	private CsService csService;
 	@Autowired
 	private SaholicReservationInventorySnapshotRepository saholicReservationInventorySnapshotRepository;
 	@Autowired
 	@Autowired
 	private SchemeRepository schemeRepository;
 	@Autowired
 	private AddressRepository addressRepository;
 	@Autowired
 	private ScheduledTasks scheduledTasks;
 	@Autowired
 	private SchemeItemRepository schemeItemRepository;
 		List<FofoStore> fofoStores = fofoStoreRepository.selectAll();
 		System.out.println("id\tcode\tactive\tname\tcity\tstate\twallet amount\tsum\twallet amount-sum");
 		for (FofoStore fofoStore : fofoStores) {
 			UserWallet uw = userWalletRepository.selectByRetailerId(fofoStore.getId());
 			User user = userRepository.selectById(fofoStore.getId());
-			if(user==null) {
+			if (user == null) {
 				LOGGER.info("store does not exist", fofoStore.getCode());
 				continue;
+			}
 			Address address = null;
 			try {
 				address = addressRepository.selectById(user.getAddressId());
-			} catch(Exception e) {
+			} catch (Exception e) {
 				LOGGER.info("Could not find address for Store", fofoStore.getCode());
 				address = new Address();
+			}
 			double sum = userWalletHistoryRepository.selectSumByWallet(uw.getId());
-			int calculated = (int)sum;
+			int calculated = (int) sum;
 			System.out.printf("%d\t%s\t%b\t%s\t%s\t%s\t%d\t%d\t%d%n", fofoStore.getId(), fofoStore.getCode(),
-					fofoStore.isActive(), address.getName(), address.getCity(), address.getState(), uw.getAmount(), calculated,
+					fofoStore.isActive(), address.getName(), address.getCity(), address.getState(), uw.getAmount(),
-					uw.getAmount() - calculated);
+					calculated, uw.getAmount() - calculated);
+		}
+	}
 	public void changeWarehouse() throws Exception {
 		transactionService.moveWarehouses();
+	}
+	public void mailDashboardScreenshots() throws Exception {
+		System.setProperty("webdriver.chrome.driver", "/usr/bin/chromedriver");
+		WebDriver driver = new ChromeDriver();
+		driver.manage().window().maximize();
+		//Deleting all the cookies
+		driver.manage().deleteAllCookies();
+		//Specifiying pageLoadTimeout and Implicit wait
+		driver.manage().timeouts().pageLoadTimeout(40, TimeUnit.SECONDS);
+		driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS);
+		Map<String, Set<Integer>> authUserPartnerIdMapping = csService.getAuthUserPartnerIdMapping();
+		Set<Integer> allPartners = new HashSet<>();
+		Map<Integer, String> partnerSnapshotMap = new HashMap<>();
+		authUserPartnerIdMapping.values().stream().forEach(x->allPartners.addAll(x));
+		System.out.println(allPartners.size());
+		for(int fofoId : allPartners) {
+			driver.get("https://partners.smartdukaan.com/12dashboard34?fofoId=" + fofoId);
+			String base64Image = ((TakesScreenshot)driver).getScreenshotAs(OutputType.BASE64);
+			partnerSnapshotMap.put(fofoId, base64Image);
+		}
+		for(Map.Entry<String, Set<Integer>> entry : authUserPartnerIdMapping.entrySet()) {
+			String email = entry.getKey();
+			Set<Integer> partnerIds = entry.getValue();
+			StringBuffer body = new StringBuffer();
+			for(int fofoId: partnerIds) {
+				body.append(String.format("<br><image src=\"%s\" />", partnerSnapshotMap.get(fofoId)));
+			}
+			Utils.sendHtmlMailWithAttachments(mailSender, new String[]{"amit.gupta@shop2020.in"}, new String[] {}, String.format("Franchise Screenshots - %s", FormattingUtils.formatDate(LocalDateTime.now())), body.toString());
+		}
+		driver.quit();
+	}
+}
 //7015845171

Subversion Repositories SmartDukaan

(root)/trunk/profitmandi-cron/src/main/java/com/smartdukaan/cron/migrations/RunOnceTasks.java – Rev 25734 → 25749