胡绍康andClaude Opus 4.6
8d5c8a04d6
- 新增 start.py 环境变量智能初始化功能 - 自动从 .env.example 复制配置文件 - 自动生成安全的 SECRET_KEY - 添加详细的用户提示和安全警告 - 更新 README.md 添加界面截图展示 - 仪表盘、邮箱管理、验证码提取、系统设置 - 清理旧的中文命名图片文件 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
启动脚本 - 智能初始化环境变量并启动 Flask 应用
|
|
"""
|
|
import os
|
|
import secrets
|
|
import shutil
|
|
import sys
|
|
|
|
from dotenv import load_dotenv
|
|
|
|
|
|
def ensure_env_file():
|
|
"""确保 .env 文件存在且 SECRET_KEY 已正确配置"""
|
|
env_file = ".env"
|
|
env_example = ".env.example"
|
|
|
|
# 1. 如果 .env 不存在，从 .env.example 复制
|
|
if not os.path.exists(env_file):
|
|
if os.path.exists(env_example):
|
|
print(f"⚠️ 未找到 {env_file}，从 {env_example} 复制...")
|
|
shutil.copy(env_example, env_file)
|
|
else:
|
|
print(f"❌ 错误：{env_example} 文件不存在")
|
|
sys.exit(1)
|
|
|
|
# 2. 读取 .env 文件内容
|
|
with open(env_file, "r", encoding="utf-8") as f:
|
|
lines = f.readlines()
|
|
|
|
# 3. 检查 SECRET_KEY 是否需要生成
|
|
secret_key_line_index = -1
|
|
current_secret_key = None
|
|
|
|
for i, line in enumerate(lines):
|
|
if line.strip().startswith("SECRET_KEY="):
|
|
secret_key_line_index = i
|
|
current_secret_key = line.split("=", 1)[1].strip()
|
|
break
|
|
|
|
# 4. 如果 SECRET_KEY 是占位符或为空，生成新的
|
|
if current_secret_key in [None, "", "your-secret-key-here"]:
|
|
new_secret_key = secrets.token_hex(32)
|
|
|
|
if secret_key_line_index >= 0:
|
|
lines[secret_key_line_index] = f"SECRET_KEY={new_secret_key}\n"
|
|
else:
|
|
# 如果没有找到 SECRET_KEY 行，添加到文件开头
|
|
lines.insert(0, f"SECRET_KEY={new_secret_key}\n")
|
|
|
|
# 写回文件
|
|
with open(env_file, "w", encoding="utf-8") as f:
|
|
f.writelines(lines)
|
|
|
|
print("=" * 60)
|
|
print("✅ 已自动生成新的 SECRET_KEY")
|
|
print(f"🔑 SECRET_KEY: {new_secret_key}")
|
|
print("=" * 60)
|
|
print("⚠️ 重要提醒：")
|
|
print(" 1. SECRET_KEY 用于加密数据库中的敏感信息")
|
|
print(" 2. 请务必备份 .env 文件，丢失后将无法解密旧数据")
|
|
print(" 3. 不要将 .env 文件提交到 Git 仓库")
|
|
print("=" * 60)
|
|
else:
|
|
print(f"✅ 使用现有的 SECRET_KEY（已保护数据安全）")
|
|
|
|
|
|
# 初始化环境文件
|
|
ensure_env_file()
|
|
|
|
# 加载 .env 文件中的环境变量
|
|
load_dotenv()
|
|
|
|
# 确保环境变量已加载
|
|
if not os.getenv("SECRET_KEY"):
|
|
print("❌ 错误：SECRET_KEY 环境变量未设置")
|
|
sys.exit(1)
|
|
|
|
# 直接运行 web_outlook_app.py
|
|
if __name__ == "__main__":
|
|
# 使用 exec 运行 web_outlook_app.py，保持 __name__ == "__main__"
|
|
with open("web_outlook_app.py", "r", encoding="utf-8") as f:
|
|
code = f.read()
|
|
exec(code, {"__name__": "__main__", "__file__": "web_outlook_app.py"})
|