From 398ccce3b53f4919efe09a59f721284a836bede6 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mateusz=20Gruszczy=C5=84ski?= Date: Tue, 4 Nov 2025 10:26:40 +0100 Subject: [PATCH] rewrite --- database/__init__.py | 43 +++++++++++++++++++++++++++++++------------ 1 file changed, 31 insertions(+), 12 deletions(-) diff --git a/database/__init__.py b/database/__init__.py index f1a74a2..bed6798 100644 --- a/database/__init__.py +++ b/database/__init__.py @@ -2,25 +2,44 @@ 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""" + """Initialize database - create tables""" + with app.app_context(): - db.create_all() - print("[DB] All tables created successfully", flush=True) + 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 - admin = User.query.filter_by(username='admin').first() - if not admin: - 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) - else: - print("[DB] Admin user already exists", flush=True) + 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