#!/usr/bin/env python3
"""Create default admin: admin / admin123"""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

from app.auth import hash_password
from app.database import Base, SessionLocal, engine
from app.models import User


def main():
    Base.metadata.create_all(bind=engine)
    db = SessionLocal()
    if db.query(User).filter(User.username == "admin").first():
        print("admin already exists")
        return
    db.add(
        User(
            username="admin",
            password_hash=hash_password("admin123"),
            display_name="管理员",
            role="admin",
        )
    )
    db.add(
        User(
            username="grader",
            password_hash=hash_password("grader123"),
            display_name="阅卷员",
            role="grader",
        )
    )
    db.commit()
    print("Created users: admin/admin123, grader/grader123")


if __name__ == "__main__":
    main()