package foundation.ai;
|
|
import java.io.File;
|
import java.io.FileInputStream;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.util.List;
|
import java.util.Objects;
|
|
import org.apache.logging.log4j.LogManager;
|
import org.apache.logging.log4j.Logger;
|
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.fastjson.JSONObject;
|
|
import foundation.ai.dao.InvoiceVerificationDao;
|
import foundation.ai.logic.OCRResultKingDee;
|
import foundation.dao.Filter;
|
import foundation.dao.Settings;
|
import foundation.data.entity.EntitySet;
|
import foundation.data.object.DataObject;
|
import foundation.util.ContentBuilder;
|
import foundation.util.MD5Utils;
|
import foundation.util.Util;
|
import okhttp3.MediaType;
|
import okhttp3.OkHttpClient;
|
import okhttp3.Request;
|
import okhttp3.RequestBody;
|
import okhttp3.Response;
|
import okhttp3.ResponseBody;
|
import sun.misc.BASE64Encoder;
|
|
public class KingdeeAIProvider extends AIProvider {
|
|
private static Logger logger;
|
|
private static String HOST = "https://api.piaozone.com";
|
private static String CLIENT_ID = "Pa0umVUNzg6EiWSEpGht";
|
private static String CLIENT_SECRET = "a3cbabd08e684413a30aae1803c773d5";
|
private static String TOKEN_PATH = "/base/oauth/token";
|
protected static String AES_KEY = "goopyi5VTKD5WIgs";
|
|
public static int TimeOut_Second_Write = 10;
|
public static int TimeOut_Second_Read = 10;
|
public static boolean Debug = true;
|
private static OkHttpClient httpClient = null;
|
|
static {
|
logger = LogManager.getLogger(KingdeeAIProvider.class);
|
}
|
|
@Override
|
protected String getName() {
|
return "金蝶AI";
|
}
|
|
@Override
|
public OCRResult executeAction(Operator operator, File file) throws Exception {
|
boolean isReal = false, isBill = false;
|
|
// 1. 得到图片的Base64
|
String base64 = imageToBase64(file);
|
if (base64 == null) {
|
logger.error("无法对图片{}进行Base64编码，无法进行AI识别", file);
|
return null;
|
}
|
|
// 2. 获取access_token
|
Response tokenResponse = getAccessToken();
|
if (!tokenResponse.isSuccessful()) {
|
return new OCRResult(operator, "error");
|
}
|
ResponseBody body = tokenResponse.body();
|
String bodyStr = body.string();
|
JSONObject tokenJson = JSONObject.parseObject(bodyStr);
|
body.close();
|
|
// 3. 创建请求
|
OperateUrl operation = OperateUrl.parse(operator);
|
Request request = new Request.Builder().url(HOST + operation.requestUrl + tokenJson.getString("access_token"))
|
.post(RequestBody.create(MediaType.parse("text/plain"), base64)).build();
|
|
// 4. 发送请求
|
String value = "";
|
try {
|
value = send(operation, request);
|
} catch (Exception e) {
|
e.printStackTrace();
|
value = "fail";
|
}
|
|
// 5.分析校验结果
|
OCRResultKingDee resultKingDee = OCRResultKingDee.parseBatch(value);
|
List<InvoiceVerificationDao> invoiceVerificationList = resultKingDee.getInvoiceVerificationList();
|
|
if(invoiceVerificationList.size()> 0 || !resultKingDee.isSuccess()) {
|
isReal = false; isBill = false;
|
}
|
|
// 5.1 发票唯一性校验
|
boolean checkDistinct = Settings.getBoolean("InvoiceDistinctCheck", false);
|
|
if (checkDistinct) {
|
for (InvoiceVerificationDao invoice : invoiceVerificationList) {
|
boolean distincCheck = distinctCheck(invoice.getCode(), invoice.getNumber());
|
invoice.setDistinct(distincCheck);
|
}
|
}
|
|
// 6. 返回结果
|
value = Util.ocrJson(value);
|
OCRResult result = new OCRResult(operator, value);
|
result.setErrcode(resultKingDee.getErrcode());
|
result.setDescription(resultKingDee.getDescription());
|
result.setTraceId(resultKingDee.getTraceId());
|
result.setIs_real(isReal);
|
result.setIs_bill(isBill);
|
result.setJsonArray(resultKingDee.getData());
|
result.setInvoiceList(invoiceVerificationList);
|
|
return result;
|
}
|
|
|
@Override
|
public OCRResult executeAction(Operator operator, Object paramObject) throws Exception {
|
boolean isReal = false, isBill = false;
|
|
// 1. 获取发票信息
|
JSONObject vatInvoiceVerify = (JSONObject) paramObject;
|
if (vatInvoiceVerify == null || Util.isEmpty(vatInvoiceVerify)) {
|
logger.error("未获取到相关发票信息，无法查询发票");
|
return null;
|
}
|
|
// 2.获取access_token
|
Response tokenResponse = getAccessToken();
|
if (!tokenResponse.isSuccessful()) {
|
return new OCRResult(operator, "error");
|
}
|
ResponseBody body = tokenResponse.body();
|
String bodyStr = body.string();
|
JSONObject tokenJson = JSONObject.parseObject(bodyStr);
|
body.close();
|
|
// 3. 创建请求
|
String value = "";
|
OperateUrl operation = OperateUrl.parse(operator);
|
JSONObject jsonBody = new JSONObject();
|
jsonBody.put("invoiceCode", vatInvoiceVerify.getString("invoiceCode"));
|
jsonBody.put("invoiceNo", vatInvoiceVerify.getString("invoiceNo"));
|
jsonBody.put("invoiceDate", vatInvoiceVerify.getString("invoiceDate"));
|
jsonBody.put("invoiceMoney", vatInvoiceVerify.getString("invoiceMoney"));
|
jsonBody.put("totalAmount", vatInvoiceVerify.getString("totalAmount"));
|
jsonBody.put("checkCode", vatInvoiceVerify.getString("checkCode"));
|
jsonBody.put("isCreateUrl", vatInvoiceVerify.getString("isCreateUrl"));
|
Request request = new Request.Builder().url(HOST + operation.requestUrl + tokenJson.getString("access_token"))
|
.post(RequestBody.create(MediaType.parse("application/json"), jsonBody.toJSONString())).build();
|
|
// 4. 发送请求
|
try {
|
value = send(operation, request);
|
} catch (Exception e) {
|
e.printStackTrace();
|
value = e.getMessage();
|
}
|
|
// 5. 分析查询结果
|
OCRResultKingDee resultKingDee = OCRResultKingDee.parseOne(value);
|
List<InvoiceVerificationDao> invoiceVerificationList = resultKingDee.getInvoiceVerificationList();
|
|
// 5.1 发票唯一性校验
|
boolean checkDistinct = Settings.getBoolean("InvoiceDistinctCheck", false);
|
|
|
if (checkDistinct) {
|
for (InvoiceVerificationDao invoice : invoiceVerificationList) {
|
boolean distincCheck = distinctCheck(invoice.getCode(), invoice.getNumber());
|
invoice.setDistinct(distincCheck);
|
}
|
}
|
|
if (Util.isEmpty(invoiceVerificationList) || !resultKingDee.isSuccess()) {
|
isReal = false;
|
isBill = false;
|
}
|
|
|
// 6. 返回结果
|
value = Util.ocrJson(value);
|
OCRResult result = new OCRResult(operator, value);
|
result.setErrcode(resultKingDee.getErrcode());
|
result.setDescription(resultKingDee.getDescription());
|
result.setTraceId(resultKingDee.getTraceId());
|
result.setIs_real(isReal);
|
result.setIs_bill(isBill);
|
result.setJsonArray(resultKingDee.getData());
|
result.setInvoiceList(invoiceVerificationList);
|
|
return result;
|
|
}
|
|
private String send(OperateUrl operation, Request request) throws IOException {
|
httpClient = new OkHttpClient();
|
String json = null;
|
|
Response response = httpClient.newCall(request).execute();
|
if (response.isSuccessful()) {
|
try {
|
json = response.body().string();
|
} catch (Exception e) {
|
e.printStackTrace();
|
} finally {
|
response.close();
|
}
|
}
|
|
return json;
|
}
|
|
private static Response getAccessToken() throws Exception {
|
httpClient = new OkHttpClient();
|
long currentTimeStamp = System.currentTimeMillis();
|
String sign = MD5Utils.MD5Encode(CLIENT_ID + CLIENT_SECRET + currentTimeStamp, "UTF-8");
|
|
JSONObject body = new JSONObject();
|
// 1. set parameter
|
body.put("client_id", CLIENT_ID);
|
body.put("client_secret", CLIENT_SECRET);
|
body.put("sign", sign);
|
body.put("timestamp", currentTimeStamp);
|
|
MediaType jsonMediaType = MediaType.get("application/json");
|
String bodyStr = JSON.toJSONString(body);
|
|
Request request = new Request.Builder().url(HOST + TOKEN_PATH).post(RequestBody.create(jsonMediaType, bodyStr))
|
.build();
|
|
// 3. do sendRequest
|
Response response = httpClient.newCall(request).execute();
|
|
return response;
|
}
|
|
private static String imageToBase64(File file) {
|
byte[] data = null;
|
|
InputStream in = null;
|
try {
|
// 1. bytes
|
in = new FileInputStream(file);
|
data = new byte[in.available()];
|
in.read(data);
|
in.close();
|
|
// 2. encode
|
BASE64Encoder encoder = new BASE64Encoder();
|
return encoder.encode(Objects.requireNonNull(data));
|
} catch (IOException e) {
|
e.printStackTrace();
|
} finally {
|
try {
|
in.close();
|
} catch (IOException e) {
|
e.printStackTrace();
|
}
|
}
|
|
return null;
|
}
|
|
public static boolean distinctCheck(String invoiceCode , String invoiceNumber) throws Exception {
|
DataObject dataObject = DataObject.getInstance("so_implant_invoice_list");
|
Filter filter = new Filter();
|
|
ContentBuilder builder = new ContentBuilder(", ");
|
builder.append(Util.quotedStr("input"));
|
builder.append(Util.quotedStr("cancel"));
|
builder.append(Util.quotedStr("close"));
|
|
filter.add("so_implant_invoice.code", invoiceCode);
|
filter.add("so_implant_invoice.number", invoiceNumber);
|
filter.add("so_implant.state_code", "not in ", "(" + builder.toString() + ")");
|
EntitySet entitySet = dataObject.getBrowseEntitySet(filter);
|
|
if (entitySet.size() > 0 ) {
|
return false;
|
}
|
return true;
|
}
|
}
|
