-集成百度智能识别企业营业执照

new
MsFang 1 year ago
parent 814b99043d
commit 8b091f6f05

@ -19,5 +19,25 @@
<description>
报销系统业务模块
</description>
<dependencies>
<dependency>
<groupId>com.baidu.aip</groupId>
<artifactId>java-sdk</artifactId>
<version>4.8.0</version>
</dependency>
<dependency>
<groupId>commons-io</groupId>
<artifactId>commons-io</artifactId>
<version>2.6</version>
</dependency>
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>fastjson</artifactId>
<version>1.2.80</version>
</dependency>
</dependencies>
</project>

@ -1,5 +1,6 @@
package cn.iocoder.yudao.module.bs.controller.admin.suppliercompany;
import cn.iocoder.yudao.module.bs.utils.BusinessLicense;
import org.springframework.web.bind.annotation.*;
import javax.annotation.Resource;
import org.springframework.validation.annotation.Validated;
@ -22,11 +23,13 @@ import cn.iocoder.yudao.framework.excel.core.util.ExcelUtils;
import cn.iocoder.yudao.framework.operatelog.core.annotations.OperateLog;
import static cn.iocoder.yudao.framework.operatelog.core.enums.OperateTypeEnum.*;
import static cn.iocoder.yudao.module.bs.utils.BaiduOcrHandler.handleBusinessLicense;
import cn.iocoder.yudao.module.bs.controller.admin.suppliercompany.vo.*;
import cn.iocoder.yudao.module.bs.dal.dataobject.suppliercompany.SupplierCompanyDO;
import cn.iocoder.yudao.module.bs.convert.suppliercompany.SupplierCompanyConvert;
import cn.iocoder.yudao.module.bs.service.suppliercompany.SupplierCompanyService;
import org.springframework.web.multipart.MultipartFile;
@Tag(name = "管理后台 - 供应商信息")
@RestController
@ -37,6 +40,17 @@ public class SupplierCompanyController {
@Resource
private SupplierCompanyService supplierCompanyService;
@PostMapping("/buildBusinessLicense")
@Operation(summary = "百度云营业执照识别")
@PreAuthorize("@ss.hasPermission('bs:supplier-company:create')")
public CommonResult<BusinessLicense> buildBusinessLicense(MultipartFile file) {
try {
return success(handleBusinessLicense(file));
} catch (Exception e) {
throw new RuntimeException(e);
}
}
@PostMapping("/create")
@Operation(summary = "创建供应商信息")
@PreAuthorize("@ss.hasPermission('bs:supplier-company:create')")

@ -0,0 +1,43 @@
package cn.iocoder.yudao.module.bs.utils;
/**
* @author MrFang
* @date 20230815 11:36
*/
public interface BaiduOcrConstant {
/**
*
*/
String CHARSET = "UTF-8";
/**
* API Key
*/
String CLIENT_ID = "YLQSdO65OZjARM8A8ZN62O11";
/**
* Secret Key
*/
String CLIENT_SECRET = "4fU8odNqBDqNk5W3BS0PDrdk20TGV9sO";
/**
* OCR
*/
String ID_CARD_FRONT_URL = "https://aip.baidubce.com/rest/2.0/ocr/v1/idcard";
/**
* OCR
*/
String ID_CARD_MULTI_URL = "https://aip.baidubce.com/rest/2.0/ocr/v1/multi_idcard";
/**
* OCR
*/
String BUSINESS_LICENSE_URL = "https://aip.baidubce.com/rest/2.0/ocr/v1/business_license";
/**
* OCR accessToken
*/
String ACCESS_TOKEN_URL = "https://aip.baidubce.com/oauth/2.0/token";
}

@ -0,0 +1,281 @@
package cn.iocoder.yudao.module.bs.utils;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.baidu.aip.auth.DevAuth;
import com.baidu.aip.util.AipClientConfiguration;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.net.URLEncoder;
import java.util.Map;
/**
* @author MrFang
* @date 20230815 11:36
*/
@Slf4j
public class BaiduOcrHandler {
/**
* OCR
*
* @param idCardFrontImage {@linkplain File}{@linkplain MultipartFile}
* @return IdCard
*/
/* public static IdCard handleFrontIdCard(Object idCardFrontImage) throws Exception {
String accessToken = getAccessToken();
if (StringUtils.isEmpty(accessToken)) {
throw new RuntimeException("获取token失败");
}
String base64ImgParam = getImageBase64Param(idCardFrontImage);
String param = "id_card_side=front&image=" + base64ImgParam;
String jsonStr = HttpUtil.post(BaiduOcrConstant.ID_CARD_FRONT_URL, accessToken, param);
JSONObject jsonObject = JSON.parseObject(jsonStr);
log.info("身份证正面OCR识别结果=>{}", jsonObject);
// 解析身份证正面数据
String status = jsonObject.getString("image_status");
if (!"normal".equals(status)) {
throw new RuntimeException("身份证图片识别失败:"+status );
}
JSONObject wordsResult = jsonObject.getJSONObject("words_result");
if (wordsResult.isEmpty()) {
throw new RuntimeException("没有解析到任何数据");
}
IdCard idCard = new IdCard();
buildIdCard(idCard, wordsResult);
return idCard;
}*/
/**
* OCR
*
* @param idCardFrontImage {@linkplain File}{@linkplain MultipartFile}
* @return IdCard
*/
/* public static IdCard handleMultiIdCard(Object idCardFrontImage) throws Exception {
String accessToken = getAccessToken();
if (StringUtils.isEmpty(accessToken)) {
throw new RuntimeException("获取token失败");
}
String base64ImgParam = getImageBase64Param(idCardFrontImage);
String param = "image=" + base64ImgParam;
String jsonStr = HttpUtil.post(BaiduOcrConstant.ID_CARD_MULTI_URL, accessToken, param);
JSONObject jsonObject = JSON.parseObject(jsonStr);
log.info("身份证正反面OCR识别结果=>{}", jsonObject);
// 解析正反面
JSONArray wordsResult = jsonObject.getJSONArray("words_result");
if (wordsResult.isEmpty()) {
throw new RuntimeException("没有解析到任何数据");
}
IdCard idCard = new IdCard();
for (Object o : wordsResult) {
JSONObject obj = (JSONObject) o;
JSONObject cardResult = obj.getJSONObject("card_result");
if (cardResult == null) {
continue;
}
buildIdCard(idCard, cardResult);
}
return idCard;
}*/
/**
* OCR
*
* @param businessLicenseImage {@linkplain File}{@linkplain MultipartFile}
* @return IdCard
*/
public static BusinessLicense handleBusinessLicense(Object businessLicenseImage) throws Exception {
String accessToken = getAccessToken();
if (StringUtils.isEmpty(accessToken)) {
throw new RuntimeException("获取token失败");
}
String base64ImgParam = getImageBase64Param(businessLicenseImage);
String param = "detect_direction=true&" + "image=" + base64ImgParam;
String jsonStr = HttpUtil.post(BaiduOcrConstant.BUSINESS_LICENSE_URL, accessToken, param);
JSONObject jsonObject = JSON.parseObject(jsonStr);
// log.info("营业执照OCR识别结果=>{}", jsonObject);
if (!jsonObject.containsKey("words_result")) {
throw new RuntimeException("识别营业执照接口返回信息状态失败");
}
BusinessLicense license = new BusinessLicense();
JSONObject wordsResult = jsonObject.getJSONObject("words_result");
buildBusinessLicense(license, wordsResult);
return license;
}
/**
*
*
* @param idCard
* @param meta
*//*
private static void buildIdCard(IdCard idCard, JSONObject meta) {
for (Map.Entry<String, Object> entry : meta.entrySet()) {
String key = entry.getKey();
String resultStr = entry.getValue().toString();
JSONObject result = JSON.parseObject(resultStr);
String value = result.getString("words");
switch (key) {
case IDCardKeywords.FRONT_NAME:
idCard.setName(value);
break;
case IDCardKeywords.FRONT_SEX:
idCard.setSex(value);
break;
case IDCardKeywords.FRONT_NATION:
idCard.setNation(value);
break;
case IDCardKeywords.FRONT_BIRTH:
idCard.setBirthDate(value);
break;
case IDCardKeywords.FRONT_ADDRESS:
idCard.setAddress(value);
break;
case IDCardKeywords.FRONT_ID:
idCard.setCardNumber(value);
break;
case IDCardKeywords.BACK_SIGN_ORG:
idCard.setSignOrg(value);
break;
case IDCardKeywords.BACK_SIGN_DATE:
idCard.setSignDate(value);
break;
case IDCardKeywords.BACK_EXPIRE:
idCard.setExpiredDate(value);
break;
default:
}
}
}
*/
/**
*
*
* @param license
* @param meta
*/
private static void buildBusinessLicense(BusinessLicense license, JSONObject meta) {
for (Map.Entry<String, Object> entry : meta.entrySet()) {
String key = entry.getKey();
String resultStr = entry.getValue().toString();
JSONObject result = JSON.parseObject(resultStr);
String value = result.getString("words");
switch (key) {
case LicenseKeywords.UNIT_NAME:
license.setUnitName(value);
break;
case LicenseKeywords.UNIT_TYPE:
license.setUnitType(value);
break;
case LicenseKeywords.LEGAL_PERSON:
license.setLegalPerson(value);
break;
case LicenseKeywords.ADDRESS:
license.setAddress(value);
break;
case LicenseKeywords.CODE:
license.setCode(value);
break;
case LicenseKeywords.ID_NUMBER:
license.setIdNumber(value);
break;
case LicenseKeywords.VALIDITY:
license.setValidity(value);
break;
case LicenseKeywords.REGISTERED_CAPITAL:
license.setRegisteredCapital(value);
break;
case LicenseKeywords.COMPANY_CREATE_DATE:
license.setCompanyCreateDate(value);
break;
case LicenseKeywords.LAYOUT:
license.setLayout(value);
break;
case LicenseKeywords.BUSINESS_SCOPE:
license.setBusinessScope(value);
break;
default:
}
}
}
/**
*
*
* @param file FileMultipartFile
* @return
* @throws Exception
*/
private static File transferToFile(Object file) throws Exception {
File img;
if (file instanceof File) {
img = (File) file;
} else if (file instanceof MultipartFile) {
img = FileUtils.multipartFileToFile((MultipartFile) file);
} else {
throw new UnsupportedOperationException("仅支持File或MultipartFile类型的参数");
}
return img;
}
/**
* base64
*
* @param file
* @return
* @throws Exception
*/
private static String getImageBase64Param(Object file) throws Exception {
File img;
if (file instanceof File) {
img = (File) file;
} else if (file instanceof MultipartFile) {
img = FileUtils.multipartFileToFile((MultipartFile) file);
} else {
throw new UnsupportedOperationException("仅支持File或MultipartFile类型的参数");
}
byte[] bytes = FileUtil.readFileByBytes(img);
String encode = Base64Util.encode(bytes);
return URLEncoder.encode(encode, BaiduOcrConstant.CHARSET);
}
/**
* APIaccess_token
* grant_type - client_credentials
* client_id - API Key
* client_secret - Secret Key
*
* @return access_token
*/
public static String getAccessToken() {
try {
org.json.JSONObject jsonObject = DevAuth.oauth(BaiduOcrConstant.CLIENT_ID, BaiduOcrConstant.CLIENT_SECRET, config());
return jsonObject.getString("access_token");
} catch (Exception e) {
log.error(e.getMessage(), e);
}
return null;
}
/**
*
*
*/
private static AipClientConfiguration config() {
AipClientConfiguration config = new AipClientConfiguration();
config.setConnectionTimeoutMillis(60000);
config.setSocketTimeoutMillis(60000);
return config;
}
}

@ -0,0 +1,67 @@
package cn.iocoder.yudao.module.bs.utils;
/**
* @author MrFang
* @date 20230815 11:38
*/
public class Base64Util {
private static final char last2byte = (char) Integer.parseInt("00000011", 2);
private static final char last4byte = (char) Integer.parseInt("00001111", 2);
private static final char last6byte = (char) Integer.parseInt("00111111", 2);
private static final char lead6byte = (char) Integer.parseInt("11111100", 2);
private static final char lead4byte = (char) Integer.parseInt("11110000", 2);
private static final char lead2byte = (char) Integer.parseInt("11000000", 2);
private static final char[] encodeTable = new char[]{'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', '+', '/'};
public Base64Util() {
}
public static String encode(byte[] from) {
StringBuilder to = new StringBuilder((int) ((double) from.length * 1.34D) + 3);
int num = 0;
char currentByte = 0;
int i;
for (i = 0; i < from.length; ++i) {
for (num %= 8; num < 8; num += 6) {
switch (num) {
case 0:
currentByte = (char) (from[i] & lead6byte);
currentByte = (char) (currentByte >>> 2);
case 1:
case 3:
case 5:
default:
break;
case 2:
currentByte = (char) (from[i] & last6byte);
break;
case 4:
currentByte = (char) (from[i] & last4byte);
currentByte = (char) (currentByte << 2);
if (i + 1 < from.length) {
currentByte = (char) (currentByte | (from[i + 1] & lead2byte) >>> 6);
}
break;
case 6:
currentByte = (char) (from[i] & last2byte);
currentByte = (char) (currentByte << 4);
if (i + 1 < from.length) {
currentByte = (char) (currentByte | (from[i + 1] & lead4byte) >>> 4);
}
}
to.append(encodeTable[currentByte]);
}
}
if (to.length() % 4 != 0) {
for (i = 4 - to.length() % 4; i > 0; --i) {
to.append("=");
}
}
return to.toString();
}
}

@ -0,0 +1,46 @@
package cn.iocoder.yudao.module.bs.utils;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
/**
* @author MrFang
* @date 20230815 11:37
*/
@Data
public class BusinessLicense {
@Schema(description = "单位名称")
private String unitName;
@Schema(description = "类型")
private String unitType;
@Schema(description = "公司法人")
private String legalPerson;
@Schema(description = "地址")
private String address;
@Schema(description = "有效期")
private String validity;
@Schema(description = "证件编号")
private String idNumber;
@Schema(description = "社会信用代码")
private String code;
@Schema(description = "注册资本")
private String registeredCapital;
@Schema(description = "成立日期")
private String companyCreateDate;
@Schema(description = "组成形式")
private String layout;
@Schema(description = "经营范围")
private String businessScope;
}

@ -0,0 +1,66 @@
package cn.iocoder.yudao.module.bs.utils;
import java.io.*;
/**
* @author MrFang
* @date 20230815 11:39
*/
public class FileUtil {
/**
*
*/
public static String readFileAsString(String filePath) throws IOException {
File file = new File(filePath);
if (!file.exists()) {
throw new FileNotFoundException(filePath);
}
if (file.length() > 1024 * 1024 * 1024) {
throw new IOException("File is too large");
}
StringBuilder sb = new StringBuilder((int) (file.length()));
// 创建字节输入流
FileInputStream fis = new FileInputStream(filePath);
// 创建一个长度为10240的Buffer
byte[] bbuf = new byte[10240];
// 用于保存实际读取的字节数
int hasRead = 0;
while ((hasRead = fis.read(bbuf)) > 0) {
sb.append(new String(bbuf, 0, hasRead));
}
fis.close();
return sb.toString();
}
/**
* byte[]
*/
public static byte[] readFileByBytes(String filePath) throws IOException {
File file = new File(filePath);
if (!file.exists()) {
throw new FileNotFoundException(filePath);
}
return readFileByBytes(file);
}
/**
* byte[]
*/
public static byte[] readFileByBytes(File file) throws IOException {
try (ByteArrayOutputStream bos = new ByteArrayOutputStream((int) file.length())) {
BufferedInputStream in = null;
in = new BufferedInputStream(new FileInputStream(file));
short bufSize = 1024;
byte[] buffer = new byte[bufSize];
int len1;
while (-1 != (len1 = in.read(buffer, 0, bufSize))) {
bos.write(buffer, 0, len1);
}
byte[] var7 = bos.toByteArray();
return var7;
}
}
}

@ -0,0 +1,127 @@
package cn.iocoder.yudao.module.bs.utils;
import org.springframework.core.io.ClassPathResource;
import org.springframework.web.multipart.MultipartFile;
import java.io.*;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.Objects;
/**
* @author MrFang
* @date 20230815 11:38
*/
public class FileUtils {
/**
* classpath
*
* @param fileClassPath classpath
* @return
*/
public static InputStream readFileStreamWithClassPath(String fileClassPath) {
return Thread.currentThread().getContextClassLoader().getResourceAsStream(fileClassPath);
}
/**
* classpath
*
* @param fileClassPath classpath
* @return
*/
public static File readFileWithClassPath(String fileClassPath) throws IOException {
return new ClassPathResource(fileClassPath).getFile();
}
/**
* classpath
*
* @param absoluteFilePath
* @return
*/
public static InputStream readFile(String absoluteFilePath) throws IOException {
return Files.newInputStream(Paths.get(absoluteFilePath));
}
/**
*
*
* @param sourceFilePath
* @param targetFilePath
* @throws IOException
*/
public static void copy(String sourceFilePath, String targetFilePath) throws IOException {
Files.copy(Paths.get(sourceFilePath), Paths.get(targetFilePath));
}
/**
* MultipartFileFile
* <p>
* 使java 使 MultipartFile.transferto()
*
* @param multipartFile
* @return
*/
public static File transferToFile(MultipartFile multipartFile) {
File file = null;
try {
String originalFilename = multipartFile.getOriginalFilename();
String[] filename = originalFilename.split("\\.");
file = File.createTempFile(filename[0], filename[1]);
multipartFile.transferTo(file);
file.deleteOnExit();
} catch (IOException e) {
e.printStackTrace();
}
return file;
}
/**
* MultipartFile File
*
* @param file
* @throws Exception
*/
public static File multipartFileToFile(MultipartFile file) throws Exception {
File toFile = null;
InputStream ins;
ins = file.getInputStream();
toFile = new File(Objects.requireNonNull(file.getOriginalFilename()));
inputStreamToFile(ins, toFile);
ins.close();
return toFile;
}
/**
*
* @param ins
* @param file
*/
private static void inputStreamToFile(InputStream ins, File file) {
try {
OutputStream os = new FileOutputStream(file);
int bytesRead = 0;
byte[] buffer = new byte[8192];
while ((bytesRead = ins.read(buffer, 0, 8192)) != -1) {
os.write(buffer, 0, bytesRead);
}
os.close();
ins.close();
} catch (Exception e) {
e.printStackTrace();
}
}
/**
*
* @param file
*/
public static boolean deleteTempFile(File file) {
if (file != null) {
File del = new File(file.toURI());
return del.delete();
}
return false;
}
}

@ -0,0 +1,77 @@
package cn.iocoder.yudao.module.bs.utils;
import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.List;
import java.util.Map;
/**
* @author MrFang
* @date 20230815 11:39
*/
public class HttpUtil {
public static String post(String requestUrl, String accessToken, String params)
throws Exception {
String contentType = "application/x-www-form-urlencoded";
return HttpUtil.post(requestUrl, accessToken, contentType, params);
}
public static String post(String requestUrl, String accessToken, String contentType, String params)
throws Exception {
String encoding = "UTF-8";
if (requestUrl.contains("nlp")) {
encoding = "GBK";
}
return HttpUtil.post(requestUrl, accessToken, contentType, params, encoding);
}
public static String post(String requestUrl, String accessToken, String contentType, String params, String encoding)
throws Exception {
String url = requestUrl + "?access_token=" + accessToken;
return HttpUtil.postGeneralUrl(url, contentType, params, encoding);
}
public static String postGeneralUrl(String generalUrl, String contentType, String params, String encoding)
throws Exception {
URL url = new URL(generalUrl);
// 打开和URL之间的连接
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
connection.setRequestMethod("POST");
// 设置通用的请求属性
connection.setRequestProperty("Content-Type", contentType);
connection.setRequestProperty("Connection", "Keep-Alive");
connection.setUseCaches(false);
connection.setDoOutput(true);
connection.setDoInput(true);
// 得到请求的输出流对象
DataOutputStream out = new DataOutputStream(connection.getOutputStream());
out.write(params.getBytes(encoding));
out.flush();
out.close();
// 建立实际的连接
connection.connect();
// 获取所有响应头字段
Map<String, List<String>> headers = connection.getHeaderFields();
// 遍历所有的响应头字段
for (String key : headers.keySet()) {
System.err.println(key + "--->" + headers.get(key));
}
// 定义 BufferedReader输入流来读取URL的响应
BufferedReader in = null;
in = new BufferedReader(
new InputStreamReader(connection.getInputStream(), encoding));
String result = "";
String getLine;
while ((getLine = in.readLine()) != null) {
result += getLine;
}
in.close();
System.err.println("result:" + result);
return result;
}
}

@ -0,0 +1,41 @@
package cn.iocoder.yudao.module.bs.utils;
import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;
/**
* @author MrFang
* @date 20230815 11:37
*/
@Data
public class IdCard {
@Schema(description = "姓名")
private String name;
@Schema(description = "性别")
private String sex;
@Schema(description = "民族")
private String nation;
@Schema(description = "出生日期")
private String birthDate;
@Schema(description = "住址")
private String address;
@Schema(description = "公民身份号码")
private String cardNumber;
@Schema(description = "签发机关")
private String signOrg;
@Schema(description = "签发日期")
private String signDate;
@Schema(description = "失效日期")
private String expiredDate;
}

@ -0,0 +1,19 @@
package cn.iocoder.yudao.module.bs.utils;
/**
* @author MrFang
* @date 20230815 14:08
*/
public class LicenseKeywords {
public static final String UNIT_NAME = "单位名称";
public static final String BUSINESS_SCOPE = "经营范围";
public static final String LAYOUT = "组成形式";
public static final String UNIT_TYPE = "类型";
public static final String LEGAL_PERSON = "法定代表人";
public static final String ADDRESS = "地址";
public static final String CODE = "社会信用代码";
public static final String ID_NUMBER = "证件编号";
public static final String VALIDITY = "有效期";
public static final String REGISTERED_CAPITAL = "注册资本";
public static final String COMPANY_CREATE_DATE = "成立日期";
}
Loading…
Cancel
Save