- 添加重试

- 提示消息全部变为中文

Author: doveppp

main.py

@@ -1,12 +1,32 @@
 import os
 import random
 import time
+import functools
 
 from loguru import logger
 from playwright.sync_api import sync_playwright
 from tabulate import tabulate
 
 
+def retry_decorator(retries=3):
+    def decorator(func):
+        @functools.wraps(func)
+        def wrapper(*args, **kwargs):
+            for attempt in range(retries):
+                try:
+                    return func(*args, **kwargs)
+                except Exception as e:
+                    if attempt == retries - 1:  # 最后一次尝试
+                        logger.error(f"函数 {func.__name__} 最终执行失败: {str(e)}")
+                    logger.warning(f"函数 {func.__name__} 第 {attempt + 1}/{retries} 次尝试失败: {str(e)}")
+                    time.sleep(1)
+            return None
+
+        return wrapper
+
+    return decorator
+
+
 os.environ.pop("DISPLAY", None)
 os.environ.pop("DYLD_LIBRARY_PATH", None)
 
@@ -25,7 +45,7 @@ def __init__(self) -> None:
         self.page.goto(HOME_URL)
 
     def login(self):
-        logger.info("Login")
+        logger.info("开始登录")
         self.page.click(".login-button .d-button-label")
         time.sleep(2)
         self.page.fill("#login-account-name", USERNAME)
@@ -36,36 +56,39 @@ def login(self):
         time.sleep(10)
         user_ele = self.page.query_selector("#current-user")
         if not user_ele:
-            logger.error("Login failed")
+            logger.error("登录失败")
             return False
         else:
-            logger.info("Login success")
+            logger.info("登录成功")
             return True
 
     def click_topic(self):
         topic_list = self.page.query_selector_all("#list-area .title")
-        logger.info(f"Click {len(topic_list)} topics")
+        logger.info(f"发现 {len(topic_list)} 个主题帖")
         for topic in topic_list:
-            logger.info("Click topic: " + topic.get_attribute("href"))
-            page = self.context.new_page()
-            page.goto(HOME_URL + topic.get_attribute("href"))
-            if random.random() < 0.3:  # 0.3 * 30 = 9
-                self.click_like(page)
-            self.browse_post(page)
-            page.close()
+            self.click_one_topic(topic.get_attribute("href"))
+
+    @retry_decorator()
+    def click_one_topic(self, topic_url):
+        page = self.context.new_page()
+        page.goto(HOME_URL + topic_url)
+        if random.random() < 0.3:  # 0.3 * 30 = 9
+            self.click_like(page)
+        self.browse_post(page)
+        page.close()
 
     def browse_post(self, page):
         prev_url = None
         # 开始自动滚动，最多滚动10次
         for _ in range(10):
             # 随机滚动一段距离
             scroll_distance = random.randint(550, 650)  # 随机滚动 550-650 像素
-            logger.info(f"Scrolling down by {scroll_distance} pixels...")
+            logger.info(f"向下滚动 {scroll_distance} 像素...")
             page.evaluate(f"window.scrollBy(0, {scroll_distance})")
-            logger.info(f"Loaded: {page.url}")
+            logger.info(f"已加载页面: {page.url}")
 
             if random.random() < 0.03:  # 33 * 4 = 132
-                logger.success("Randomly exit")
+                logger.success("随机退出浏览")
                 break
 
             # 检查是否到达页面底部
@@ -74,12 +97,12 @@ def browse_post(self, page):
             if current_url != prev_url:
                 prev_url = current_url
             elif at_bottom and prev_url == current_url:
-                logger.success("Reached the bottom of the page. Exiting.")
+                logger.success("已到达页面底部，退出浏览")
                 break
 
             # 动态随机等待
             wait_time = random.uniform(2, 4)  # 随机等待 2-4 秒
-            logger.info(f"Waiting for {wait_time:.2f} seconds...")
+            logger.info(f"等待 {wait_time:.2f} 秒...")
             time.sleep(wait_time)
 
     def run(self):
@@ -103,7 +126,7 @@ def click_like(self, page):
             logger.error(f"点赞失败: {str(e)}")
 
     def print_connect_info(self):
-        logger.info("Print connect info")
+        logger.info("获取连接信息")
         page = self.context.new_page()
         page.goto("https://connect.linux.do/")
         rows = page.query_selector_all("table tr")