package jnpf.base.model.ocr.model;
|
|
import com.alibaba.fastjson.JSONObject;
|
import io.swagger.v3.oas.annotations.media.Schema;
|
import jnpf.base.model.ocr.OcrModel;
|
import jnpf.util.DateUtil;
|
import jnpf.util.StringUtil;
|
import lombok.Data;
|
import lombok.EqualsAndHashCode;
|
import lombok.NoArgsConstructor;
|
|
import java.util.ArrayList;
|
import java.util.Arrays;
|
import java.util.Date;
|
import java.util.List;
|
import java.util.regex.Matcher;
|
import java.util.regex.Pattern;
|
|
/**
|
* ocr解析驾驶证模型
|
*
|
* @author JNPF开发平台组
|
* @version v6.1.0
|
* @copyright 引迈信息技术有限公司
|
* @date 2025/7/29 16:58:47
|
*/
|
@EqualsAndHashCode(callSuper = true)
|
@Data
|
@Schema(description = "ocr解析驾驶证模型")
|
@NoArgsConstructor
|
public class DrivingLicModel extends OcrModel {
|
@Schema(description = "证号")
|
private String idNum;
|
@Schema(description = "姓名")
|
private String name;
|
@Schema(description = "性别")
|
private String sex;
|
@Schema(description = "国籍")
|
private String nationality;
|
@Schema(description = "住址")
|
private String address;
|
@Schema(description = "出生日期")
|
private Long birthDate;
|
@Schema(description = CCLZRQ)
|
private Long issueDate;
|
@Schema(description = "准驾车型")
|
private String classType;
|
@Schema(description = "有效期限")
|
private String validPeriod;
|
@Schema(description = "证件有效期起始日期")
|
private Long startDate;
|
@Schema(description = "证件有效期结束日期")
|
private Long endDate;
|
|
private static final String CCLZRQ = "初次领证日期";
|
private static final String DATE_DD = "yyyy-MM-dd";
|
|
@Override
|
public void extract(String url, JSONObject body) {
|
super.extract(url, body);
|
if (StringUtil.isEmpty(this.getContent()) || !this.getContent().contains("驾驶证")) {
|
return;
|
}
|
this.setInfo();
|
this.setContent(null);
|
this.setStrList(null);
|
}
|
|
private void setInfo() {
|
// 使用正则表达式提取关键信息
|
String ocrText = this.getContent();
|
this.cardNumber(ocrText);
|
this.setThisName();
|
this.sex();
|
this.validPeriod();
|
//for循环取单行数据，避免整体取到其他行合并信息
|
List<String> strList = this.getStrList();
|
for (int i = 0; i < strList.size(); i++) {
|
if (StringUtil.isEmpty(this.nationality) && strList.get(i).contains("国籍")) {
|
this.nationality = strList.get(i).endsWith("国籍") ? this.getStrList().get(i + 1) :
|
strList.get(i).substring(strList.get(i).indexOf("国籍") + 2);
|
}
|
if (StringUtil.isEmpty(this.address) && strList.get(i).contains("住址")) {
|
this.address = strList.get(i).endsWith("住址") ? this.getStrList().get(i + 1) :
|
strList.get(i).substring(strList.get(i).indexOf("住址") + 2);
|
}
|
if (StringUtil.isEmpty(this.classType) && strList.get(i).contains("准驾车型")) {
|
this.classType = this.getStrList().get(i + 1);
|
}
|
|
if (this.birthDate == null && strList.get(i).contains("出生日期")) {
|
String s = strList.get(i).endsWith("出生日期") ? this.getStrList().get(i + 1) :
|
strList.get(i).substring(strList.get(i).indexOf("出生日期") + 6);
|
if (StringUtil.isNotEmpty(s)) {
|
Date date = DateUtil.checkDate(s, DATE_DD);
|
this.birthDate = date != null ? date.getTime() : null;
|
}
|
}
|
if (this.issueDate == null && strList.get(i).contains(CCLZRQ)) {
|
String s = strList.get(i).endsWith(CCLZRQ) ? this.getStrList().get(i + 1) :
|
strList.get(i).substring(strList.get(i).indexOf(CCLZRQ) + 6);
|
if (StringUtil.isNotEmpty(s)) {
|
Date date = DateUtil.checkDate(s, DATE_DD);
|
this.issueDate = date != null ? date.getTime() : null;
|
}
|
}
|
}
|
|
//补充
|
this.birthDate();
|
this.address();
|
this.nationality();
|
}
|
|
private List<String> getKeyword() {
|
return new ArrayList<>(Arrays.asList("中华人民", "Driving", "证号", "姓名", "性别", "国籍", "Name", "Na", "住址", "Add", "Address",
|
"Address", "初次", "领证日期", "准驾车型", "有效期限", "Valid", "Period", "出生日期", "of", "Birth", "Data", "First", "Issue", "Class"));
|
}
|
|
private boolean containKeyword(String str) {
|
for (String s : getKeyword()) {
|
if (str.contains(s)) {
|
return true;
|
}
|
}
|
return false;
|
}
|
|
/**
|
* 获取身份证号
|
*/
|
private void cardNumber(String str) {
|
// 18位身份证正则表达式
|
Pattern r = Pattern.compile("[1-9]\\d{13,16}[a-zA-Z0-9]");
|
Matcher m = r.matcher(str);
|
if (m.find()) {
|
this.idNum = m.group();
|
}
|
}
|
|
private void sex() {
|
// 取倒身份证倒数第二位的数字的奇偶性判断性别，二代身份证18位
|
if (StringUtil.isNotEmpty(idNum)) {
|
String substring = idNum.substring(idNum.length() - 2, idNum.length() - 1);
|
int parseInt = Integer.parseInt(substring);
|
if (parseInt % 2 == 0) {
|
sex = "女";
|
} else {
|
sex = "男";
|
}
|
}
|
}
|
|
private void setThisName() {
|
List<String> nameList = new ArrayList<>();
|
int i = this.getStrList().indexOf("住址");
|
if (i < 3) {
|
i = 6;
|
}
|
for (String str : this.getStrList().subList(0, i)) {
|
if (StringUtil.isNotEmpty(str)) {
|
boolean flag = false;
|
for (String n : this.getKeyword()) {
|
if (str.contains(n)) {
|
flag = true;
|
break;
|
}
|
}
|
if (!flag) {
|
nameList.add(str);
|
}
|
}
|
}
|
if (!nameList.isEmpty()) {
|
for (String str : nameList) {
|
String pattern = "[一-龥]{1,4}";
|
Pattern r = Pattern.compile(pattern);
|
Matcher m = r.matcher(str);
|
if (m.matches()) {
|
this.name = str;
|
return;
|
}
|
}
|
}
|
}
|
|
private void birthDate() {
|
if (birthDate == null && StringUtil.isNotEmpty(idNum)) {
|
String dateStr = idNum.substring(6, 14);
|
String year = dateStr.substring(0, 4);
|
String month = dateStr.substring(4, 6);
|
String day = dateStr.substring(6, 8);
|
String bd = year + "-" + month + "-" + day;
|
birthDate = DateUtil.stringToDates(bd).getTime();
|
}
|
}
|
|
private void validPeriod() {
|
for (String s : this.getStrList()) {
|
if (s.contains("-") && s.contains("至") && s.length() > 14) {
|
this.validPeriod = s;
|
}
|
}
|
if (StringUtil.isNotEmpty(validPeriod)) {
|
// 拆分字符串
|
String[] dates = validPeriod.replace(" ", "").split("至");
|
// 解析开始日期
|
Date sdate = DateUtil.checkDate(dates[0], DATE_DD);
|
startDate = sdate != null ? sdate.getTime() : null;
|
// 解析结束日期
|
Date edate = DateUtil.checkDate(dates[1], DATE_DD);
|
endDate = edate != null ? edate.getTime() : null;
|
}
|
}
|
|
private void address() {
|
if (StringUtil.isEmpty(address) || containKeyword(address)) {
|
for (String s : this.getStrList()) {
|
if ((s.contains("省") || s.contains("市") || s.contains("区")) && s.length() > 6) {
|
if (s.startsWith("住址")) s = s.substring(2);
|
this.address = s;
|
return;
|
}
|
}
|
}
|
}
|
|
private void nationality() {
|
if (StringUtil.isEmpty(nationality) || containKeyword(nationality)) {
|
for (String s : this.getStrList()) {
|
if (s.contains("CHN") || s.contains("中国")) {
|
this.nationality = s;
|
return;
|
}
|
}
|
}
|
}
|
}
|
