rewrite
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
"""Database initialization"""
|
||||
"""Database module initialization"""
|
||||
|
||||
from flask_sqlalchemy import SQLAlchemy
|
||||
from flask_migrate import Migrate
|
||||
@@ -6,22 +6,21 @@ from flask_migrate import Migrate
|
||||
db = SQLAlchemy()
|
||||
migrate = Migrate()
|
||||
|
||||
|
||||
def init_db(app):
|
||||
"""Initialize database with app"""
|
||||
db.init_app(app)
|
||||
migrate.init_app(app, db)
|
||||
|
||||
# Create tables
|
||||
"""Initialize database - create tables"""
|
||||
with app.app_context():
|
||||
db.create_all()
|
||||
print("[DB] All tables created successfully", flush=True)
|
||||
|
||||
# Create default admin user if not exists
|
||||
from database.models import User
|
||||
admin = User.query.filter_by(username='admin').first()
|
||||
|
||||
if not admin:
|
||||
from config.settings import DEFAULT_ADMIN_USERNAME, DEFAULT_ADMIN_PASSWORD
|
||||
admin = User(username=DEFAULT_ADMIN_USERNAME)
|
||||
admin.set_password(DEFAULT_ADMIN_PASSWORD)
|
||||
admin = User(username='admin', is_admin=True)
|
||||
admin.set_password('admin123')
|
||||
db.session.add(admin)
|
||||
db.session.commit()
|
||||
print(f"[DB] Created default admin user: {DEFAULT_ADMIN_USERNAME}", flush=True)
|
||||
print("[DB] Default admin user created (admin/admin123)", flush=True)
|
||||
else:
|
||||
print("[DB] Admin user already exists", flush=True)
|
||||
|
||||
Reference in New Issue
Block a user