update

toolbox/douyin_selenium/tasks/bless_bag_collector.py

@@ -70,7 +70,8 @@ class BlessBagCollector(object):
         return self.get_element_by_xpath(xpath)
 
     def get_live_ended_element(self):
-        xpath = "//div[@id='TipsLayout']/div[3]/span"
+        # xpath = "//div[@id='TipsLayout']/div[3]/span"
+        xpath = "//div[@id='TipsLayout']/div[3]/span/div/div"
         return self.get_element_by_xpath(xpath)
 
     def get_bless_bag_element(self):
@@ -182,10 +183,10 @@ class BlessBagCollector(object):
                     logger.info(f"{self.flag}达到观看直播时长设置: {self.watch_duration//60}分钟,{self.watch_duration%60}秒.")
                     break
 
-                # live_ended_element = self.get_live_ended_element()
-                # if live_ended_element is not None and live_ended_element.text == "直播已结束":
-                #     logger.info(f"{self.flag}直播已结束: {live_ended_element.text}.")
-                #     break
+                live_ended_element = self.get_live_ended_element()
+                if live_ended_element is not None and live_ended_element.text in ("直播已结束", "直播已结束，猜你喜欢"):
+                    logger.info(f"{self.flag}直播已结束: {live_ended_element.text}.")
+                    break
 
                 online_user_element = self.get_online_user_element()
                 if online_user_element is not None: