Apache POI入门学习

Apache POI入门学习

    • 官网地址
  • excel中使用到的类
  • 读取excel表格内容
    • 表格内容
    • maven依赖
    • 方式一
      • 测试结果
    • 方式二
      • 测试结果
  • 向excel中写入数据
    • 方式一
    • 方式二
    • 方式三
    • 测试结果
  • 从 Excel 工作表中的公式单元格读取数据
    • 测试结果
  • Excel 工作表中写入公式单元格
  • 从受密码保护的Excel中读取数据
    • 方式一
      • 测试结果
    • 方式二
      • 测试结果
  • 填充背景色和前景色到单元格中
    • 测试结果
  • 将HashMap格式写入表格
    • 测试结果
  • 从excel中读取数据到HashMap中
    • 测试结果
  • 从数据库中读取数据并写入Excel
    • 测试结果
  • 从Excel中读取数据并写入数据库表
    • excel内容
    • 生成的数据表
  • Selenium中的数据驱动测试
    • maven依赖
    • 获取提交按钮的Xpath
    • 工具类
    • 案例代码
  • 在Selenium中将WebTable数据写入Excel表（网页数据抽取）
    • 测试结果
  • 在Excel中单元格内容为日期格式
    • 测试结果
  • 代码地址

官网地址

https://poi.apache.org/index.html
Apache POI提供API给Java程序对Microsoft Office格式档案读和写的功能。POI为“Poor Obfuscation Implementation”的首字母缩写，意为“简洁版的模糊实现”。

excel中使用到的类

一个excel表格文件包含一个工作簿，一个工作簿包含多个工作表，每个工作表包含多个行，每行包含多个单元格。
在这里插入图片描述

读取excel表格内容

表格内容

在这里插入图片描述

maven依赖

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd"><modelVersion>4.0.0</modelVersion><parent><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-parent</artifactId><version>3.2.4</version><relativePath/> <!-- lookup parent from repository --></parent><groupId>org.example</groupId><artifactId>ApachePoi-Demo</artifactId><version>1.0-SNAPSHOT</version><properties><maven.compiler.source>17</maven.compiler.source><maven.compiler.target>17</maven.compiler.target><project.build.sourceEncoding>UTF-8</project.build.sourceEncoding></properties><dependencies><!--poi 是基础依赖，提供了操作 Excel 文件的核心功能--><dependency><groupId>org.apache.poi</groupId><artifactId>poi</artifactId><version>5.2.5</version></dependency><!--poi-ooxml 是操作 Office Open XML 格式文件（如 .xlsx、.docx 等）的扩展库。--><dependency><groupId>org.apache.poi</groupId><artifactId>poi-ooxml</artifactId><version>5.2.5</version></dependency><!--spring web--><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-web</artifactId></dependency><!--lombok--><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId><optional>true</optional></dependency><!--spring boot单元测试--><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-test</artifactId><scope>test</scope></dependency></dependencies><build><plugins><plugin><groupId>org.springframework.boot</groupId><artifactId>spring-boot-maven-plugin</artifactId><version>3.1.2</version><configuration><excludes><exclude><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId></exclude></excludes></configuration></plugin></plugins></build></project>

方式一

for循环的方式读取数据

