diff --git a/init_db.py b/init_db.py index 336c7e8..b3ea5a9 100644 --- a/init_db.py +++ b/init_db.py @@ -1,4 +1,4 @@ -from app import app, db +from app import app, db, User from flask_migrate import upgrade def init_db(): @@ -7,6 +7,15 @@ def init_db(): db.create_all() # Run any pending migrations upgrade() + + # Create default admin user if it doesn't exist + admin = User.query.filter_by(username='admin').first() + if not admin: + admin = User(username='admin') + admin.set_password('admin123') # You should change this password after first login + db.session.add(admin) + db.session.commit() + print("Default admin user created!") if __name__ == '__main__': init_db() \ No newline at end of file