company_official/backend/seed.py

from app import create_app, db
from app.models import User
import sys

app = create_app()

try:
    with app.app_context():
        print("Checking for admin user...")
        if not User.query.filter_by(username='admin').first():
            print("Creating admin user...")
            user = User(username='admin')
            user.set_password('123456')
            db.session.add(user)
            db.session.commit()
            print("Admin user created.")
        else:
            print("Admin user already exists.")
except Exception as e:
    print(f"Error: {e}")
    sys.exit(1)