change: set found live/feed msg as const

This commit is contained in:
lilong.129
2023-09-20 21:03:09 +08:00
parent 6464dacdfb
commit 517f128b29
+14 -12
View File
@@ -44,7 +44,6 @@ type VideoCrawler struct {
// used to help checking if swipe success // used to help checking if swipe success
failedCount int64 failedCount int64
lastVideo *Video
FeedCount int `json:"feed_count"` FeedCount int `json:"feed_count"`
FeedStat map[string]int `json:"feed_stat"` // 分类统计 feed 数量:视频/图文/广告/特效/模板/购物 FeedStat map[string]int `json:"feed_stat"` // 分类统计 feed 数量:视频/图文/广告/特效/模板/购物
@@ -117,6 +116,11 @@ func (vc *VideoCrawler) exitLiveRoom() error {
return vc.driverExt.Driver.PressBack() return vc.driverExt.Driver.PressBack()
} }
const (
FOUND_FEED_SUCCESS = "found feed success"
FOUND_LIVE_SUCCESS = "found live success"
)
func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) { func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
if dExt.plugin == nil { if dExt.plugin == nil {
return errors.New("miss plugin for video crawler") return errors.New("miss plugin for video crawler")
@@ -135,12 +139,10 @@ func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
configs: configs, configs: configs,
failedCount: 0, failedCount: 0,
lastVideo: &Video{}, FeedCount: 0,
FeedStat: make(map[string]int),
FeedCount: 0, LiveCount: 0,
FeedStat: make(map[string]int), LiveStat: make(map[string]int),
LiveCount: 0,
LiveStat: make(map[string]int),
} }
defer func() { defer func() {
dExt.cacheStepData.videoCrawler = crawler dExt.cacheStepData.videoCrawler = crawler
@@ -187,7 +189,6 @@ func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
crawler.failedCount++ crawler.failedCount++
continue continue
} }
crawler.lastVideo = feedVideo
screenResult := &ScreenResult{ screenResult := &ScreenResult{
Resolution: dExt.windowSize, Resolution: dExt.windowSize,
@@ -211,7 +212,7 @@ func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
log.Info(). log.Info().
Strs("tags", screenResult.Tags). Strs("tags", screenResult.Tags).
Interface("video", feedVideo). Interface("video", feedVideo).
Msg("found live success") Msg(FOUND_LIVE_SUCCESS)
// simulation watch feed video // simulation watch feed video
sleepStrict(swipeFinishTime, feedVideo.PlayDuration) sleepStrict(swipeFinishTime, feedVideo.PlayDuration)
break break
@@ -224,7 +225,8 @@ func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
} }
log.Info().Msg("tap screen center to enter live room") log.Info().Msg("tap screen center to enter live room")
if err := crawler.driverExt.TapAbsXY(entryPoint.X, entryPoint.Y); err != nil { if err := crawler.driverExt.TapAbsXY(entryPoint.X, entryPoint.Y,
WithOffsetRandomRange(-20, 20)); err != nil {
log.Error().Err(err).Msg("tap live video failed") log.Error().Err(err).Msg("tap live video failed")
continue continue
} }
@@ -236,7 +238,7 @@ func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
log.Info(). log.Info().
Strs("tags", screenResult.Tags). Strs("tags", screenResult.Tags).
Interface("video", feedVideo). Interface("video", feedVideo).
Msg("found live success") Msg(FOUND_LIVE_SUCCESS)
// take screenshot and get screen texts by OCR // take screenshot and get screen texts by OCR
screenResultFromOCR, err := crawler.driverExt.GetScreenResult( screenResultFromOCR, err := crawler.driverExt.GetScreenResult(
@@ -291,7 +293,7 @@ func (dExt *DriverExt) VideoCrawler(configs *VideoCrawlerConfigs) (err error) {
log.Info(). log.Info().
Strs("tags", screenResult.Tags). Strs("tags", screenResult.Tags).
Interface("video", feedVideo). Interface("video", feedVideo).
Msg("found feed success") Msg(FOUND_FEED_SUCCESS)
// simulation watch feed video // simulation watch feed video
sleepStrict(swipeFinishTime, screenResult.Video.PlayDuration) sleepStrict(swipeFinishTime, screenResult.Video.PlayDuration)