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
|
from flask_migrate import upgrade
|
||||||
|
|
||||||
def init_db():
|
def init_db():
|
||||||
@@ -7,6 +7,15 @@ def init_db():
|
|||||||
db.create_all()
|
db.create_all()
|
||||||
# Run any pending migrations
|
# Run any pending migrations
|
||||||
upgrade()
|
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__':
|
if __name__ == '__main__':
|
||||||
init_db()
|
init_db()
|
||||||
Reference in New Issue
Block a user