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 lombok.extern.slf4j.Slf4j;
|
|
import java.util.ArrayList;
|
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
|
*/
|
@Slf4j
|
@EqualsAndHashCode(callSuper = true)
|
@Data
|
@Schema(description = "ocr解析火车票模型")
|
@NoArgsConstructor
|
public class TrainTicketModel extends OcrModel {
|
@Schema(description = "票号")
|
private String ticketNum;
|
@Schema(description = "出发地")
|
private String startingStation;
|
@Schema(description = "目的地")
|
private String destinationStation;
|
@Schema(description = "车次")
|
private String trainNum;
|
|
@Schema(description = "发车日期")
|
private Long dateTime;
|
@Schema(description = "票价")
|
private String ticketRates;
|
@Schema(description = "售票地点")
|
private String ticketLocation;
|
@Schema(description = "序列号")
|
private String serialNum;
|
@Schema(description = "姓名")
|
private String name;
|
@Schema(description = "身份证号")
|
private String idNum;
|
@Schema(description = "座位号")
|
private String seat;
|
@Schema(description = "座位类型")
|
private String seatCategory;
|
@Schema(description = "检票口")
|
private String ticketGate;
|
|
@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() {
|
List<String> list = new ArrayList<>(this.getStrList());
|
ticketNum = list.get(0).matches("(?:[A-Za-z]*\\d+[A-Za-z]*\\d*|[A-Za-z]+\\d+)") ? list.get(0) : list.get(1);
|
//起始终止站
|
startingStation();
|
//序列号
|
serialNum();
|
//身份证
|
idNum();
|
int index = 0;
|
for (String str : list) {
|
dateTime(str);
|
ticketRates(str);
|
ticketLocation(str);
|
seat(str);
|
seatCategory(str);
|
ticketGate(str);
|
//用户名称
|
if (StringUtil.isNotEmpty(idNum) && str.contains(idNum)) {
|
if (!str.equals(idNum)) {
|
if (isAllChinese(str.substring(idNum.length()))) {
|
name = str.substring(idNum.length());
|
}
|
} else if (index + 1 < list.size()) {
|
if (isAllChinese(list.get(index + 1))) {
|
name = list.get(index + 1);
|
} else if (isAllChinese(list.get(index - 1))) {
|
name = list.get(index - 1);
|
}
|
}
|
}
|
index++;
|
}
|
|
}
|
|
private void startingStation() {
|
if (StringUtil.isEmpty(startingStation)) {
|
int index = 0;
|
List<String> strList = this.getStrList();
|
int size = strList.size();
|
|
for (String str : strList) {
|
if (isAllChinese(str) && index + 2 < size && isTrainNum(strList.get(index + 1))
|
&& (isAllChinese(strList.get(index + 2)) || isAllCaracter(strList.get(index + 2)))) {
|
startingStation = str.equals("站") ? strList.get(index - 1) + str : str;
|
trainNum = strList.get(index + 1);
|
destinationStation = strList.get(index + 2);
|
if (isAllCaracter(strList.get(index + 2))) {
|
String[] split = strList.get(index + 1).split("次");
|
trainNum = split.length > 0 ? split[0] : null;
|
destinationStation = split.length > 1 ? split[1] : null;
|
}
|
break;
|
}
|
|
index++;
|
}
|
//当顺序不正常，做一下补充填写
|
int firstPY = 0;
|
for (int i = 0; i < strList.size(); i++) {
|
if (isAllCaracter(strList.get(i))) {
|
firstPY = i;
|
break;
|
}
|
}
|
if (StringUtil.isEmpty(destinationStation) && firstPY > 0) {
|
for (String s : strList.subList(firstPY - 3, firstPY)) {
|
if (containsNumber(s)) {
|
trainNum = s;
|
} else {
|
// 第一个非数字字符串作为出发站，第二个作为目的站
|
if (StringUtil.isEmpty(startingStation)) {
|
startingStation = s;
|
} else {
|
destinationStation = s;
|
}
|
}
|
}
|
}
|
}
|
}
|
|
private void dateTime(String str) {
|
if (dateTime == null && str.matches("\\d{4}年\\d{1,2}月\\d{1,2}日\\d{1,2}[:：]\\d{1,2}")) {
|
try {
|
String[] split = str.split("年");
|
String year = split[0];
|
String[] split2 = split[1].split("月");
|
String month = String.format("%02d", Integer.parseInt(split2[0]));
|
String[] split3 = split2[1].split("日");
|
String day = String.format("%02d", Integer.parseInt(split3[0]));
|
String[] split4 = split3[1].contains(":") ? split3[1].split(":") : split3[1].split("：");
|
String hh = String.format("%02d", Integer.parseInt(split4[0]));
|
String mmStr = split4[1].length() > 2 || split4[1].contains("开") ? split4[1].split("开")[0] : split4[1];
|
String mm = String.format("%02d", Integer.parseInt(mmStr));
|
String bd = year + "-" + month + "-" + day + " " + hh + ":" + mm;
|
dateTime = DateUtil.checkDate(bd, "yyyy-MM-dd HH:mm").getTime();
|
} catch (Exception e) {
|
log.error("火车票解析错误：" + e.getMessage());
|
}
|
}
|
}
|
|
private void ticketRates(String str) {
|
if (StringUtil.isEmpty(ticketRates) && StringUtil.isNotEmpty(str) && str.matches("[￥¥#]?\\d+(?:\\.\\d+)?")) {
|
ticketRates = str.replaceAll("[￥¥#元]", "");
|
}
|
}
|
|
private void ticketLocation(String str) {
|
if (StringUtil.isEmpty(ticketLocation) && str.endsWith("售")) {
|
ticketLocation = str.substring(0, str.length() - 1);
|
}
|
}
|
|
private void serialNum() {
|
if (StringUtil.isEmpty(serialNum)) {
|
List<String> strList = this.getStrList();
|
int size = strList.size();
|
for (int i = size - 1; i >= 0; i--) {
|
if (isNumAndCarater(strList.get(i)) && strList.get(i).length() > 4) {
|
serialNum = strList.get(i);
|
return;
|
}
|
}
|
}
|
}
|
|
private void seat(String str) {
|
if (StringUtil.isEmpty(seat) && str.matches("\\d{1,2}车(?:\\d{1,3}[A-Z]?号?|无座)(?:下铺|上铺|中铺|硬座|硬卧|软卧)?")) {
|
seat = str;
|
}
|
}
|
|
private void seatCategory(String str) {
|
String regex = "(?:新空调)?(?:软卧|硬卧|软座|硬座|商务座|一等座|二等座|特等座|动卧|一等卧|二等卧|高级动卧|无座|站票)(?:普快)?";
|
Matcher matcher = Pattern.compile(regex).matcher(str);
|
if (StringUtil.isEmpty(seatCategory) || matcher.matches()) {
|
seatCategory = str;
|
}
|
}
|
|
private void idNum() {
|
if (StringUtil.isEmpty(idNum)) {
|
String regex = "[1-9]\\d{5}(?:\\d{4}|\\*{4})(?:\\d{4}|\\*{4})\\d{3}[0-9Xx]";
|
Matcher matcher = Pattern.compile(regex).matcher(this.getContent());
|
if (matcher.find()) {
|
idNum = matcher.group();
|
}
|
|
}
|
}
|
|
private void ticketGate(String str) {
|
if (StringUtil.isEmpty(ticketGate) && (str.contains("检票") || str.contains("候车"))) {
|
ticketGate = str;
|
}
|
}
|
|
private boolean isAllChinese(String str) {
|
return str.matches("[一-龥]+");
|
}
|
|
private boolean isAllCaracter(String str) {
|
return str.matches("[a-zA-Z]+");
|
}
|
|
private boolean isNumAndCarater(String str) {
|
return str.matches("[0-9a-zA-Z]+");
|
}
|
|
private boolean isTrainNum(String str) {
|
return str.matches("([GCDZTK]?\\d{1,4})(次)?(?:\\s*(\\S+))?");
|
}
|
|
public static boolean containsNumber(String str) {
|
Pattern pattern = Pattern.compile("\\d");
|
Matcher matcher = pattern.matcher(str);
|
return matcher.find();
|
}
|
}
|
