load_env.py 965 B

12345678910111213141516171819202122232425262728293031
  1. #!/usr/bin/env python3
  2. """
  3. 加载环境变量脚本
  4. """
  5. import os
  6. from dotenv import load_dotenv
  7. def load_environment():
  8. """加载环境变量"""
  9. # 尝试加载.env文件
  10. env_files = ['.env', '.env.development', '.env.local']
  11. for env_file in env_files:
  12. if os.path.exists(env_file):
  13. print(f"加载环境变量文件: {env_file}")
  14. load_dotenv(env_file)
  15. return True
  16. print("未找到环境变量文件,使用默认配置")
  17. return False
  18. if __name__ == "__main__":
  19. load_environment()
  20. # 显示关键配置
  21. print("\n关键配置:")
  22. print(f"DATABASE_URL: {os.getenv('DATABASE_URL', '未设置')}")
  23. print(f"SECRET_KEY: {'已设置' if os.getenv('SECRET_KEY') else '未设置'}")
  24. print(f"JWT_SECRET_KEY: {'已设置' if os.getenv('JWT_SECRET_KEY') else '未设置'}")
  25. print(f"DEBUG: {os.getenv('DEBUG', 'True')}")
  26. print(f"PORT: {os.getenv('PORT', '8000')}")