feat: add screen resolution in screen results

This commit is contained in:
lilong.129
2023-06-01 17:42:40 +08:00
parent 233d5e1b64
commit a24e1ff251
2 changed files with 10 additions and 1 deletions

View File

@@ -210,6 +210,11 @@ func (dExt *DriverExt) GetStepCacheData() map[string]interface{} {
cacheData["screenshots"] = dExt.cacheStepData.screenShots
cacheData["screenshots_urls"] = dExt.cacheStepData.screenShotsUrls
screenSize, err := dExt.Driver.WindowSize()
if err != nil {
log.Warn().Err(err).Msg("get screen resolution failed")
screenSize = Size{}
}
screenResults := make(map[string]interface{})
for imagePath, screenResult := range dExt.cacheStepData.screenResults {
o, _ := json.Marshal(screenResult.Texts)
@@ -217,6 +222,10 @@ func (dExt *DriverExt) GetStepCacheData() map[string]interface{} {
"tags": screenResult.Tags,
"texts": string(o),
"popularity": screenResult.Popularity,
"resolution": map[string]int{
"width": screenSize.Width,
"height": screenSize.Height,
},
}
screenResults[imagePath] = data