Rev 30317 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | RSS feed
package com.smartdukaan.cron.scheduled;import nu.pattern.OpenCV;import org.apache.commons.io.FileUtils;import org.apache.logging.log4j.LogManager;import org.apache.logging.log4j.Logger;import org.opencv.core.Mat;import org.opencv.imgcodecs.Imgcodecs;import org.opencv.imgproc.Imgproc;import org.openqa.selenium.*;import org.openqa.selenium.chrome.ChromeDriver;import org.openqa.selenium.chrome.ChromeOptions;import org.openqa.selenium.interactions.Actions;import org.openqa.selenium.support.ui.ExpectedConditions;import org.openqa.selenium.support.ui.WebDriverWait;import javax.imageio.ImageIO;import java.awt.image.BufferedImage;import java.io.File;import java.time.LocalDate;import java.time.format.DateTimeFormatter;import java.util.HashMap;import java.util.List;import java.util.Map;import java.util.concurrent.Callable;import java.util.concurrent.TimeUnit;import java.util.stream.Collectors;public class CheckOppoWarrantyTask implements Callable<Map<String, LocalDate>> {private static final Logger LOGGER = LogManager.getLogger(CheckOppoWarrantyTask.class);private List<String> imeis;public CheckOppoWarrantyTask(List<String> imeis) {this.imeis = imeis;}@Overridepublic Map<String, LocalDate> call() throws Exception {return checkWarranty(imeis);}private Map<String, LocalDate> checkWarranty(List<String> imeis) throws Exception {String name = Thread.currentThread().getName();Map<String, LocalDate> dateMap = new HashMap<>();OpenCV.loadShared();LOGGER.info("Initiating webdriver...");System.setProperty("webdriver.chrome.driver", "/usr/bin/chromedriver");Map<String, Object> prefsMap = new HashMap<>();prefsMap.put("profile.default_content_settings.popups", 0);prefsMap.put("download.prompt", false);prefsMap.put("download.directory_upgrade", true);ChromeOptions options = new ChromeOptions();options.setExperimentalOption("prefs", prefsMap);//options.setExperimentalOption("detach", true);options.addArguments("--headless");options.addArguments("--user-agent=\"Mozilla/5.0 (Windows Phone 10.0; Android 4.2.1; Microsoft; Lumia 640 XL LTE) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/42.0.2311.135 Mobile Safari/537.36 Edge/12.10166\"");options.addArguments("--no-sandbox");options.addArguments("start-maximized");options.addArguments("disable-infobars");options.addArguments("--disable-extensions");options.addArguments("--test-type");WebDriver driver = new ChromeDriver(options);driver.manage().timeouts().setScriptTimeout(10, TimeUnit.SECONDS);driver.manage().window().setSize(new Dimension(1600, 900));driver.manage().window().maximize();// Deleting all the cookiesdriver.manage().deleteAllCookies();// Specifiying pageLoadTimeout and Implicit waitdriver.manage().timeouts().pageLoadTimeout(40, TimeUnit.SECONDS);driver.manage().timeouts().implicitlyWait(30, TimeUnit.SECONDS);//Get all imeis//List<String> imeis = Arrays.asList("861066056452377", "866558036988517", "868474039102395", "868474038057590", "868474038058515", "868474038058333", "868474038056691", "868839031100490", "867765031744873", "867765031744352", "868474037740014", "866778040667770", "867765031743354", "865468040669130", "865468041617237", "869146030904131", "866778047106830", "866778049780715", "866778049779550", "865468040779491", "865053047507739");//List<String> imeis = Arrays.asList("861066056452377", "866558036988517", "868474039102395");driver.get("https://support.oppo.com/in/warranty-check/");int mainCount = 0;List<String> remainingImeis = imeis;do {for (int i = 0; i < remainingImeis.size(); i++) {try {String imei = remainingImeis.get(i);System.out.println("Starting fresh with new IMEI " + imei);driver.findElement(By.className("el-input__inner")).sendKeys(imei);driver.findElement(By.className("el-button--primary")).click();WebDriverWait wait10Sec = new WebDriverWait(driver, 10);WebElement slideButton = null;Actions actionProvider = new Actions(driver);// Perform click-and-hold action on the elementint counter = 1;String fileName = "/tmp/" + name + "-simple.png";int moveBy = 0;do {System.out.println("Starting Do");wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_wrapper_1")));wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_slider_1")));slideButton = driver.findElement(By.id("dx_captcha_basic_slider_1"));actionProvider.moveToElement(slideButton).clickAndHold().moveByOffset(10, 0).perform();Mat circles = getMatCircles(driver, fileName);if (circles.cols() == 2) {double firstCircleX = circles.get(0, 0)[0];double secondCircleX = circles.get(0, 1)[0];double distance = Math.abs(firstCircleX - secondCircleX);System.out.println("Starting Checkdistance method");moveBy = checkDistance(driver, wait10Sec, actionProvider, distance);if (moveBy > 0) {System.out.println("Ending Checkdistance method " + moveBy);actionProvider.moveByOffset(moveBy, 0).release().perform();System.out.println("Move click performed");try {try {Thread.sleep(2000);WebElement verifyFailed = wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_bar-fail_1")));System.out.println("Failed = " + counter);counter++;moveBy = 0;} catch (Exception notFailedException) {System.out.println("Failed due to = + " + notFailedException.getMessage());System.out.println("Success at attempt " + counter);wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.className("warranty-service--result_info__wrapper")));try {String warrantyDate = driver.findElement(By.className("warranty-service--result_info__dateValue")).getText();System.out.println("Date -- " + warrantyDate);dateMap.put(imei, LocalDate.parse(warrantyDate, DateTimeFormatter.ofPattern("yyyy/MM/dd")).minusYears(1));} catch (Exception e) {System.out.println("Could not capture date");dateMap.put(imei, null);}WebElement checkAgainButton = driver.findElement(By.className("van-button--normal"));checkAgainButton.click();//Thread.sleep(1000);break;}} catch (Throwable e) {System.out.println("Failed = " + counter + " ----------" + e.getMessage());counter++;moveBy = 0;}}}System.out.println("After circles conditions");wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_wrapper_1")));wait10Sec.until(ExpectedConditions.visibilityOfElementLocated(By.id("dx_captcha_basic_slider_1")));slideButton = driver.findElement(By.id("dx_captcha_basic_slider_1"));actionProvider.moveToElement(slideButton).doubleClick().perform();System.out.println("Ending Do");} while (moveBy <= 0);} catch (Exception e) {e.printStackTrace();}}mainCount++;remainingImeis = imeis.stream().filter(x -> dateMap.containsKey(x)).collect(Collectors.toList());} while (mainCount != 10 && dateMap.size() != imeis.size());//Thread.sleep(5000);driver.close();driver.quit();return dateMap;}private int checkDistance(WebDriver driver, WebDriverWait wait3Sec, Actions actionProvider, double distance) throws Exception {//actionProvider.moveToElement(slideButton).clickAndHold().moveByOffset(1, 0).perform();actionProvider.moveByOffset(15, 0).perform();//String fileName2 = "/tmp/" + Thread.currentThread().getName() + "-moved.png";Mat movedCircles = getMatCircles(driver, fileName2);if (movedCircles.cols() != 2) return 0;double movedFirst = movedCircles.get(0, 0)[0];double movedSecond = movedCircles.get(0, 1)[0];//System.out.println("Test Offset Move " + testMove);System.out.println("Found Moved " + movedFirst + " ," + movedSecond);double movedDistance = Math.abs(movedFirst - movedSecond);double moved = distance - movedDistance;/*System.out.println("Original distance " + distance);System.out.println("Moved Distance " + movedDistance);System.out.println("Moved " + moved);*/if (moved == 0) return 0;int moveBy = (int) ((movedDistance) * (15 / moved));//int moveBy = (int) distance - 21;System.out.println("Move by = " + moveBy);return moveBy;}private Mat getMatCircles(WebDriver driver, String fileName) throws Exception {File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);BufferedImage fullImg = ImageIO.read(screenshot);// Get the location of element on the pageWebElement imageElement = driver.findElement(By.id("dx_captcha_basic_bg_1"));Point point = imageElement.getLocation();// Get width and height of the elementint eleWidth = imageElement.getSize().getWidth();int eleHeight = imageElement.getSize().getHeight();// Crop the entire page screenshot to get only element screenshotBufferedImage eleScreenshot = fullImg.getSubimage(point.getX(), point.getY(),eleWidth, eleHeight);ImageIO.write(eleScreenshot, "png", screenshot);// Copy the element screenshot to diskFile screenshotLocation = new File(fileName);FileUtils.copyFile(screenshot, screenshotLocation);Imgcodecs imageCodecs = new Imgcodecs();Mat grayMatrix = imageCodecs.imread(fileName, Imgcodecs.IMREAD_COLOR);Imgproc.cvtColor(grayMatrix, grayMatrix, Imgproc.COLOR_BGR2GRAY);Mat medianBlur = new Mat();Imgproc.medianBlur(grayMatrix, medianBlur, 5);Mat circles = new Mat();Imgproc.HoughCircles(medianBlur, circles, Imgproc.HOUGH_GRADIENT, 1, 15, 50, 20, 17, 25);return circles;}}