调整日志级别
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user