81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
#!/usr/bin/env python
|
||
"""
|
||
项目初始化脚本
|
||
用于快速初始化新项目
|
||
"""
|
||
import os
|
||
import shutil
|
||
import secrets
|
||
import sys
|
||
|
||
def generate_secret_key():
|
||
"""生成安全的密钥"""
|
||
return secrets.token_hex(32)
|
||
|
||
def init_project():
|
||
"""初始化项目"""
|
||
print("=" * 50)
|
||
print("Python项目初始化脚本")
|
||
print("=" * 50)
|
||
|
||
# 获取项目名称
|
||
project_name = input("请输入项目名称（用于替换your_app）: ").strip()
|
||
if not project_name:
|
||
print("错误: 项目名称不能为空")
|
||
sys.exit(1)
|
||
|
||
# 生成密钥
|
||
secret_key = generate_secret_key()
|
||
print(f"\n生成的SECRET_KEY: {secret_key}")
|
||
|
||
# 创建.env文件
|
||
if not os.path.exists('.env'):
|
||
print("\n创建.env文件...")
|
||
shutil.copy('env.example', '.env')
|
||
|
||
# 替换SECRET_KEY
|
||
with open('.env', 'r') as f:
|
||
content = f.read()
|
||
content = content.replace('your-secret-key-here-change-this-in-production', secret_key)
|
||
with open('.env', 'w') as f:
|
||
f.write(content)
|
||
print("✓ .env文件已创建")
|
||
else:
|
||
print("⚠ .env文件已存在，跳过创建")
|
||
|
||
# 创建必要目录
|
||
directories = ['logs', 'uploads', 'migrations']
|
||
for directory in directories:
|
||
if not os.path.exists(directory):
|
||
os.makedirs(directory)
|
||
print(f"✓ 创建目录: {directory}")
|
||
|
||
# 重命名应用目录（可选）
|
||
rename = input(f"\n是否重命名应用目录 src/your_app 为 src/{project_name}? (y/n): ").strip().lower()
|
||
if rename == 'y':
|
||
old_path = 'src/your_app'
|
||
new_path = f'src/{project_name}'
|
||
if os.path.exists(old_path) and not os.path.exists(new_path):
|
||
shutil.move(old_path, new_path)
|
||
print(f"✓ 已重命名: {old_path} -> {new_path}")
|
||
print(f"⚠ 请手动更新以下文件中的导入路径:")
|
||
print(f" - run_dev.py")
|
||
print(f" - run_production.py")
|
||
print(f" - gunicorn.conf.py")
|
||
print(f" - tests/conftest.py")
|
||
print(f" - 所有测试文件")
|
||
|
||
print("\n" + "=" * 50)
|
||
print("初始化完成！")
|
||
print("=" * 50)
|
||
print("\n下一步:")
|
||
print("1. 编辑.env文件，配置数据库和其他设置")
|
||
print("2. 安装依赖: pip install -r requirements/base.txt -r requirements/development.txt")
|
||
print("3. 初始化数据库: flask db init && flask db migrate -m 'Initial migration' && flask db upgrade")
|
||
print("4. 运行项目: python run_dev.py")
|
||
print("\n")
|
||
|
||
if __name__ == '__main__':
|
||
init_project()
|
||
|