【java 菜鸟自动化实践之二】采用PIO将图片存入指定的excel单元格中(支持excel2003和2007)

实现目的：selenium实时截取浏览器图，并采用PIO将图片存入指定的excel单元格中(支持excel2003和2007)

package common;

import org.apache.log4j.Logger;
import org.apache.poi.hssf.usermodel.HSSFClientAnchor;
import org.apache.poi.hssf.usermodel.HSSFPatriarch;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFClientAnchor;
import org.apache.poi.xssf.usermodel.XSSFDrawing;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import java.awt.image.BufferedImage;
import java.io.*;

import javax.imageio.ImageIO;

/**
 * User: XZee Date: 
 * 2011-10-10 Time: 16:10:29
 */
 
public class write_excel {

	private static Logger log = Logger.getLogger(write_excel.class);
	/** 总行数 */
	private int totalRows = 0;
	/** 总列数 */
	private int totalCells = 0;
	/** 错误信息 */
	private static String errorInfo;
	private static XSSFWorkbook workbook;
	private static HSSFWorkbook workbook2;
	private static WebDriver driver;

	/**
	 * 构造方法
	 * 
	 * @return
	 */

	public int getTotalRows() {
		return totalRows;
	}

	public int getTotalCells() {
		return totalCells;
	}

	public String getErrorInfo() {
		return errorInfo;
	}

	public static boolean validateExcel(String filePath) {
		/** 检查文件名是否为空或者是否是Excel格式的文件 */
		if (filePath == null
				|| !(isExcel2003(filePath) || isExcel2007(filePath))) {
			errorInfo = "文件名不是excel格式";
			return false;
		}
		/** 检查文件是否存在 */
		File file = new File(filePath);
		if (file == null || !file.exists()) {
			errorInfo = "文件不存在";
			return false;
		}
		return true;
	}

	public static boolean isExcel2003(String filePath) {
		return filePath.matches("^.+\\.(?i)(xls)$");
	}

	public static boolean isExcel2007(String filePath) {
		return filePath.matches("^.+\\.(?i)(xlsx)$");
	}

	public static void update_Excel_Screenshot(String filePath, int sheetIndex, int row, int col, File value) {
		BufferedImage bufferImage = null;
		FileOutputStream fileOutputStream = null;
		try {
			FileInputStream fis = new FileInputStream(filePath);
			// 先把读入的图片放到第一个 ByteArrayOutputStream 中，用于产生ByteArray
			ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
			bufferImage = ImageIO.read(value);
			ImageIO.write(bufferImage, "JPEG", byteArrayOutputStream);
			if (!validateExcel(filePath)) {
				log.error(errorInfo);
			}
			if (isExcel2007(filePath)) {
				workbook = new XSSFWorkbook(fis);
				XSSFSheet sheet = workbook.getSheetAt(sheetIndex);
				// 准备插入图片
				XSSFDrawing patriarch = sheet.createDrawingPatriarch();
				int cols = col + 1;
				int rows = row - 1;
				XSSFClientAnchor anchor = new XSSFClientAnchor(0, 0, 100, 100, (short) col, rows, (short) cols, row);
				// anchor.setAnchorType(3);
				// 准备插入图片
				byte[] pictureData = byteArrayOutputStream.toByteArray();
				int pictureFormat = XSSFWorkbook.PICTURE_TYPE_JPEG;
				int pictureIndex = workbook.addPicture(pictureData,pictureFormat);
				patriarch.createPicture(anchor, pictureIndex);
			} else if (isExcel2003(filePath)) {
				workbook2 = new HSSFWorkbook(fis);
				HSSFSheet sheet = workbook2.getSheetAt(sheetIndex);
				HSSFPatriarch patriarch = sheet.createDrawingPatriarch();
				int cols = col + 1;
				int rows = row - 1;
				// 准备插入图片
				HSSFClientAnchor anchor = new HSSFClientAnchor(0, 0, 100, 100, (short) col, rows, (short) cols, row);
				byte[] pictureData = byteArrayOutputStream.toByteArray();
				int pictureFormat = HSSFWorkbook.PICTURE_TYPE_JPEG;
				int pictureIndex = workbook.addPicture(pictureData, pictureFormat);
				patriarch.createPicture(anchor, pictureIndex);
			}
			fileOutputStream = new FileOutputStream(filePath);
			// 写入excel
			workbook.write(fileOutputStream);
			byteArrayOutputStream.close();
			fileOutputStream.close();
			fis.close();
		} catch (Exception e) {
			log.info(e.getMessage());
		} finally {
			if (fileOutputStream != null) {
				try {
					fileOutputStream.close();
				} catch (IOException io) {
					log.info(io.getMessage());
				}
			}
		}
	}

	// 实时截取页面图，并存入到excel中
	public static void write_excel_Screenshot(String filePath, int i) throws Exception {
		try {
			// 截图
			File screenShotFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
			// 将图片存入第1行的第3列的单元格中
			update_Excel_Screenshot(filePath, 0, i, 3, screenShotFile); 
			log.info("用例执行失败后，成功保存当前操作失败的图片！");
		} catch (Exception e) {
			log.error(e.getMessage());
		}
	}

	public static void main(String[] args) throws Exception {
		File file_chrome = new File("C:\\Program Files (x86)\\Google\\Chrome\\Application\\chromedriver.exe");
		System.setProperty("webdriver.chrome.driver", file_chrome.getAbsolutePath());
		ChromeOptions options = new ChromeOptions();
		options.addArguments("disable-infobars");
		// 设置成用户自己的数据目录
		options.addArguments("user-data-dir=C:\\Users\\Administrator\\AppData\\Local\\Google\\Chrome\\User Data"); 
		// 打开chrome浏览器
		WebDriver driver = new ChromeDriver(options);
		System.out.println("End chrome browser...");
		driver.get("http://www.baidu.com");
		int row = 1; // 行
		String filePath =".\\xxxxx\\xxxx.xlsx";
		write_excel_Screenshot(filePath, row);
	}
}