package com.apache.poi.demo.controller;import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;import java.io.InputStream;@RestController
public class DemoController {@PostMapping(value = "/poi/excel/import")public void excelImport(@RequestParam("file") MultipartFile file) throws Exception {// 输入流InputStream inputStream = file.getInputStream();// 创建一个工作簿XSSFWorkbook workbook = new XSSFWorkbook(inputStream);// 获取工作表getSheetAt,索引从0开始表示第一个工作表XSSFSheet sheet = workbook.getSheetAt(0);// 获取工作表中对应的行数int rows = sheet.getLastRowNum();for (int r = 0; r <= rows; r++) {// 获取指定的行XSSFRow row = sheet.getRow(r);// 获取指定的行数所对应的单元格int cols = sheet.getRow(r).getLastCellNum();for (int c = 0; c < cols; c++) {// 获取指定行指定的单元格XSSFCell cell = row.getCell(c);// 单元格类型,可以是字符串也可以是数字，也可以是布尔值CellType cellType = cell.getCellType();switch (cellType) {case STRING -> {// 字符串类型获取对应的字符串值System.out.print(cell.getStringCellValue());break;}case NUMERIC -> {// 数字类型获取对应的数字System.out.print(cell.getNumericCellValue());break;}case BOOLEAN -> {// 布尔类型获取对应的布尔值System.out.print(cell.getBooleanCellValue());break;}}System.out.print(" | ");}System.out.println();}}
}

测试结果

在这里插入图片描述
在这里插入图片描述

方式二

使用Iterator进行遍历表格数据

package com.apache.poi.demo.controller;import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;import java.io.InputStream;
import java.util.Iterator;@RestController
public class DemoTwoController {@PostMapping(value = "/poi/excel/import2")public void excelImportTwo(@RequestParam("file") MultipartFile file) throws Exception {// 输入流InputStream inputStream = file.getInputStream();// 创建一个工作簿XSSFWorkbook workbook = new XSSFWorkbook(inputStream);// 获取工作表getSheetAt,索引从0开始表示第一个工作表XSSFSheet sheet = workbook.getSheetAt(0);Iterator<Row> rowIterator = sheet.rowIterator();while (rowIterator.hasNext()) {// 获取指定的行XSSFRow row = (XSSFRow) rowIterator.next();// 获取单元格Iterator<Cell> cellIterator = row.cellIterator();while (cellIterator.hasNext()) {XSSFCell cell = (XSSFCell) cellIterator.next();CellType cellType = cell.getCellType();switch (cellType) {// 字符串类型获取对应的字符串值case STRING -> System.out.print(cell.getStringCellValue());// 数字类型获取对应的数字case NUMERIC -> System.out.print(cell.getNumericCellValue());//布尔类型获取对应的布尔值case BOOLEAN -> System.out.print(cell.getBooleanCellValue());}System.out.print(" | ");}System.out.println();}}
}

测试结果

在这里插入图片描述
在这里插入图片描述

向excel中写入数据

方式一

通过使用for循环的方式向excel中写入数据

package com.apache.poi.demo;import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;/*** 向excel中写入数据的步骤* Workbook -> Sheet -> Rows -> Cells*/
public class WriteExcelDemo {public static void main(String[] args) throws IOException {// 创建工作簿XSSFWorkbook workbook = new XSSFWorkbook();// 工作表名称为Emp InfoXSSFSheet sheet = workbook.createSheet("Emp Info");Object empdata[][] = {{"EmpID", "Name", "Job"},{101, "David", "Enginner"},{102, "Smith", "Manager"},{103, "Scott", "Analyst"}};for (int r = 0; r < empdata.length; r++) {// 创建表格行XSSFRow row = sheet.createRow(r);for (int c = 0; c < empdata[r].length; c++) {// 创建单元格Cell cell = row.createCell(c);Object value = empdata[r][c];// 判断值的类型,然后进行转换if (value instanceof String) {cell.setCellValue((String) value);} else if (value instanceof Integer) {cell.setCellValue((Integer) value);} else if (value instanceof Boolean) {cell.setCellValue((Boolean) value);}}}File dir = new File(".\\datafiles");if (!dir.exists()) {// 文件夹不存在则自动创建dir.mkdir();}String filePath = ".\\datafiles\\employee.xlsx";// new FileOutputStream(filePath, true);true表示文件不存在时候，自动生成FileOutputStream fileOutputStream = new FileOutputStream(filePath, true);workbook.write(fileOutputStream);// 关闭流workbook.close();fileOutputStream.close();System.out.println("employee.xlsx写入成功");}
}

方式二

package com.apache.poi.demo;import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;/*** 向excel中写入数据的步骤* Workbook -> Sheet -> Rows -> Cells*/
public class WriteExcelDemo2 {public static void main(String[] args) throws IOException {// 创建工作簿XSSFWorkbook workbook = new XSSFWorkbook();// 工作表名称为Emp InfoXSSFSheet sheet = workbook.createSheet("Emp Info");Object empdata[][] = {{"EmpID", "Name", "Job"},{101, "David", "Enginner"},{102, "Smith", "Manager"},{103, "Scott", "Analyst"}};int rowCount = 0;// 数据存储在一维数组emp[]中,for (Object emp[] : empdata) {// rowCount++,每次值使用之后，自动加1XSSFRow row = sheet.createRow(rowCount++);int columnCount = 0;for (Object value : emp) {XSSFCell cell = row.createCell(columnCount++);// 判断值的类型,然后进行转换if (value instanceof String) {cell.setCellValue((String) value);} else if (value instanceof Integer) {cell.setCellValue((Integer) value);} else if (value instanceof Boolean) {cell.setCellValue((Boolean) value);}}}File dir = new File(".\\datafiles");if (!dir.exists()) {// 文件夹不存在则自动创建dir.mkdir();}// 写入当前项目下的datafiles目录下的employee.xlsxString filePath = ".\\datafiles\\employee.xlsx";// new FileOutputStream(filePath, true);true表示文件不存在时候，自动生成FileOutputStream fileOutputStream = new FileOutputStream(filePath, true);workbook.write(fileOutputStream);// 关闭流workbook.close();fileOutputStream.close();System.out.println("employee.xlsx写入成功");}
}

方式三

package com.apache.poi.demo;import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;/*** 向excel中写入数据的步骤* Workbook -> Sheet -> Rows -> Cells*/
public class WriteExcelDemo3 {public static void main(String[] args) throws IOException {// 创建工作簿XSSFWorkbook workbook = new XSSFWorkbook();// 工作表名称为Emp InfoXSSFSheet sheet = workbook.createSheet("Emp Info");ArrayList<Object[]> empData = new ArrayList<>();empData.add(new Object[]{"EmpID", "Name", "Job"});empData.add(new Object[]{101, "David", "Enginner"});empData.add(new Object[]{102, "Smith", "Manager"});empData.add(new Object[]{103, "Scott", "Analyst"});int rowNum = 0;for (Object[] emp : empData) {XSSFRow row = sheet.createRow(rowNum++);int cellNum = 0;for (Object value : emp) {XSSFCell cell = row.createCell(cellNum++);// 判断值的类型,然后进行转换if (value instanceof String) {cell.setCellValue((String) value);} else if (value instanceof Integer) {cell.setCellValue((Integer) value);} else if (value instanceof Boolean) {cell.setCellValue((Boolean) value);}}}// 写入当前项目下的datafiles目录下的employee.xlsxString filePath = ".\\datafiles\\employee.xlsx";// new FileOutputStream(filePath, true);true表示文件不存在时候，自动生成FileOutputStream fileOutputStream = new FileOutputStream(filePath, true);workbook.write(fileOutputStream);// 关闭流workbook.close();fileOutputStream.close();System.out.println("employee.xlsx写入成功");}
}

测试结果

方式一、方式二和方式三的代码，都会在项目的根目录下的datafiles文件夹中生成employee.xlsx
在这里插入图片描述
employee.xlsx的文件内容如下
在这里插入图片描述

从 Excel 工作表中的公式单元格读取数据

package com.apache.poi.demo;import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileInputStream;
import java.io.IOException;public class ReadDataFromFormulaCell {public static void main(String[] args) throws IOException {String filePath = ".\\datafiles\\readFormula.xlsx";FileInputStream fileInputStream = new FileInputStream(filePath);XSSFWorkbook workbook = new XSSFWorkbook(fileInputStream);XSSFSheet sheet = workbook.getSheetAt(0);int rows = sheet.getLastRowNum();int cols = sheet.getRow(0).getLastCellNum();for (int r = 0; r <= rows; r++) {XSSFRow row = sheet.getRow(r);for (int c = 0; c < cols; c++) {XSSFCell cell = row.getCell(c);switch (cell.getCellType()) {case STRING -> System.out.print(cell.getStringCellValue());case NUMERIC, FORMULA -> System.out.print(cell.getNumericCellValue());case BOOLEAN -> System.out.print(cell.getBooleanCellValue());}System.out.print("|");}System.out.println();}fileInputStream.close();}
}

在这里插入图片描述
在这里插入图片描述

测试结果

在这里插入图片描述

Excel 工作表中写入公式单元格

package com.apache.poi.demo;import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileOutputStream;
import java.io.IOException;/*** Excel 工作表中写入公式单元格*/
public class WriteDataFromFormulaCell {public static void main(String[] args) throws IOException {XSSFWorkbook workbook = new XSSFWorkbook();String sheetName = "Number";XSSFSheet sheet = workbook.createSheet(sheetName);XSSFRow row = sheet.createRow(0);row.createCell(0).setCellValue(10);row.createCell(1).setCellValue(20);row.createCell(2).setCellValue(30);/*** setCellFormula设置单元格公式,A1*B1*C1，单元格列名，* A1*B1*C1相当于单元格的值等于第一行第一个单元格乘以第一行第二个单元格乘以第一行第三个单元格*/row.createCell(3).setCellFormula("A1*B1*C1");String filePath = ".\\datafiles\\calc.xlsx";// new FileOutputStream(filePath, true);true表示文件不存在，自动生成FileOutputStream fileOutputStream = new FileOutputStream(filePath, true);workbook.write(fileOutputStream);fileOutputStream.close();workbook.close();}
}

生成文件
在这里插入图片描述
文件内容
在这里插入图片描述

从受密码保护的Excel中读取数据

一个excel文件，被密码进行保护，如何读取里面的数据。表格数据如下
在这里插入图片描述

方式一

package com.apache.poi.demo;import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileInputStream;
import java.io.IOException;/*** 从受密码保护的Excel中读取数据*/
public class ReadingPasswordProtectedExcel {public static void main(String[] args) throws IOException {String path = ".\\datafiles\\customers.xlsx";FileInputStream fis = new FileInputStream(path);// 密码String password = "test123";// 根据可能受密码保护的给定InputStream创建适当的HSSFWorkbook/XSSFWorkbook。XSSFWorkbook workbook = (XSSFWorkbook) WorkbookFactory.create(fis, password);// 获取指定的工作表XSSFSheet sheet = workbook.getSheetAt(0);// 一共有5行数据，索引从0开始int rows = sheet.getLastRowNum();// 一共有3列，索引从1开始int cols = sheet.getRow(0).getLastCellNum();// 使用for循环读取数据for (int r = 0; r <= rows; r++) {XSSFRow row = sheet.getRow(r);for (int c = 0; c < cols; c++) {XSSFCell cell = row.getCell(c);switch (cell.getCellType()) {case NUMERIC, FORMULA -> System.out.print(cell.getNumericCellValue());case STRING -> System.out.print(cell.getStringCellValue());case BOOLEAN -> System.out.print(cell.getBooleanCellValue());}System.out.print(" | ");}System.out.println();}// 关闭流workbook.close();fis.close();}
}

测试结果

在这里插入图片描述

方式二

package com.apache.poi.demo;import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.WorkbookFactory;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileInputStream;
import java.io.IOException;
import java.util.Iterator;/*** 从受密码保护的Excel中读取数据*/
public class ReadingPasswordProtectedExcel2 {public static void main(String[] args) throws IOException {String path = ".\\datafiles\\customers.xlsx";FileInputStream fis = new FileInputStream(path);// 密码String password = "test123";// 根据可能受密码保护的给定InputStream创建适当的HSSFWorkbook/XSSFWorkbook。XSSFWorkbook workbook = (XSSFWorkbook) WorkbookFactory.create(fis, password);// 获取指定的工作表XSSFSheet sheet = workbook.getSheetAt(0);// 使用迭代器来读取表格中的数据Iterator<Row> rowIterator = sheet.rowIterator();while (rowIterator.hasNext()) {Row row = rowIterator.next();Iterator<Cell> cellIterator = row.cellIterator();while (cellIterator.hasNext()) {Cell cell = cellIterator.next();switch (cell.getCellType()) {case NUMERIC, FORMULA -> System.out.print(cell.getNumericCellValue());case STRING -> System.out.print(cell.getStringCellValue());case BOOLEAN -> System.out.print(cell.getBooleanCellValue());}System.out.print(" | ");}System.out.println();}// 关闭流workbook.close();fis.close();}
}

测试结果

在这里插入图片描述

填充背景色和前景色到单元格中

package com.apache.poi.demo;import org.apache.poi.ss.usermodel.FillPatternType;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.xssf.usermodel.*;import java.io.FileOutputStream;
import java.io.IOException;public class FormattingCellColor {public static void main(String[] args) throws IOException {XSSFWorkbook workbook = new XSSFWorkbook();XSSFSheet sheet = workbook.createSheet("Sheet1");XSSFRow row = sheet.createRow(1);// 设置背景色XSSFCellStyle style = workbook.createCellStyle();style.setFillBackgroundColor(IndexedColors.BRIGHT_GREEN.getIndex());style.setFillPattern(FillPatternType.BIG_SPOTS);// 创建第一个单元格XSSFCell cell = row.createCell(1);cell.setCellValue("welcome");cell.setCellStyle(style);//设置前景色style = workbook.createCellStyle();style.setFillBackgroundColor(IndexedColors.YELLOW.getIndex());style.setFillPattern(FillPatternType.BIG_SPOTS);// 创建第二个单元格cell = row.createCell(2);cell.setCellValue("Automation");cell.setCellStyle(style);String path = ".\\datafiles\\style.xlsx";FileOutputStream fos = new FileOutputStream(path, true);workbook.write(fos);workbook.close();fos.close();System.out.println("Done!!!");}
}

测试结果

在这里插入图片描述
生成的表格样式
在这里插入图片描述

将HashMap格式写入表格

package com.apache.poi.demo;import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;/****/
public class HashMapExcel {public static void main(String[] args) throws IOException {XSSFWorkbook workbook = new XSSFWorkbook();XSSFSheet sheet = workbook.createSheet("Student data");Map<String, String> data = new HashMap<>();data.put("101", "John");data.put("102", "Smith");data.put("103", "Scott");data.put("104", "Kim");data.put("105", "Mary");int rowNo = 0;for (Map.Entry entry : data.entrySet()) {XSSFRow row = sheet.createRow(rowNo++);row.createCell(0).setCellValue((String) entry.getKey());row.createCell(1).setCellValue((String) entry.getValue());}FileOutputStream fos = new FileOutputStream(".\\datafiles\\student.xlsx", true);workbook.write(fos);workbook.close();fos.close();System.out.println("Excel Written Successfully!");}
}

测试结果

在这里插入图片描述
控制台输出
在这里插入图片描述

从excel中读取数据到HashMap中

package com.apache.poi.demo;import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileInputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.Map;/*** 把表格数据读取到HashMap中*/
public class ExcelToHashMap {public static void main(String[] args) throws IOException {String path = ".\\datafiles\\student.xlsx";FileInputStream fis = new FileInputStream(path);XSSFWorkbook workbook = new XSSFWorkbook(fis);XSSFSheet sheet = workbook.getSheet("Student data");int rows = sheet.getLastRowNum();HashMap<String, String> data = new HashMap<>();// 读取数据从表格到HashMapfor (int r = 0; r <= rows; r++) {String key = sheet.getRow(r).getCell(0).getStringCellValue();String value = sheet.getRow(r).getCell(1).getStringCellValue();data.put(key, value);}// 读取数据从HashMap中for (Map.Entry entry : data.entrySet()) {System.out.println(entry.getKey() + " " + entry.getValue());}System.out.println(data);workbook.close();fis.close();}
}

测试结果

在这里插入图片描述

从数据库中读取数据并写入Excel

springboot的maven项目添加postgresql数据库依赖

 <!--postgresql依赖--><dependency><groupId>org.postgresql</groupId><artifactId>postgresql</artifactId><scope>runtime</scope></dependency>

package com.apache.poi.demo;import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileOutputStream;
import java.io.IOException;
import java.sql.*;/*** 从数据库中读取数据并写入Excel*/
public class DataBaseToExcel {/*** url*/private static final String url = "jdbc:postgresql://localhost:5432/cps";/*** 用户名*/private static final String username = "postgres";/*** 密码*/private static final String password = "admin";public static void main(String[] args) throws SQLException, IOException {// 连接到数据库Connection connection = DriverManager.getConnection(url, username, password);// statement/queryStatement statement = connection.createStatement();String sql = "select * from departments";ResultSet rs = statement.executeQuery(sql);// 表格XSSFWorkbook workbook = new XSSFWorkbook();XSSFSheet sheet = workbook.createSheet("Locations Data");// 创建表格标题行XSSFRow row = sheet.createRow(0);row.createCell(0).setCellValue("department_id");row.createCell(1).setCellValue("department_name");row.createCell(2).setCellValue("manager_id");row.createCell(3).setCellValue("location_id");int rowNum = 1;while (rs.next()) {String departmentId = rs.getString("department_id");String departmentName = rs.getString("department_name");String managerId = rs.getString("manager_id");String locationId = rs.getString("location_id");row = sheet.createRow(rowNum++);row.createCell(0).setCellValue(departmentId);row.createCell(1).setCellValue(departmentName);row.createCell(2).setCellValue(managerId);row.createCell(3).setCellValue(locationId);}FileOutputStream fos = new FileOutputStream(".\\datafiles\\department.xlsx");workbook.write(fos);workbook.close();fos.close();//关闭数据库连接connection.close();}
}

测试结果

在这里插入图片描述

从Excel中读取数据并写入数据库表

excel内容

在这里插入图片描述

package com.apache.poi.demo;import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileInputStream;
import java.io.IOException;
import java.sql.*;/*** 从Excel中读取数据并写入数据库表*/
public class ExcelToDatabase {/*** url*/private static final String url = "jdbc:postgresql://localhost:5432/cps";/****/private static final String username = "postgres";/*** 密码*/private static final String password = "admin";public static void main(String[] args) throws SQLException, IOException {// 连接到数据库Connection connection = DriverManager.getConnection(url, username, password);// statement/queryStatement statement = connection.createStatement();String sql = "create table test.departments (department_id int4 not null,department_name varchar(30) not null,manager_id int4 null,location_id int4 null)";statement.execute(sql);// 表格FileInputStream fis = new FileInputStream(".\\datafiles\\department.xlsx");XSSFWorkbook workbook = new XSSFWorkbook(fis);XSSFSheet sheet = workbook.getSheet("Locations Data");int rows = sheet.getLastRowNum();for (int r = 1; r <= rows; r++) {XSSFRow row = sheet.getRow(r);double departmentId = Double.parseDouble(row.getCell(0).getStringCellValue());String departmentName = row.getCell(1).getStringCellValue();int managerId = Integer.parseInt(row.getCell(2).getStringCellValue());int locationId = Integer.parseInt(row.getCell(3).getStringCellValue());sql = "insert into test.departments values(?,?,?,?)";PreparedStatement preparedStatement = connection.prepareStatement(sql);preparedStatement.setDouble(1, departmentId);preparedStatement.setString(2, departmentName);preparedStatement.setInt(3, managerId);preparedStatement.setInt(4, locationId);preparedStatement.executeUpdate();// 提交sqlstatement.execute("commit");}workbook.close();fis.close();connection.close();}
}

生成的数据表

在这里插入图片描述

Selenium中的数据驱动测试

maven依赖

<!--selenium的java依赖-->
<!-- https://mvnrepository.com/artifact/org.seleniumhq.selenium/selenium-java --><dependency><groupId>org.seleniumhq.selenium</groupId><artifactId>selenium-java</artifactId><version>4.20.0</version></dependency><!--TestNg的依赖--><!-- https://mvnrepository.com/artifact/org.testng/testng --><dependency><groupId>org.testng</groupId><artifactId>testng</artifactId><version>7.10.1</version><scope>test</scope></dependency>

获取提交按钮的Xpath

在这里插入图片描述

自动化登录的地址:https://admin-demo.nopcommerce.com/login

工具类

package com.apache.poi.demo;import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.DataFormatter;
import org.apache.poi.ss.usermodel.FillPatternType;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;public class XLUtility {public FileInputStream fis;public FileOutputStream fos;public XSSFWorkbook workbook;public XSSFSheet sheet;public XSSFRow row;public XSSFCell cell;public CellStyle style;String path = null;XLUtility(String path) {this.path = path;}/*** 获取工作表中的行数** @param sheetName 工作表名* @return 表格行数* @throws IOException 抛出IO异常*/public int getRowCount(String sheetName) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);int rowCount = sheet.getLastRowNum();workbook.close();fis.close();return rowCount;}/*** 获取工作表中行所在的单元格数量** @param sheetName 工作表名* @param rowNum    表格行号* @return 行所在的单元格数量* @throws IOException 抛出IO异常*/public int getCellCount(String sheetName, int rowNum) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);row = sheet.getRow(rowNum);int cellCount = row.getLastCellNum();workbook.close();fis.close();return cellCount;}/*** 获取单元格内容** @param sheetName 工作表名* @param rowNum    表格行号* @param cellNum   单元格号* @return 单元格内容* @throws IOException 抛出IO异常*/public String getCellData(String sheetName, int rowNum, int cellNum) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);row = sheet.getRow(rowNum);cell = row.getCell(cellNum);DataFormatter formatter = new DataFormatter();String data;try {data = formatter.formatCellValue(cell);} catch (Exception e) {data = "";}workbook.close();fis.close();return data;}/*** 设置表格单元格内容** @param sheetName 工作表名* @param rowNum    表格行号* @param cellNum   单元格号* @param data      单元格内容* @throws IOException 抛出IO异常*/public void setCellData(String sheetName, int rowNum, int cellNum, String data) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);row = sheet.getRow(rowNum);cell = row.getCell(cellNum);cell.setCellValue(data);fos = new FileOutputStream(path);workbook.write(fos);workbook.close();fis.close();fos.close();}/*** 单元格填充颜色** @param sheetName 工作表名* @param rowNum    表格行号* @param cellNum   单元格号* @throws IOException 抛出IO异常*/public void fillGreenColor(String sheetName, int rowNum, int cellNum) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);row = sheet.getRow(rowNum);cell = row.getCell(cellNum);style = workbook.createCellStyle();style.setFillForegroundColor(IndexedColors.GREEN.getIndex());style.setFillPattern(FillPatternType.SOLID_FOREGROUND);cell.setCellStyle(style);workbook.write(fos);workbook.close();fis.close();fos.close();}
}

案例代码

package com.apache.poi.demo;import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.testng.Assert;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;import java.io.IOException;
import java.util.concurrent.TimeUnit;public class DataDriverTest {WebDriver driver;@BeforeClasspublic void setup() {// 将chromedriver下载放到自己的一个目录中，并在代码中设置System.setProperty("webdriver.chrome.driver", "..\\selenium-java-demo\\driver\\chromedriver.exe");//使用驱动实例并开启对话driver = new ChromeDriver();// 建立等待策略driver.manage().timeouts().implicitlyWait(500000, TimeUnit.SECONDS);// 窗口最大化driver.manage().window().maximize();}@Test(dataProvider = "LoginData")public void loginTest(String user, String pwd, String exp) {String url = "https://admin-demo.nopcommerce.com/login";// 打开的urldriver.get(url);WebElement txtEmail = driver.findElement(By.id("Email"));// 清空元素的输入txtEmail.clear();// 输入内容txtEmail.sendKeys(user);WebElement txtPassword = driver.findElement(By.id("Password"));txtPassword.clear();txtPassword.sendKeys(pwd);// LOG IN按钮的Xpath陆路径String xPath = "/html/body/div[6]/div/div/div/div/div[2]/div[1]/div/form/div[3]/button";//  点击LOG IN按钮driver.findElement(By.xpath(xPath)).click();String expTitle = "Dashboard / nopCommerce administration";String actTitle = driver.getTitle();if (exp.equals("Valid")) {if (expTitle.equals(actTitle)) {driver.findElement(By.linkText("Logout")).click();Assert.assertTrue(true);} else {Assert.assertTrue(false);}} else if (exp.equals("Invalid")) {if (expTitle.equals(actTitle)) {driver.findElement(By.linkText("Logout")).click();Assert.assertTrue(false);} else {Assert.assertTrue(true);}}}/*** 数据提供者* 必须返回Object[][]数组类型*/@DataProvider(name = "LoginData")public String[][] getData() throws IOException {// 从excel中获取数据String path = ".\\datafiles\\loginData.xlsx";XLUtility util = new XLUtility(path);int totalRows = util.getRowCount("Sheet1");int totalCols = util.getCellCount("Sheet1", 1);String loginData[][] = new String[totalRows][totalCols];for (int i = 1; i < totalRows; i++) {for (int j = 0; j < totalCols; j++) {String cellData = util.getCellData("Sheet1", i, j);loginData[i-1][j] = cellData;}}return loginData;}@AfterClassvoid tearDown() {driver.quit();}
}

在Selenium中将WebTable数据写入Excel表（网页数据抽取）

package com.apache.poi.demo;import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.DataFormatter;
import org.apache.poi.ss.usermodel.FillPatternType;
import org.apache.poi.ss.usermodel.IndexedColors;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;public class XLUtility {public FileInputStream fis;public FileOutputStream fos;public XSSFWorkbook workbook;public XSSFSheet sheet;public XSSFRow row;public XSSFCell cell;public CellStyle style;String path = null;XLUtility(String path) {this.path = path;}/*** 获取工作表中的行数** @param sheetName 工作表名* @return 表格行数* @throws IOException 抛出IO异常*/public int getRowCount(String sheetName) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);int rowCount = sheet.getLastRowNum();workbook.close();fis.close();return rowCount;}/*** 获取工作表中行所在的单元格数量** @param sheetName 工作表名* @param rowNum    表格行号* @return 行所在的单元格数量* @throws IOException 抛出IO异常*/public int getCellCount(String sheetName, int rowNum) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);row = sheet.getRow(rowNum);int cellCount = row.getLastCellNum();workbook.close();fis.close();return cellCount;}/*** 获取单元格内容** @param sheetName 工作表名* @param rowNum    表格行号* @param cellNum   单元格号* @return 单元格内容* @throws IOException 抛出IO异常*/public String getCellData(String sheetName, int rowNum, int cellNum) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);row = sheet.getRow(rowNum);cell = row.getCell(cellNum);DataFormatter formatter = new DataFormatter();String data;try {data = formatter.formatCellValue(cell);} catch (Exception e) {data = "";}workbook.close();fis.close();return data;}/*** 设置表格单元格内容** @param sheetName 工作表名* @param rowNum    表格行号* @param cellNum   单元格号* @param data      单元格内容* @throws IOException 抛出IO异常*/public void setCellData(String sheetName, int rowNum, int cellNum, String data) throws IOException {File xlFile = new File(path);if (!xlFile.exists()) {// 如果文件不存在就创建一个新文件workbook = new XSSFWorkbook();fos = new FileOutputStream(path);workbook.write(fos);}fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);if (workbook.getSheetIndex(sheetName) == -1) {// 工作表不存在就创建workbook.createSheet(sheetName);}sheet = workbook.getSheet(sheetName);if (sheet.getRow(rowNum) == null) {// 表格行不存在就创建row = sheet.createRow(rowNum);}row = sheet.getRow(rowNum);cell = row.createCell(cellNum);cell.setCellValue(data);fos = new FileOutputStream(path);workbook.write(fos);workbook.close();fis.close();fos.close();}/*** 单元格填充颜色** @param sheetName 工作表名* @param rowNum    表格行号* @param cellNum   单元格号* @throws IOException 抛出IO异常*/public void fillGreenColor(String sheetName, int rowNum, int cellNum) throws IOException {fis = new FileInputStream(path);workbook = new XSSFWorkbook(fis);sheet = workbook.getSheet(sheetName);row = sheet.getRow(rowNum);cell = row.getCell(cellNum);style = workbook.createCellStyle();style.setFillForegroundColor(IndexedColors.GREEN.getIndex());style.setFillPattern(FillPatternType.SOLID_FOREGROUND);cell.setCellStyle(style);workbook.write(fos);workbook.close();fis.close();fos.close();}
}
package com.apache.poi.demo;import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;import java.io.IOException;
import java.util.concurrent.TimeUnit;/*** 在Selenium中将WebTable数据写入Excel表（网页数据抽取）*/
public class WebTableToExcel {public static void main(String[] args) throws IOException {//设置环境变量String driverPath = ".\\selenium-java-demo\\driver\\chromedriver.exe";System.setProperty("webdriver.chrome.driver", driverPath);// 使用驱动实例开启会话WebDriver driver = new ChromeDriver();//等待策略driver.manage().timeouts().implicitlyWait(5, TimeUnit.SECONDS);// 窗口最大化driver.manage().window().maximize();//导航到的urlString url = "https://en.wikipedia.org/wiki/List_of_countries_and_dependencies_by_population";driver.get(url);String path = ".\\datafiles\\population.xlsx";XLUtility utility = new XLUtility(path);// 设置标题行utility.setCellData("Sheet1", 0, 0, "Location");utility.setCellData("Sheet1", 0, 1, "Population");utility.setCellData("Sheet1", 0, 2, "% of world");utility.setCellData("Sheet1", 0, 3, "Date");utility.setCellData("Sheet1", 0, 4, "Source");// 捕获表格数据WebElement table = driver.findElement(By.xpath("//*[@id=\"mw-content-text\"]/div[1]/table/tbody"));// 在网页中表格的行int size = table.findElements(By.xpath("tr")).size();for (int r = 1; r <= size; r++) {String location = table.findElement(By.xpath("tr[" + r + "]/td[1]")).getText();String population = table.findElement(By.xpath("tr[" + r + "]/td[2]")).getText();String world = table.findElement(By.xpath("tr[" + r + "]/td[3]")).getText();String date = table.findElement(By.xpath("tr[" + r + "]/td[4]")).getText();String source = table.findElement(By.xpath("tr[" + r + "]/td[5]")).getText();System.out.println(location + " " + population + " " + world + " " + date + " " + source);utility.setCellData("Sheet1", r, 0, location);utility.setCellData("Sheet1", r, 1, population);utility.setCellData("Sheet1", r, 2, world);utility.setCellData("Sheet1", r, 3, date);utility.setCellData("Sheet1", r, 4, source);}System.out.println("Web scraping is done successfully...");driver.close();}
}

测试结果

生成population.xlsx
在这里插入图片描述

在Excel中单元格内容为日期格式

package com.apache.poi.demo;import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.CreationHelper;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Date;public class WorkingWithDateCells {public static void main(String[] args) throws IOException {// 创建一个空的工作簿XSSFWorkbook workbook = new XSSFWorkbook();// 创建一个工作表格XSSFSheet sheet = workbook.createSheet("Date formats");sheet.setColumnWidth(0, 256 * 20);XSSFCell cell = sheet.createRow(0).createCell(0);cell.setCellValue(new Date());CreationHelper creationHelper = workbook.getCreationHelper();// 格式化  yyyy-MM-dd HH:mm:ssCellStyle style = workbook.createCellStyle();style.setDataFormat(creationHelper.createDataFormat().getFormat("yyyy-MM-dd HH:mm:ss"));cell.setCellStyle(style);FileOutputStream fos = new FileOutputStream(".\\datafiles\\dataformats.xlsx", true);workbook.write(fos);fos.close();workbook.close();}
}

测试结果

在这里插入图片描述

代码地址

https://gitee.com/BAIXUEQIAN/java-study/tree/develop/Apache-Poi-Demo

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.hqwc.cn/news/671944.html

如若内容造成侵权/违法违规/事实不符，请联系编程知识网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

2024年03月 Scratch 图形化（三级）真题解析#中国电子学会#全国青少年软件编程等级考试

2024年03月 Scratch 图形化（三级）真题解析#中国电子学会#全国青少年软件编程等级考试

Scratch图形化等级考试（1~4级）全部真题・点这里 一、单选题（共18题，共50分） 第1题 运行程序后，角色一定不会说出的数字是？（ ） A：2 B：4 C：6 D：8 答案：A 程序中随机数的取值最小为 2，最大为 20 ，那么随机数加上 2 之后的结果的最小值为 4 ，最大值为 22 。所…
阅读更多...
查看微信小程序主包大小

查看微信小程序主包大小

前言 略 查看微信小程序主包大小 在微信开发者工具右上角找到“详情->基本信息” 查看微信小程序主包构成 通过微信开发者工具中的“代码依赖分析”工具查看
阅读更多...
第十三章 计算机网络

第十三章 计算机网络

这里写目录标题 1.网络设备2.协议簇2.1电子邮件(传输层)2.2地址解析(网际层)2.3DHCP(动态主动配置协议)2.4URL(统一资源定位器)2.5IP地址和子网掩码 1.网络设备 物理层&#xff1a;中继器&#xff0c;集线器(多路中继器) 数据链路层&#xff1a;网桥&#xff0c;交换机(多端口…
阅读更多...
Java常用5大集合类详解（实战）

Java常用5大集合类详解（实战）

文章目录 1、Collection1.1 iterator 迭代器 2、List 有序集合2.1 ArrayList ⭐2.2 LinkedList2.3 Queue 3、Set 无序集合3.1 HashSet ⭐3.2 TreeSet3.3 LinkedHashSet 4、Map 键值集合4.1 HashMap ⭐4.2 TreeMap / LinkedHashMap 5、工具类5.1 Collections5.2 Arrays 【拓展】…
阅读更多...
多组间比较散点图+误差棒（自备）

多组间比较散点图+误差棒（自备）

目录 数据 计算四分位值 作图 数据 rm(list ls()) library(ggplot2) library(dplyr) library(ggpubr) library(reshape2) library(tidyverse)data <- iris##鸢尾花数据集 dat <- data[,c(5,1)]#单个数据进行分析 计算四分位值 #根据分组计算四分位及中位数 dat1 …
阅读更多...
SegFix:预测边界和预测方向来修正边界

SegFix:预测边界和预测方向来修正边界

论文标题&#xff1a;SegFix: Model-Agnostic Boundary Refinement for Segmentation 论文地址&#xff1a;https://arxiv.org/pdf/2007.04269.pdf 代码地址&#xff1a;https://github.com/openseg-group/openseg.pytorch 两种loss监督 八种方向变回归问题为分类问题 代码地…
阅读更多...
GitOps介绍

GitOps介绍

基础设施即代码 IaC 在理解 GitOps 之前&#xff0c;需要先理解什么是基础设施即代码。 基础设施即代码&#xff08;Infrastructure as Code&#xff0c;简称IaC&#xff09;是一种软件工程实践&#xff0c;它将基础设施的管理和配置过程像管理代码一样进行版本控制、自动化和…
阅读更多...
电源小白入门学习7——USB充电、供电、电源路径管理

电源小白入门学习7——USB充电、供电、电源路径管理

电源小白入门学习7——USB充电、供电、电源路径管理 USB充电系统需要考虑的因素开关充电和线性充电充电路径管理输入限流路径管理&#xff08;动态功率管理&#xff09;理想二极管帮助提高电池利用率输入过充抑制 上期我们介绍了锂离子电池的电池特性&#xff0c;及充电电路设计…
阅读更多...
websevere服务器从零搭建到上线（二）｜Linux上的五种IO模型

websevere服务器从零搭建到上线（二）｜Linux上的五种IO模型

文章目录 阻塞 blocking非阻塞 non-blockingIO复用 IO multiplexing信号驱动 signal-driven异步 asynchronous拓展知识 看过上篇文章英国基本能理解本文五张图的内容websevere服务器从零搭建到上线&#xff08;一&#xff09;&#xff5c;阻塞、非阻塞、同步、异步 本文要能够在…
阅读更多...
关于冯诺依曼体系结构 和 操作系统（Operator System）的概念讲解（冯诺依曼体系结构，操作系统的作用等）

关于冯诺依曼体系结构 和 操作系统（Operator System）的概念讲解（冯诺依曼体系结构，操作系统的作用等）

目录 一、冯诺依曼体系结构 二、操作系统 1. 概念 2. 设计操作系统的目的 3.系统调用和库函数概念 4.总结 三、完结撒❀ 一、冯诺依曼体系结构 我们常见的计算机&#xff0c;如笔记本。我们不常见的计算机&#xff0c;如服务器&#xff0c;大部分都遵守冯诺依曼体系。 截…
阅读更多...
工业光源环形系列一平面无影光源特点

工业光源环形系列一平面无影光源特点

产品特点 ◆LED灯珠均匀排布经过漫射板特殊角度反射达到漫射效果&#xff1a; ◆光源均匀性高&#xff0c;漫射效果好。
阅读更多...
Navicat for MySQL Mac：数据库管理与开发的理想工具

Navicat for MySQL Mac：数据库管理与开发的理想工具

Navicat for MySQL Mac是一款功能强大的数据库管理与开发工具&#xff0c;专为Mac用户设计&#xff0c;旨在提供高效、便捷的数据库操作体验。 它支持创建、管理和维护MySQL和MariaDB数据库&#xff0c;通过直观的图形界面&#xff0c;用户可以轻松进行数据库连接、查询、编辑和…
阅读更多...
推荐文章
最新文章

Copyright @ 2022~2023