diff --git a/utils/sehuatang/shehuatang.py b/utils/sehuatang/shehuatang.py index f76c426..67cc577 100644 --- a/utils/sehuatang/shehuatang.py +++ b/utils/sehuatang/shehuatang.py @@ -38,7 +38,7 @@ def download_image(url): image = BytesIO(response.content) return image except requests.exceptions.RequestException as e: - logger.info(f"下载图片失败: {e}") + logger.warning(f"下载图片失败: {e}") return None @@ -68,16 +68,16 @@ def fetch_and_create_pdf(url): driver = webdriver.Chrome(service=Service(chrome_driver_path), options=options) except Exception as e: - logger.info(f"初始化ChromeDriver失败: {e}") + logger.debug(f"初始化ChromeDriver失败: {e}") chrome_driver_path = ChromeDriverManager().install() driver = webdriver.Chrome(service=Service(chrome_driver_path), options=options) # 如果本地没有chromedriver.exe,则使用webdriver_manager下载一次 if not os.path.exists(chrome_driver_path): chrome_driver_path = ChromeDriverManager().install() - logger.info(f"ChromeDriver已下载到: {chrome_driver_path}") + logger.debug(f"ChromeDriver已下载到: {chrome_driver_path}") else: - logger.info(f"使用本地ChromeDriver: {chrome_driver_path}") + logger.debug(f"使用本地ChromeDriver: {chrome_driver_path}") driver = webdriver.Chrome(service=Service(chrome_driver_path), options=options) @@ -89,10 +89,10 @@ def fetch_and_create_pdf(url): try: enter_button = driver.find_element(By.XPATH, '//a[contains(text(), "满18岁,请点此进入")]') enter_button.click() - logger.info("点击了满18岁按钮") + logger.debug("点击了满18岁按钮") time.sleep(5) except Exception as e: - logger.info("未找到满18岁按钮,跳过此步骤", e) + logger.warning("未找到满18岁按钮,跳过此步骤", e) # 解析页面 html = driver.page_source @@ -203,9 +203,9 @@ def fetch_and_create_pdf(url): # 添加图片到内容中,使用计算后的尺寸 content.append(Image(img_stream, width=new_width, height=new_height)) content.append(Spacer(1, 4)) - logger.info(f"处理图片: 原始尺寸 {img_width}x{img_height}, 新尺寸 {new_width}x{new_height}") + logger.debug(f"处理图片: 原始尺寸 {img_width}x{img_height}, 新尺寸 {new_width}x{new_height}") except Exception as e: - logger.info(f"处理图片时出错: {e}") + logger.error(f"处理图片时出错: {e}") # 在每个帖子后添加分页符(除了最后一页) if post != today_posts[-1]: @@ -223,12 +223,12 @@ def fetch_and_create_pdf(url): return absolute_pdf_path except Exception as e: - logger.info(f"生成PDF时出错: {e}") + logger.error(f"生成PDF时出错: {e}") driver.quit() # 如果生成失败,返回一个默认路径或空字符串 return "" except Exception as e: - logger.info(f"抓取帖子时出错: {e}") + logger.error(f"抓取帖子时出错: {e}") # 如果抓取失败,返回一个默认路径或空字符串 driver.quit() return "" @@ -246,9 +246,9 @@ def add_pdf_encryption(pdf_file, password="4000"): pdf_writer.encrypt(password) with open(pdf_file, "wb") as output_pdf: pdf_writer.write(output_pdf) - logger.info(f"PDF加密成功,密码为: {password}") + logger.debug(f"PDF加密成功,密码为: {password}") except Exception as e: - logger.info(f"PDF加密失败: {e}") + logger.error(f"PDF加密失败: {e}") def pdf_file_path(): @@ -264,7 +264,7 @@ def pdf_file_path(): logger.info(f"PDF生成失败,返回默认路径: {default_path}") return default_path except Exception as e: - logger.info(f"生成PDF路径时出错: {e}") + logger.error(f"生成PDF路径时出错: {e}") # 返回一个默认路径 default_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), "default.pdf") return default_path