Initial commit: 知识管理平台
主要功能: - 多用户管理系统 - 浏览器自动化(Playwright) - 任务编排和执行 - Docker容器化部署 - 数据持久化和日志管理 技术栈: - Flask 3.0.0 - Playwright 1.40.0 - SQLite with connection pooling - Docker + Docker Compose 部署说明详见README.md
This commit is contained in:
Yu Yon
214
browser_installer.py
Executable file
214
browser_installer.py
Executable file
@@ -0,0 +1,214 @@
|
||||
#!/usr/bin/env python3
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
浏览器自动下载安装模块
|
||||
检测本地是否有Playwright浏览器，如果没有则自动下载安装
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import shutil
|
||||
import subprocess
|
||||
from pathlib import Path
|
||||
|
||||
# 设置浏览器安装路径（支持Docker和本地环境）
|
||||
# Docker环境: PLAYWRIGHT_BROWSERS_PATH环境变量已设置为 /ms-playwright
|
||||
# 本地环境: 使用Playwright默认路径
|
||||
if 'PLAYWRIGHT_BROWSERS_PATH' in os.environ:
|
||||
BROWSERS_PATH = os.environ['PLAYWRIGHT_BROWSERS_PATH']
|
||||
else:
|
||||
# Windows: %USERPROFILE%\AppData\Local\ms-playwright
|
||||
# Linux: ~/.cache/ms-playwright
|
||||
if sys.platform == 'win32':
|
||||
BROWSERS_PATH = str(Path.home() / "AppData" / "Local" / "ms-playwright")
|
||||
else:
|
||||
BROWSERS_PATH = str(Path.home() / ".cache" / "ms-playwright")
|
||||
os.environ["PLAYWRIGHT_BROWSERS_PATH"] = BROWSERS_PATH
|
||||
|
||||
|
||||
class BrowserInstaller:
|
||||
"""浏览器安装器"""
|
||||
|
||||
def __init__(self, log_callback=None):
|
||||
"""
|
||||
初始化安装器
|
||||
|
||||
Args:
|
||||
log_callback: 日志回调函数
|
||||
"""
|
||||
self.log_callback = log_callback
|
||||
|
||||
def log(self, message):
|
||||
"""输出日志"""
|
||||
if self.log_callback:
|
||||
self.log_callback(message)
|
||||
else:
|
||||
try:
|
||||
print(message)
|
||||
except UnicodeEncodeError:
|
||||
# 如果打印Unicode字符失败，替换特殊字符
|
||||
safe_message = message.replace('✓', '[OK]').replace('✗', '[X]')
|
||||
print(safe_message)
|
||||
|
||||
def check_playwright_installed(self):
|
||||
"""检查Playwright是否已安装"""
|
||||
try:
|
||||
import playwright
|
||||
self.log("✓ Playwright已安装")
|
||||
return True
|
||||
except ImportError:
|
||||
self.log("✗ Playwright未安装")
|
||||
return False
|
||||
|
||||
def check_chromium_installed(self):
|
||||
"""检查Chromium浏览器是否已安装"""
|
||||
try:
|
||||
from playwright.sync_api import sync_playwright
|
||||
|
||||
# 尝试启动浏览器检查是否可用
|
||||
with sync_playwright() as p:
|
||||
try:
|
||||
# 使用超时快速检查
|
||||
browser = p.chromium.launch(headless=True, timeout=5000)
|
||||
browser.close()
|
||||
self.log("✓ Chromium浏览器已安装且可用")
|
||||
return True
|
||||
except Exception as e:
|
||||
error_msg = str(e)
|
||||
self.log(f"✗ Chromium浏览器不可用: {error_msg}")
|
||||
|
||||
# 检查是否是路径不存在的错误
|
||||
if "Executable doesn't exist" in error_msg:
|
||||
self.log("检测到浏览器文件缺失，需要重新安装")
|
||||
|
||||
return False
|
||||
except Exception as e:
|
||||
self.log(f"✗ 检查浏览器时出错: {str(e)}")
|
||||
return False
|
||||
|
||||
def install_chromium(self):
|
||||
"""安装Chromium浏览器"""
|
||||
try:
|
||||
self.log("正在安装 Chromium 浏览器...")
|
||||
|
||||
# 查找 playwright 可执行文件
|
||||
playwright_cli = None
|
||||
possible_paths = [
|
||||
os.path.join(os.path.dirname(sys.executable), "Scripts", "playwright.exe"),
|
||||
os.path.join(os.path.dirname(sys.executable), "playwright.exe"),
|
||||
os.path.join(os.path.dirname(sys.executable), "Scripts", "playwright"),
|
||||
os.path.join(os.path.dirname(sys.executable), "playwright"),
|
||||
"playwright", # 系统PATH中
|
||||
]
|
||||
|
||||
for path in possible_paths:
|
||||
if os.path.exists(path) or shutil.which(path):
|
||||
playwright_cli = path
|
||||
break
|
||||
|
||||
# 如果找到了 playwright CLI，直接调用
|
||||
if playwright_cli:
|
||||
self.log(f"使用 Playwright CLI: {playwright_cli}")
|
||||
result = subprocess.run(
|
||||
[playwright_cli, "install", "chromium"],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=300
|
||||
)
|
||||
else:
|
||||
# 检测是否是 Nuitka 编译的程序
|
||||
is_nuitka = hasattr(sys, 'frozen') or '__compiled__' in globals()
|
||||
|
||||
if is_nuitka:
|
||||
self.log("检测到 Nuitka 编译环境")
|
||||
self.log("✗ 无法找到 playwright CLI 工具")
|
||||
self.log("请手动运行: playwright install chromium")
|
||||
return False
|
||||
else:
|
||||
# 使用 python -m
|
||||
result = subprocess.run(
|
||||
[sys.executable, "-m", "playwright", "install", "chromium"],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=300
|
||||
)
|
||||
|
||||
if result.returncode == 0:
|
||||
self.log("✓ Chromium浏览器安装成功")
|
||||
return True
|
||||
else:
|
||||
self.log(f"✗ 浏览器安装失败: {result.stderr}")
|
||||
return False
|
||||
|
||||
except subprocess.TimeoutExpired:
|
||||
self.log("✗ 浏览器安装超时")
|
||||
return False
|
||||
except Exception as e:
|
||||
self.log(f"✗ 浏览器安装出错: {str(e)}")
|
||||
return False
|
||||
|
||||
def auto_install(self):
|
||||
"""
|
||||
自动检测并安装所需环境
|
||||
|
||||
Returns:
|
||||
是否成功安装或已安装
|
||||
"""
|
||||
self.log("=" * 60)
|
||||
self.log("检查浏览器环境...")
|
||||
self.log("=" * 60)
|
||||
|
||||
# 1. 检查Playwright是否安装
|
||||
if not self.check_playwright_installed():
|
||||
self.log("✗ Playwright未安装，无法继续")
|
||||
self.log("请确保程序包含 Playwright 库")
|
||||
return False
|
||||
|
||||
# 2. 检查Chromium浏览器是否安装
|
||||
if not self.check_chromium_installed():
|
||||
self.log("\n未检测到Chromium浏览器，开始自动安装...")
|
||||
|
||||
# 安装浏览器
|
||||
if not self.install_chromium():
|
||||
self.log("✗ 浏览器安装失败")
|
||||
self.log("\n您可以尝试以下方法：")
|
||||
self.log("1. 手动执行: playwright install chromium")
|
||||
self.log("2. 检查网络连接后重试")
|
||||
self.log("3. 检查防火墙设置")
|
||||
return False
|
||||
|
||||
self.log("\n" + "=" * 60)
|
||||
self.log("✓ 浏览器环境检查完成，一切就绪！")
|
||||
self.log("=" * 60 + "\n")
|
||||
|
||||
return True
|
||||
|
||||
|
||||
def check_and_install_browser(log_callback=None):
|
||||
"""
|
||||
便捷函数：检查并安装浏览器
|
||||
|
||||
Args:
|
||||
log_callback: 日志回调函数
|
||||
|
||||
Returns:
|
||||
是否成功
|
||||
"""
|
||||
installer = BrowserInstaller(log_callback)
|
||||
return installer.auto_install()
|
||||
|
||||
|
||||
# 测试代码
|
||||
if __name__ == "__main__":
|
||||
print("浏览器自动安装工具")
|
||||
print("=" * 60)
|
||||
|
||||
installer = BrowserInstaller()
|
||||
success = installer.auto_install()
|
||||
|
||||
if success:
|
||||
print("\n✓ 安装成功！您现在可以运行主程序了。")
|
||||
else:
|
||||
print("\n✗ 安装失败，请查看上方错误信息。")
|
||||
|
||||
print("=" * 60)
|
||||
Reference in New Issue
Block a user