Update init_db.py
This commit is contained in:
11
init_db.py
11
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():
|
||||
@@ -8,5 +8,14 @@ def init_db():
|
||||
# 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()
|
||||
Reference in New Issue
Block a user