调整日志级别

This commit is contained in:
liuwei
2025-06-11 16:03:44 +08:00
parent c85cf16368
commit f3f7dcce20

View File

@@ -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