eta_api/services/aliyun.go

package services

import (
	"crypto/tls"
	"encoding/base64"
	"encoding/json"
	"errors"
	"fmt"
	openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client"
	ocr20191230 "github.com/alibabacloud-go/ocr-20191230/v3/client"
	util "github.com/alibabacloud-go/tea-utils/v2/service"
	"github.com/alibabacloud-go/tea/tea"
	"hongze/hz_eta_api/utils"
	"io/ioutil"
	"net/http"
	"strings"

	rhttp "github.com/rdlucklib/rdluck_tools/http"
)

const (
	AlAppKey    = "203889624"
	AlAppSecret = "voampGGl0yGNx5xA1sZdmZlV2EiBct2P"
	AlAppCode   = "22553c4ba74545568aba70ac6cfd441d"
)

type OcrItem struct {
	Image string `json:"image"`
}

type OcrResult struct {
	Addr               []string `json:"addr"`
	Company            []string `json:"company"`
	Success            bool     `json:"success"`
	TelCell            []string `json:"tel_cell"`
	Departments        []string `json:"departments"`
	Name               string   `json:"name"`
	OfficePhoneNumbers []string `json:"office_phone_numbers"`
	Titles             []string `json:"titles"`
}

// /ocr_business_card.json
func AliyunBusinessCard(imgUrl string) (result *OcrResult, err error) {
	postUrl := `https://dm-57.data.aliyun.com/rest/160601/ocr/ocr_business_card.json`

	imgBody, err := rhttp.Get(imgUrl)
	if err != nil {
		return
	}
	imageBase64 := base64.StdEncoding.EncodeToString(imgBody)

	item := new(OcrItem)
	item.Image = imageBase64
	jsonBody, err := json.Marshal(item)
	if err != nil {
		fmt.Println("json.Marshal Err:" + err.Error())
		return
	}
	tr := &http.Transport{
		TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
	}
	client := &http.Client{Transport: tr}
	req, err := http.NewRequest("POST", postUrl, strings.NewReader(string(jsonBody)))
	if err != nil {
		fmt.Println("err:" + err.Error())
		return
	}
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("Authorization", "APPCODE "+AlAppCode)
	resp, err := client.Do(req)
	if err != nil {
		fmt.Println("client.Do Err:" + err.Error())
		return
	}
	fmt.Println("resp")
	fmt.Println(resp)
	defer resp.Body.Close()
	body, err := ioutil.ReadAll(resp.Body)
	if err != nil {
		fmt.Println("err:" + err.Error())
		return
	}
	fmt.Println("body:", string(body))
	utils.FileLog.Info(fmt.Sprintf("aliyun card;imgUrl:%s; Body:%s", imgUrl, string(body)))
	err = json.Unmarshal(body, &result)
	return
}

// AliyunBusinessCardOcr 阿里云的ocr识别
func AliyunBusinessCardOcr(imgUrl string) (result OcrResult, err error) {
	if utils.AccessKeyId == `` {
		err = errors.New("阿里云信息未配置")
		return
	}
	config := &openapi.Config{
		// 2、"YOUR_ACCESS_KEY_ID", "YOUR_ACCESS_KEY_SECRET" 的生成请参考https://help.aliyun.com/document_detail/175144.html
		// 如果您是用的子账号AccessKey，还需要为子账号授予权限AliyunVIAPIFullAccess，请参考https://help.aliyun.com/document_detail/145025.html
		// 您的 AccessKey ID
		AccessKeyId: tea.String(utils.AccessKeyId),
		// 您的 AccessKey Secret
		AccessKeySecret: tea.String(utils.AccessKeySecret),
	}
	// 3、访问的域名。注意：这个地方需要求改为相应类目的域名，参考：https://help.aliyun.com/document_detail/143103.html
	config.Endpoint = tea.String("ocr.cn-shanghai.aliyuncs.com")
	// 4、这里只是以ocr为例，其他能力请使用相应类目的包

	client, err := ocr20191230.NewClient(config)
	if err != nil {
		return
	}
	httpClient := http.Client{}
	file, _ := httpClient.Get(imgUrl)

	// 5、这里只是以ocr的RecognizeBankCard为例，其他能力请使用相应类目的包和类
	recognizeBusinessCardAdvanceRequest := &ocr20191230.RecognizeBusinessCardAdvanceRequest{
		ImageURLObject: file.Body,
	}
	runtime := &util.RuntimeOptions{}
	// 6、这里只是以ocr的RecognizeBankCard为例，其他能力请使用相应类目的包和类，注意RecognizeBankCardWithOptions也需要改成对应能力的方法名。方法名是根据能力名称按照一定规范形成的，如能力名称为SegmentCommonImage，对应方法名应该为SegmentCommonImageWithOptions。
	recognizeBusinessCardResponse, err := client.RecognizeBusinessCardAdvance(recognizeBusinessCardAdvanceRequest, runtime)

	if err != nil {
		// 获取整体报错信息
		//fmt.Println(err.Error())
		utils.FileLog.Info(fmt.Sprintf("aliyun card;imgUrl:%s; err:%s", imgUrl, err))
		return
	}

	// 获取整体结果
	//fmt.Println(recognizeBusinessCardResponse)
	//// 获取单个字段，这里只是一个例子，具体能力下的字段需要看具体能力的文档
	//fmt.Println(recognizeBusinessCardResponse.Body.Data.Name)

	// 地址
	for _, v := range recognizeBusinessCardResponse.Body.Data.Addresses {
		result.Addr = append(result.Addr, *v)
	}
	// company
	for _, v := range recognizeBusinessCardResponse.Body.Data.Companies {
		result.Company = append(result.Company, *v)
	}
	// tel_cell
	for _, v := range recognizeBusinessCardResponse.Body.Data.CellPhoneNumbers {
		result.TelCell = append(result.TelCell, *v)
	}
	// Departments
	for _, v := range recognizeBusinessCardResponse.Body.Data.Departments {
		result.Departments = append(result.Departments, *v)
	}
	// OfficePhoneNumbers
	for _, v := range recognizeBusinessCardResponse.Body.Data.OfficePhoneNumbers {
		result.OfficePhoneNumbers = append(result.OfficePhoneNumbers, *v)
	}
	// Titles
	for _, v := range recognizeBusinessCardResponse.Body.Data.Titles {
		result.Titles = append(result.Titles, *v)
	}
	if recognizeBusinessCardResponse.Body.Data.Name != nil {
		result.Name = *recognizeBusinessCardResponse.Body.Data.Name
	}
	result.Success = true

	return
}