Python虚拟环境中图像文件保存策略：避免FileNotFoundError

import sys
import os

def get_venv_root():
    """获取当前虚拟环境的根路径，如果不在虚拟环境中则抛出错误。"""
    venv_root = sys.prefix
    # sys.base_prefix 是非虚拟环境的Python安装路径
    if venv_root == sys.base_prefix:
        raise RuntimeError("当前Python脚本未在虚拟环境中运行。")
    return venv_root

try:
    venv_path = get_venv_root()
    print(f"虚拟环境根路径: {venv_path}")
except RuntimeError as e:
    print(e)
    # 如果不在venv中，可以考虑使用其他默认路径，或直接退出
    # 这里为了演示，我们假设必须在venv中运行
    sys.exit(1)

# 承接上一步的 venv_path
image_dir_name = "images"
absolute_image_save_path = os.path.join(venv_path, image_dir_name)

print(f"图像保存目录的绝对路径: {absolute_image_save_path}")

# 承接上一步的 absolute_image_save_path
os.makedirs(absolute_image_save_path, exist_ok=True)
print(f"确保目录 '{absolute_image_save_path}' 存在。")

import os
import sys
from Screenshot import Screenshot
from selenium import webdriver

# --- 步骤一：获取虚拟环境根路径并验证 ---
def get_venv_root():
    venv_root = sys.prefix
    if venv_root == sys.base_prefix:
        raise RuntimeError("当前Python脚本未在虚拟环境中运行。")
    return venv_root

try:
    venv_path = get_venv_root()
except RuntimeError as e:
    print(e)
    sys.exit(1)

# --- 步骤二：构建图像保存目录的绝对路径 ---
image_dir_name = "images"
absolute_image_save_path = os.path.join(venv_path, image_dir_name)

# --- 步骤三：创建目录（如果不存在） ---
os.makedirs(absolute_image_save_path, exist_ok=True)
print(f"图像将保存到: {absolute_image_save_path}")

# --- 步骤四：集成到你的Selenium-Screenshot代码中 ---
ob = Screenshot.Screenshot()
driver = webdriver.Chrome() # 确保已安装ChromeDriver并配置到PATH
url = "https://www.google.com"
driver.get(url)
driver.maximize_window()

image_name = 'screenshot.png'
full_image_path = os.path.join(absolute_image_save_path, image_name)

print(f"正在保存截图至: {full_image_path}")

try:
    # 将构建好的绝对路径传递给 save_path 参数
    img_url = ob.full_screenshot(driver, save_path=absolute_image_save_path,
                                  image_name=image_name,
                                  is_load_at_runtime=True,
                                  load_wait_time=3)
    print(f"截图成功保存，返回的URL/路径: {img_url}")
except Exception as e:
    print(f"保存截图时发生错误: {e}")
finally:
    driver.close()
    driver.quit()