WebSVN – SmartDukaan – Diff – /trunk/profitmandi-cron/src/main/java/com/smartdukaan/cron/scheduled/CheckOppoWarrantyTask.java

 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.core.Size;
 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;
         Map<String, LocalDate> dateMap = new HashMap<>();
         //System.out.println(System.getProperty("os.name"));
         //System.out.println(System.getProperty("os.arch"));
         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);
+        System.setProperty("webdriver.chrome.driver", "/usr/bin/chromedriver");
         //I need this
-        //System.setProperty("webdriver.chrome.driver", "/usr/local/bin/chromedriver");
+        //System.setProperty("webdriver.chrome.driver", "/Users/amit/cft/chromedriver/chromedriver");
         ChromeOptions options = new ChromeOptions();
+        //options.setBinary("/Users/amit/cft/chrome/Google Chrome for Testing.app/Contents/MacOS/Google Chrome For Testing");
         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("--disable-extensions");
         options.addArguments("--force-device-scale-factor=1");
         options.addArguments("--test-type");
         WebDriver driver = new ChromeDriver(options);
-        driver.manage().timeouts().setScriptTimeout(10, TimeUnit.SECONDS);
+        driver.manage().timeouts().setScriptTimeout(6, TimeUnit.SECONDS);
         driver.manage().window().setSize(new Dimension(1600, 900));
         driver.manage().window().maximize();
         // Deleting all the cookies
         driver.manage().deleteAllCookies();
         // Specifiying pageLoadTimeout and Implicit wait
-        driver.manage().timeouts().pageLoadTimeout(30, TimeUnit.SECONDS);
+        driver.manage().timeouts().pageLoadTimeout(15, TimeUnit.SECONDS);
-        driver.manage().timeouts().implicitlyWait(15, TimeUnit.SECONDS);
+        driver.manage().timeouts().implicitlyWait(9, TimeUnit.SECONDS);
         int mainCount = 0;
         List<String> remainingImeis = imeis;
