"""
删除管理员用户脚本
"""
import sys
import os

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

from app.core.database import SessionLocal
from app.models.user import User


def delete_admin_user():
    """删除管理员用户"""
    db = SessionLocal()
    try:
        # 查找admin用户
        existing_user = db.query(User).filter(User.username == "admin").first()
        if existing_user:
            db.delete(existing_user)
            db.commit()
            print(f"Admin user deleted successfully")
        else:
            print(f"Admin user not found")

    except Exception as e:
        db.rollback()
        print(f"Error deleting admin user: {e}")
    finally:
        db.close()


if __name__ == "__main__":
    delete_admin_user()