1 前言
之前使用Pyecharts库在Windows系统中生成图表并转换为PNG格式图片(传送门),现将代码放于Linux服务器上运行,结果发现错误,生成html文件之后无法保存图片。
2 原理
基于Selenium库的保存方案,其原理是使用Chrome浏览器打开html文件,然后在进行截图
3 实现
3.1 环境准备
安装Chrome和Chrome Driver
1 安装 Chrome
yum install https://dl.google.com/linux/direct/google-chrome-stable_current_x86_64.rpm
chromedriver 下载地址:
https://googlechromelabs.github.io/chrome-for-testing/ (推荐,包含最新稳定版)
这里需要下载和Chrome版本一致的Driver
# 下载 chromedriver
# 进入下载目录
cd /tmp# 查看chrome 版本号
google-chrome --version# 在chromedriver下载地址中找到对应版本,下载对应版本chromedriver
# 如
wget https://edgedl.me.gvt1.com/edgedl/chrome/chrome-for-testing/118.0.5993.70/linux64/chromedriver-linux64.zip# 解压
unzip chromedriver-linux64.zip # 赋权
cd chromedriver-linux64
mv chromedriver /usr/local/bin
cd /usr/local/bin
chmod +x chromedriver
3.2 代码
import os
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Optionsdef split_filepath(filepath):"""此函数用于从文件路径中提取文件名(不含扩展名)。参数:filepath (str): 完整的文件路径返回:str: 文件名(不包含扩展名)"""# 获取目录路径和完整文件名(含扩展名)filename_with_ext = os.path.basename(filepath)# 分离文件名和扩展名filename, ext = os.path.splitext(filename_with_ext)return filenamedef save_as_png(html_file, img_file):"""该函数的作用是将指定的 HTML 文件在浏览器中打开并截图保存为 PNG 图片。参数:html_file (str): 要截图的 HTML 文件的完整路径img_file (str): 保存截图的 PNG 文件的完整路径返回:None"""# chrome-driver 安装路径DRIVER_PATH = "/usr/local/bin"# 浏览器基础配置options = Options()options.add_argument('--no-sandbox')options.add_argument('--headless') # 无头参数options.add_argument('--disable-gpu') # 禁用gpu 防止占用资源出现bugoptions.add_argument('window-size=1920x800') # 设置分窗口辨率options.add_argument('--start-maximized') # 最大化运行(全屏窗口),不设置,取元素可能会报错options.add_argument('--hide-scrollbars')s = Service(DRIVER_PATH)# 启动浏览器driver = webdriver.Chrome(service=s, options=options)# 设置隐式等待时间,在查找元素时最多等待 10 秒driver.implicitly_wait(10)try:# 调用 split_filepath 函数获取 HTML 文件的文件名(不含扩展名)img_name = split_filepath(html_file)# 打印开始截图的提示信息print("开始截图")# 在浏览器中访问指定的 HTML 文件driver.get(f"file://{html_file}")# 对当前页面进行截图并保存为指定的 PNG 文件driver.get_screenshot_as_file(img_file)# 打印完成截图的提示信息print("完成截图")except Exception as e:# 若截图过程中出现异常,打印异常信息print(e)finally:# 关闭当前浏览器窗口driver.close()# 退出浏览器驱动程序,释放资源driver.quit()
这里的重点是driver.get(f"file://{html_file}"),在需要打开的html文件路径前加上file://,否则driver会默认是url,导致超时错误。
3.3 字体
如果在html文件中存在中文的,需要保证Linux安装有中文字体
#查看已安装的字体
#查看已安装的字体
fc-list
#查看已安装的中文字体
fc-list :lang=zh#安装Linux字体程序
yum -y install fontconfig#安装中文字体
#查看服务器是否有:/usr/share/fonts这个目录,如果有,那么看下一步,如果没有,就创建
mkdir -p /usr/share/fonts#引入字体文件
#将准备好的字体放到/usr/share/fonts这个目录#安装字体索引
yum install mkfontscale#生成字体索引,在/usr/share/fonts这个目录执行下面的命令
mkfontscale#第三步验证安装结果
fc-list :lang=zh
4 参考链接
linux下安装 Chrome 和 chromedriver 以及 selenium webdriver 使用
如何给Linux系统安装中文字体
python中使用selenium打开保存在本地的html页面