GithubBackup/httprunner
1
0
Fork 0
mirror of https://github.com/httprunner/httprunner.git synced 2026-06-01 22:09:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

change: add ocr response logID

Browse Source
This commit is contained in:
debugtalk
2022-10-16 09:32:56 +08:00
parent cf6785965d
commit 2f73012cfc
4 changed files with 20 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
hrp/pkg/uixt/ocr_vedem.go
View File

@@ -71,7 +71,11 @@ func (s *veDEMOCRService) getOCRResult(imageBuf []byte) ([]OCRResult, error) {
req.Header.Add("Content-Type", bodyWriter.FormDataContentType())
resp, err := client.Do(req)
if err != nil {
return nil, fmt.Errorf("http reqeust OCR server error: %v", err)
var logID string
if resp != nil {
logID = getLogID(resp.Header)
}
return nil, fmt.Errorf("http reqeust veDEM OCR server error: %v, logID: %s", err, logID)
}
defer resp.Body.Close()
@@ -93,6 +97,18 @@ func (s *veDEMOCRService) getOCRResult(imageBuf []byte) ([]OCRResult, error) {
return ocrResult.OCRResult, nil
}
func getLogID(header http.Header) string {
if len(header) == 0 {
return ""
}
logID, ok := header["X-Tt-Logid"]
if !ok || len(logID) == 0 {
return ""
}
return logID[0]
}
func (s *veDEMOCRService) FindText(text string, imageBuf []byte, index ...int) (rect image.Rectangle, err error) {
if len(index) == 0 {
index = []int{0} // index not specified