Dockerfile 部署

This commit is contained in:
hjq
2026-06-12 17:29:48 +08:00
parent 200c7cf9ae
commit 145f31d4c1
2 changed files with 11 additions and 14 deletions

View File

@@ -5,7 +5,6 @@ ERP 登录模块 - DrissionPage
import os
import time
import datetime
import tempfile
from pathlib import Path
from dotenv import load_dotenv
from DrissionPage import ChromiumPage, ChromiumOptions
@@ -28,15 +27,14 @@ def is_docker_env() -> bool:
return os.path.exists("/.dockerenv")
def get_docker_user_data_dir() -> Path:
def get_docker_tmp_root() -> Path:
"""
为每个进程分配独立的 Chromium 用户目录。
多 worker 或异常重启后复用同一 profile容易导致 DevTools 会话错乱
指定 DrissionPage 在 Docker 中的临时根目录。
auto_port() 会在该目录下自动创建独立端口和用户目录
"""
base_dir = Path(os.getenv("DRISSION_USER_DATA_ROOT", tempfile.gettempdir())) / "drission_profiles"
user_data_dir = base_dir / f"worker_{os.getpid()}"
user_data_dir.mkdir(parents=True, exist_ok=True)
return user_data_dir
tmp_root = Path(os.getenv("DRISSION_TMP_ROOT", "/tmp")) / "DrissionPage"
tmp_root.mkdir(parents=True, exist_ok=True)
return tmp_root
# ── 日志 ──────────────────────────────────────────────────────────────────────
@@ -83,12 +81,11 @@ def get_page(headless: bool = False, port: int = 9222) -> ChromiumPage:
co.set_argument("--window-size=1440,900")
if is_docker:
# Docker 生产环境:每次启动分配随机端口,避免前一个僵尸进程占用导致 404
# Docker 生产环境:由 DrissionPage 自动分配独立端口和 profile避免僵尸会话导致 404
tmp_root = get_docker_tmp_root()
co.set_tmp_path(str(tmp_root))
co.auto_port(True)
# 为当前 worker 隔离 Chromium profile防止多个进程共享 profile 导致握手错乱
user_data_dir = get_docker_user_data_dir()
co.set_user_data_path(str(user_data_dir))
log("INFO", f"Docker Chromium 用户目录: {user_data_dir}")
log("INFO", f"Docker Drission 临时目录: {tmp_root}")
# 很多 Debian/Ubuntu 系统的 Chromium 实际上是通过 wrapper 脚本调用的
# 直接指定确切的执行路径,防止 DrissionPage 底层启动失败
if os.path.exists('/usr/bin/chromium'):