46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""Database module initialization"""
|
|
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask_migrate import Migrate
|
|
import logging
|
|
|
|
db = SQLAlchemy()
|
|
migrate = Migrate()
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def init_db(app):
|
|
"""Initialize database - create tables"""
|
|
|
|
with app.app_context():
|
|
try:
|
|
db.create_all()
|
|
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
|
|
|
|
try:
|
|
admin = User.query.filter_by(username='admin').first()
|
|
|
|
if not admin:
|
|
print("[DB] Creating default admin user...", flush=True)
|
|
admin = User(username='admin', is_admin=True)
|
|
admin.set_password('admin123')
|
|
|
|
db.session.add(admin)
|
|
db.session.commit()
|
|
print("[DB] Default admin user created (admin/admin123)", flush=True)
|
|
print(f"[DB] Hash: {admin.password_hash}", flush=True)
|
|
else:
|
|
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
|