import os
import sys

# 添加项目根目录到Python路径
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))

# 设置数据库URL为SQLite
os.environ['DATABASE_URL'] = 'sqlite:///kamaxitong.db'

from app import create_app, db
from app.models.admin import Admin
from werkzeug.security import generate_password_hash

# 创建应用实例
app = create_app()

with app.app_context():
    # 检查是否已有管理员数据
    admin_count = Admin.query.count()
    if admin_count > 0:
        print(f"Database already contains {admin_count} admin users. Skipping data insertion.")
        sys.exit(0)
    
    # 创建默认管理员账号
    print("Creating default admin users...")
    
    # 超级管理员
    admin = Admin(
        username='admin',
        email='admin@kamaxitong.com',
        role=1,  # 超级管理员
        status=1  # 正常
    )
    admin.set_password('admin123')
    db.session.add(admin)
    
    # 普通管理员
    test_admin = Admin(
        username='test_admin',
        email='test@kamaxitong.com',
        role=0,  # 普通管理员
        status=1  # 正常
    )
    test_admin.set_password('test123')
    db.session.add(test_admin)
    
    # 提交更改
    db.session.commit()
    
    print("✓ Created admin users:")
    print("  - Super admin: admin / admin123")
    print("  - Normal admin: test_admin / test123")