rewrite
This commit is contained in:
@@ -2,25 +2,44 @@
|
|||||||
|
|
||||||
from flask_sqlalchemy import SQLAlchemy
|
from flask_sqlalchemy import SQLAlchemy
|
||||||
from flask_migrate import Migrate
|
from flask_migrate import Migrate
|
||||||
|
import logging
|
||||||
|
|
||||||
db = SQLAlchemy()
|
db = SQLAlchemy()
|
||||||
migrate = Migrate()
|
migrate = Migrate()
|
||||||
|
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
|
||||||
|
|
||||||
def init_db(app):
|
def init_db(app):
|
||||||
"""Initialize database - create tables"""
|
"""Initialize database - create tables"""
|
||||||
|
|
||||||
with app.app_context():
|
with app.app_context():
|
||||||
|
try:
|
||||||
db.create_all()
|
db.create_all()
|
||||||
print("[DB] All tables created successfully", flush=True)
|
print("[DB] All tables created successfully", flush=True)
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[DB] Error creating tables: {e}", flush=True)
|
||||||
|
raise
|
||||||
|
|
||||||
from database.models import User
|
from database.models import User
|
||||||
|
|
||||||
|
try:
|
||||||
admin = User.query.filter_by(username='admin').first()
|
admin = User.query.filter_by(username='admin').first()
|
||||||
|
|
||||||
if not admin:
|
if not admin:
|
||||||
|
print("[DB] Creating default admin user...", flush=True)
|
||||||
admin = User(username='admin', is_admin=True)
|
admin = User(username='admin', is_admin=True)
|
||||||
admin.set_password('admin123')
|
admin.set_password('admin123')
|
||||||
|
|
||||||
db.session.add(admin)
|
db.session.add(admin)
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
print("[DB] Default admin user created (admin/admin123)", flush=True)
|
print("[DB] Default admin user created (admin/admin123)", flush=True)
|
||||||
|
print(f"[DB] Hash: {admin.password_hash}", flush=True)
|
||||||
else:
|
else:
|
||||||
print("[DB] Admin user already exists", flush=True)
|
print("[DB] Admin user already exists", flush=True)
|
||||||
|
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
print(f"[DB] Error creating admin: {e}", flush=True)
|
||||||
|
db.session.rollback()
|
||||||
|
raise
|
||||||
|
|||||||
Reference in New Issue
Block a user