Subversion Repositories SmartDukaan

Rev

Rev 25758 | Rev 25769 | Go to most recent revision | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 25758 Rev 25765
Line 1... Line 1...
1
package com.smartdukaan.cron.migrations;
1
package com.smartdukaan.cron.migrations;
2
 
2
 
-
 
3
import java.io.File;
3
import java.io.Serializable;
4
import java.io.Serializable;
4
import java.time.LocalDate;
5
import java.time.LocalDate;
5
import java.time.LocalDateTime;
6
import java.time.LocalDateTime;
6
import java.time.LocalTime;
7
import java.time.LocalTime;
7
import java.util.ArrayList;
8
import java.util.ArrayList;
Line 725... Line 726...
725
		driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS);
726
		driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS);
726
 
727
 
727
		
728
		
728
		Map<String, Set<Integer>> authUserPartnerIdMapping = csService.getAuthUserPartnerIdMapping();
729
		Map<String, Set<Integer>> authUserPartnerIdMapping = csService.getAuthUserPartnerIdMapping();
729
		Set<Integer> allPartners = new HashSet<>();
730
		Set<Integer> allPartners = new HashSet<>();
730
		Map<Integer, String> partnerSnapshotMap = new HashMap<>();
731
		Map<Integer, File> partnerSnapshotMap = new HashMap<>();
731
		authUserPartnerIdMapping.values().stream().forEach(x -> allPartners.addAll(x));
732
		authUserPartnerIdMapping.values().stream().forEach(x -> allPartners.addAll(x));
732
		System.out.println(allPartners.size());
733
		System.out.println(allPartners.size());
733
		for (int fofoId : allPartners) {
734
		for (int fofoId : allPartners) {
734
			driver.get("https://partners.smartdukaan.com/12dashboard34?fofoId=" + fofoId);
735
			driver.get("https://partners.smartdukaan.com/12dashboard34?fofoId=" + fofoId);
735
			String base64Image = ((TakesScreenshot) driver).getScreenshotAs(OutputType.BASE64);
736
			File file = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
736
			partnerSnapshotMap.put(fofoId, base64Image);
737
			partnerSnapshotMap.put(fofoId, file);
737
 
738
 
738
		}
739
		}
739
		for (Map.Entry<String, Set<Integer>> entry : authUserPartnerIdMapping.entrySet()) {
740
		for (Map.Entry<String, Set<Integer>> entry : authUserPartnerIdMapping.entrySet()) {
740
			String email = entry.getKey();
741
			String email = entry.getKey();
741
			Set<Integer> partnerIds = entry.getValue();
742
			Set<Integer> partnerIds = entry.getValue();
742
			StringBuffer body = new StringBuffer();
743
			StringBuffer body = new StringBuffer();
-
 
744
			
743
			for (int fofoId : partnerIds) {
745
			for (int fofoId : partnerIds) {
744
				body.append(String.format("<br><img src=\"data:image/png;base64,%s\">", partnerSnapshotMap.get(fofoId)));
746
				body.append(String.format("<br><img src=\"cid:%d\">", fofoId));
745
			}
747
			}
-
 
748
			Utils.sendEmbeddedHtmlMail(mailSender, 
746
			Utils.sendHtmlMailWithAttachments(mailSender, new String[] { "amit.gupta@shop2020.in" }, new String[] {},
749
					new String[] { "amit.gupta@shop2020.in"}, 
-
 
750
					new String[] {},
747
					String.format("Franchise Screenshots - %s", FormattingUtils.formatDate(LocalDateTime.now())),
751
					String.format("Franchise Screenshots - %s", FormattingUtils.formatDate(LocalDateTime.now())),
748
					body.toString());
752
					body.toString(), partnerSnapshotMap );
749
			break;
753
			break;
750
		}
754
		}
751
		driver.quit();
755
		driver.quit();
752
	}
756
	}
753
}
757
}