88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
|
"""自动初始化管理员账号
|
|||
|
|
|
|||
|
|
在容器启动时运行，如果管理员账号不存在则自动创建。
|
|||
|
|
"""
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
|
|||
|
|
def auto_init():
|
|||
|
|
"""自动初始化管理员账号。"""
|
|||
|
|
try:
|
|||
|
|
# 设置环境
|
|||
|
|
sys.path.insert(0, "/app/api")
|
|||
|
|
os.chdir("/app/api")
|
|||
|
|
|
|||
|
|
from app_factory import create_app
|
|||
|
|
from extensions.ext_database import db
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
from services.account_service import AccountService, TenantService
|
|||
|
|
|
|||
|
|
# 创建应用上下文
|
|||
|
|
socketio_app, flask_app = create_app()
|
|||
|
|
|
|||
|
|
with flask_app.app_context():
|
|||
|
|
# 检查是否已有账号
|
|||
|
|
result = db.session.execute(text("SELECT COUNT(*) FROM accounts"))
|
|||
|
|
count = result.scalar()
|
|||
|
|
|
|||
|
|
if count > 0:
|
|||
|
|
print("[auto_init] 管理员账号已存在，跳过初始化")
|
|||
|
|
# 确保 dify_setups 表有记录（标记初始化完成）
|
|||
|
|
_ensure_setup_complete()
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 获取配置
|
|||
|
|
email = os.environ.get("ADMIN_EMAIL", "taiyi@baodan.com")
|
|||
|
|
name = os.environ.get("ADMIN_NAME", "taiyi")
|
|||
|
|
password = os.environ.get("ADMIN_PASSWORD", "taiyi1224")
|
|||
|
|
workspace_name = os.environ.get("ADMIN_WORKSPACE", "Taiyi's Workspace")
|
|||
|
|
|
|||
|
|
print(f"[auto_init] 正在创建管理员账号: {email}")
|
|||
|
|
|
|||
|
|
# 创建账号（is_setup=True 跳过注册限制检查）
|
|||
|
|
account = AccountService.create_account(
|
|||
|
|
email=email,
|
|||
|
|
name=name,
|
|||
|
|
interface_language="zh-Hans",
|
|||
|
|
password=password,
|
|||
|
|
is_setup=True,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 创建工作空间（is_setup=True 跳过创建限制检查）
|
|||
|
|
TenantService.create_owner_tenant_if_not_exist(account, workspace_name, is_setup=True)
|
|||
|
|
|
|||
|
|
# 标记初始化完成
|
|||
|
|
_ensure_setup_complete()
|
|||
|
|
|
|||
|
|
print(f"[auto_init] 管理员账号创建成功!")
|
|||
|
|
print(f"[auto_init] 邮箱: {email}")
|
|||
|
|
print(f"[auto_init] 密码: {password}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"[auto_init] 初始化失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _ensure_setup_complete():
|
|||
|
|
"""确保 dify_setups 表有记录，标记初始化已完成。"""
|
|||
|
|
from extensions.ext_database import db
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
result = db.session.execute(text("SELECT COUNT(*) FROM dify_setups"))
|
|||
|
|
count = result.scalar()
|
|||
|
|
|
|||
|
|
if count == 0:
|
|||
|
|
db.session.execute(
|
|||
|
|
text("INSERT INTO dify_setups (version, setup_at) VALUES (:version, :setup_at)"),
|
|||
|
|
{"version": "1.0.0", "setup_at": datetime.utcnow()}
|
|||
|
|
)
|
|||
|
|
db.session.commit()
|
|||
|
|
print("[auto_init] 已标记系统初始化完成")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
auto_init()
|