liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

调整日志级别

Browse Source
This commit is contained in:
liuwei
2025-06-11 16:03:44 +08:00
parent c85cf16368
commit f3f7dcce20
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
utils/sehuatang/shehuatang.py
View File

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