package com.thanos.ebony2;

import static com.thanos.ebony2.bean.AntFormBrowser.BROWSER;

import com.thanos.ebony2.exception.NoSuchRow;

import java.util.ArrayList;

import java.util.List;

import org.openqa.selenium.By;

import org.openqa.selenium.WebElement;

import com.thanos.ebony2.bean.AntProp;

@SuppressWarnings(value = {"unchecked", "rawtypes"})

public class Test {

    public static void main(String[] args) { 　//测试代码

        BROWSER.open(AntProp.getProp("XXXURL"));

        BROWSER.pause(1);

        BROWSER.click("//*[@id=\"abc\"]/div/div/form/div[2]/div/div/div/span/div[1]/button");

        BROWSER.pause(1);

        System.out.println(getRowNo());

        System.out.println(getColNo());

        String a = getCell("A", 10); // 取A列第10行的值

        System.out.println("cellValue a: " + a);

        String b = getCell("Ｇ", 11); // 取G列第11行的值

        System.out.println("cellValue b: " + b);

    }

    /*

     * 参数: 列名 String tabName; 行数 int rowNo

     * 返回值: 某一个格子的值

     */

    public static String getCell(String colName, int rowNo) {

        int totalRow = getRowNo();

        if (rowNo < 1) {

            throw new NoSuchRow(String.format("row [%s]", rowNo));

        } else if(rowNo > totalRow) {

            throw new NoSuchRow(String.format("There's %s rows, your input %s is not reachable", totalRow, rowNo));

        } 

        List<WebElement> temp = BROWSER.getWebDriver().findElements(By.tagName("tr"));

        List<WebElement> currentRow = temp.subList(1, temp.size()-1);  //去掉表头那一行

        String[] rowData = currentRow.get(rowNo-1).getText().split(" ");  //注: 因第一行是表头(列名), 所以取单元格值时表头不计入rowNo. 即: rowNo传入1时所取的值是表头下面第一行的值.

        List<WebElement> tags = BROWSER.getWebDriver().findElements(By.tagName("th"));

        List<String> tagNames = new ArrayList();

        for (WebElement ele : tags) {

            String name = ele.getText();

            tagNames.add(name);

        }

//      Float cellValue = 0f;

        String cellValue = "";

        int index = tagNames.indexOf(colName);

//      cellValue = Float.parseFloat(rowData[index]);

        cellValue = rowData[index];

        return cellValue;

    }

    public static int getColNo() {

        List<WebElement> elements = BROWSER.getWebDriver().findElements(By.tagName("th"));

//        for (WebElement webElement : elements) {

//            System.out.println(webElement.getText());

//        }

        return elements.size();

    }

    public static int getRowNo() {

        List<WebElement> elements = BROWSER.getWebDriver().findElements(By.tagName("tr"));

//        for (WebElement webElement : elements) {

//            System.out.println(webElement.getText());

//        }

        return elements.size()-1;

    }

}

package com.thanos.ebony2;

import static com.thanos.ebony2.bean.AntFormBrowser.BROWSER;

import com.thanos.ebony2.exception.NoSuchRow;

import java.util.ArrayList;

import java.util.List;

import org.openqa.selenium.By;

import org.openqa.selenium.WebElement;

import com.thanos.ebony2.bean.AntProp;

@SuppressWarnings(value = {"unchecked", "rawtypes"})

public class Test {

    public static void main(String[] args) {  //测试代码

        BROWSER.open(AntProp.getProp("XXXURL"));

        BROWSER.pause(1);

        BROWSER.click("//*[@id=\"abc\"]/div/div/form/div[2]/div/div/div/span/div[1]/button");

        BROWSER.pause(1);

        System.out.println(getRowNo());

        System.out.println(getColNo());

        String b = getCellValue(10, 1);  //取第10行第1列的值

        System.out.println("cellValue b: " + b);

    }

    /*

     * 参数: 列 int colNo, 行 int rowNo

     * 返回值: 某一个格子的值

     */

    public static String getCellValue(int colNo, int rowNo) {

        List<WebElement> tempR = BROWSER.getWebDriver().findElements(By.tagName("tr"));

        List<WebElement> tempRow = tempR.subList(1, tempR.size()-1);  //去掉表头(表头不计入行数)

        int totalRow = getRowNo();

        int totalCol = getColNo();

        Object[][] table = new Object[totalRow][totalCol];

        for (int i = 0; i < tempRow.size(); i++) {

            List<WebElement> currCol = tempRow.get(i).findElements(By.tagName("td"));

            for (int j = 0; j < currCol.size(); j++) {

                table[i][j] = currCol.get(j).getText();

            }

        }

        return (String) table[colNo-1][rowNo-1];  //返回单元格的值

    }

    public static int getColNo() {

        List<WebElement> elements = BROWSER.getWebDriver().findElements(By.tagName("th"));

//        for (WebElement webElement : elements) {

//            System.out.println(webElement.getText());

//        }

        return elements.size();

    }

    public static int getRowNo() {

        List<WebElement> elements = BROWSER.getWebDriver().findElements(By.tagName("tr"));

//        for (WebElement webElement : elements) {

//            System.out.println(webElement.getText());

//        }

        return elements.size()-1;

    }

}