-        WebDriverWait wait10Sec = new WebDriverWait(driver, 10);
+        WebDriverWait wait10Sec = new WebDriverWait(driver, 8);
-        WebDriverWait wait5Sec = new WebDriverWait(driver, 5);
+        WebDriverWait wait5Sec = new WebDriverWait(driver, 4);
         WebElement slideButton;
         Actions actionProvider = new Actions(driver);
         do {
             driver.get("https://support.oppo.com/in/warranty-check/");
+            driver.findElement(By.cssSelector(".cp-cookie-tip a.close")).click();
             for (int i = 0; i < remainingImeis.size(); i++) {
                 String imei = remainingImeis.get(i);
                 try {
                     System.out.println("Starting fresh with new IMEI " + imei);
                     if (driver.findElement(By.className("el-input__inner")).getAttribute("value").length() > 0) {
                         System.out.println("Darn.. leaving" + driver.findElement(By.className("el-input__inner")).getAttribute("value"));
                         driver.get("https://support.oppo.com/in/warranty-check/");
                         break;
+                    }
                     driver.findElement(By.className("el-input__inner")).sendKeys(imei);
-                    driver.findElement(By.className("el-button--primary")).click();
+                    //driver.findElement(By.className("el-button--primary")).click();
+                    driver.findElement(By.cssSelector("div.searchBtn")).click();
                     // Perform click-and-hold action on the element
                     int counter = 1;
                     boolean captchaNotBroken = true;
                     do {
                         try {
                             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"));
-                            System.out.println("Slide button - " + slideButton);
+                            //System.out.println("Slide button - " + slideButton);
+                            System.out.println("Margin Left before- " + slideButton.getCssValue("margin-left"));
-                            actionProvider.moveToElement(slideButton).clickAndHold().moveByOffset(10, 0).perform();
+                            actionProvider.moveToElement(slideButton).clickAndHold()
+                                    .moveByOffset(1, 0)//.pause(2000)
+                                    .perform();
+                            //actionProvider.moveByOffset(-1,0).pause(2000).perform();
+                            System.out.println("Margin Left after - " + slideButton.getCssValue("margin-left"));
+                            //Thread.sleep(5000);
                         } catch (Exception e) {
                             System.out.println(e.getMessage());
                             driver.get("https://support.oppo.com/in/warranty-check/");
                             break;
+                        }
-                        Mat circles = getMatCircles(driver, fileName);
+                        double[] circles = getMatCircles2(driver, fileName);
-                        if (circles.cols() == 2) {
+                        if (circles != null) {
-                            double firstCircleX = circles.get(0, 0)[0];
+                            double firstCircleX = circles[0];
-                            double secondCircleX = circles.get(0, 1)[0];
+                            double secondCircleX = circles[1];
                             double distance = Math.abs(firstCircleX - secondCircleX);
+                            System.out.println("Distance is " + distance);
                             System.out.println("Starting Checkdistance method");
-                            int moveBy = checkDistance(driver, wait10Sec, actionProvider, distance);
+                            int moveByOffset = checkDistance(driver, wait10Sec, actionProvider, distance, slideButton);
-                            if (moveBy > 0) {
+                            if (moveByOffset > 0) {
+                                System.out.println("Margin Offset - " + slideButton.getCssValue("margin-left"));
-                                System.out.println("Ending Checkdistance method " + moveBy);
+                                System.out.println("Ending Checkdistance method " + moveByOffset);
-                                actionProvider.moveByOffset(moveBy, 0).release().perform();
+                                actionProvider.moveByOffset(moveByOffset, 0).release().perform();
                                 System.out.println("Move click performed");
                                 try {
                                     try {
                                         wait5Sec.until(ExpectedConditions.visibilityOfElementLocated(By.className("dx_captcha_basic_bar-inform")));
                                         System.out.println("Failed = " + counter);
                                     } catch (Exception notFailedException) {
                                         System.out.println("Success  at attempt " + counter);
-                                        List<WebElement> warrantyDateElements = driver.findElements(By.className("warranty-service--result_info__dateValue"));
+                                        //List<WebElement> warrantyDateElements = driver.findElements(By.className("warranty-service--result_info__dateValue"));
+                                        WebElement activationTimeElement = driver.findElement(By.xpath("//*[contains(text(), 'UTC+5.5')]"));
+//TODO: IF not found activationElement
-                                        if (warrantyDateElements.size() > 0) {
+                                        if (activationTimeElement != null) {
-                                            String warrantyDate = warrantyDateElements.get(0).getText();
+                                            String activationDateString = activationTimeElement.getText().trim();
-                                            System.out.println("Date --  " + warrantyDate);
+                                            System.out.println("Date --  " + activationDateString);
                                             try {
-                                                dateMap.put(imei, LocalDate.parse(warrantyDate, DateTimeFormatter.ofPattern("yyyy/MM/dd")).minusYears(1));
+                                                dateMap.put(imei, LocalDate.parse(activationDateString.split(" ")[0], DateTimeFormatter.ofPattern("dd/MM/yyyy")));
                                             } catch (Exception e) {
                                                 dateMap.put(imei, null);
+                                            }
                                         } else {
                                             System.out.println("Could not capture date");
                                             dateMap.put(imei, null);
+                                        }
-                                        WebElement checkAgainButton = driver.findElement(By.className("van-button--normal"));
+                                        WebElement checkAgainButton = driver.findElement(By.className("c-btn"));
                                         captchaNotBroken = false;
                                         checkAgainButton.click();
                                         break;
+                                    }
                                 } catch (Throwable e) {
         driver.close();
         driver.quit();
         return dateMap;
+    }
-    private int checkDistance(WebDriver driver, WebDriverWait wait3Sec, Actions actionProvider, double distance) throws Exception {
+    private int checkDistance(WebDriver driver, WebDriverWait wait3Sec, Actions actionProvider, double distance, WebElement slideButton) throws Exception {
         //actionProvider.moveToElement(slideButton).clickAndHold().moveByOffset(1, 0).perform();
-        actionProvider.moveByOffset(15, 0).perform();//
+        actionProvider.moveByOffset(15, 0).perform();
         String fileName2 = "/tmp/" + Thread.currentThread().getName() + "-moved.png";
-        Mat movedCircles = getMatCircles(driver, fileName2);
+        double[] movedCircles = getMatCircles2(driver, fileName2);
-        if (movedCircles.cols() != 2) return 0;
+        if (movedCircles == null) return 0;
-        double movedFirst = movedCircles.get(0, 0)[0];
+        double movedFirst = movedCircles[0];
-        double movedSecond = movedCircles.get(0, 1)[0];
+        double movedSecond = movedCircles[1];
         //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 {
         Mat circles = new Mat();
         Imgproc.HoughCircles(medianBlur, circles, Imgproc.HOUGH_GRADIENT, 1, 15, 50, 20, 17, 25);
         return circles;
+    }
+    private double[] getMatCircles2(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 page
+        WebElement imageElement = driver.findElement(By.id("dx_captcha_basic_bg_1"));
+        Point point = imageElement.getLocation();
+        // Get width and height of the element
+        int eleWidth = imageElement.getSize().getWidth();
+        int eleHeight = imageElement.getSize().getHeight();
+        // Crop the entire page screenshot to get only element screenshot
+        BufferedImage eleScreenshot = fullImg.getSubimage(point.getX(), point.getY(),
+                eleWidth, eleHeight);
+        ImageIO.write(eleScreenshot, "png", screenshot);
+        // Copy the element screenshot to disk
+        File screenshotLocation = new File(fileName);
+        FileUtils.copyFile(screenshot, screenshotLocation);
+        Mat src = Imgcodecs.imread(fileName);
+        if (src.empty()) {
+            System.err.println("Cannot read image: " + fileName);
+            return null;
+        }
+        Mat gray = new Mat();
+        Imgproc.cvtColor(src, gray, Imgproc.COLOR_BGR2GRAY);
+        Imgproc.GaussianBlur(gray, gray, new Size(9, 9), 2, 2);
+        Mat circles = new Mat();
+        Imgproc.HoughCircles(gray, circles, Imgproc.HOUGH_GRADIENT, 1.0,
+, // min distance between centers
+, 20, // param1 (Canny high threshold), param2 (accumulator threshold)
+, 50); // minRadius, maxRadius (adjust as needed)
+        if (circles.cols() < 2) return null;
+        for (int i = 0; i < circles.cols(); i++) {
+            double[] c1 = circles.get(0, i);
+            for (int j = i + 1; j < circles.cols(); j++) {
+                double[] c2 = circles.get(0, j);
+                if (Math.abs(c1[2] - c2[2]) < 3) { // check similar radius
+                    return new double[] {
+                            c1[0], c2[0] // x-coordinates
+                    };
+                }
+            }
+        }
+        return null;
+    }
+}

Subversion Repositories SmartDukaan

(root)/trunk/profitmandi-cron/src/main/java/com/smartdukaan/cron/scheduled/CheckOppoWarrantyTask.java – Rev 32148 → 34413