init_db.py

import os
import logging
from sqlalchemy import create_engine, select, delete
from sqlalchemy.orm import sessionmaker
from api.database import Base, User, OAuthAccount, Conversation, Message

# Setup logging
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)

# جلب URL قاعدة البيانات من المتغيرات البيئية
SQLALCHEMY_DATABASE_URL = os.getenv("SQLALCHEMY_DATABASE_URL", "sqlite:///./data/mgzon_users.db")
if not SQLALCHEMY_DATABASE_URL:
    logger.error("SQLALCHEMY_DATABASE_URL is not set in environment variables.")
    raise ValueError("SQLALCHEMY_DATABASE_URL is required.")

# إنشاء المحرك
engine = create_engine(SQLALCHEMY_DATABASE_URL, connect_args={"check_same_thread": False})

# إعداد الجلسة
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def init_db():
    logger.info("Starting database initialization...")

    # إنشاء الجداول
    Base.metadata.create_all(bind=engine)
    logger.info("Database tables created successfully.")

    # تنظيف البيانات غير المتسقة
    with SessionLocal() as session:
        try:
            # حذف سجلات oauth_accounts اللي مش مرتبطة بمستخدم موجود
            stmt = delete(OAuthAccount).where(
                OAuthAccount.user_id.notin_(select(User.id))
            )
            result = session.execute(stmt)
            deleted_count = result.rowcount
            session.commit()
            logger.info(f"Deleted {deleted_count} orphaned OAuth accounts.")

            # التأكد من إن كل المستخدمين ليهم is_active=True
            users = session.execute(select(User)).scalars().all()
            for user in users:
                if not user.is_active:
                    user.is_active = True
                    logger.info(f"Updated user {user.email} to is_active=True")
            session.commit()

            # اختبار إنشاء مستخدم ومحادثة (اختياري)
            test_user = session.query(User).filter_by(email="[email protected]").first()
            if not test_user:
                test_user = User(
                    email="[email protected]",
                    hashed_password="$2b$12$examplehashedpassword",  # استبدل بكلمة مرور مشفرة حقيقية
                    is_active=True,
                    display_name="Test User"
                )
                session.add(test_user)
                session.commit()
                logger.info("Test user created successfully.")

            test_conversation = session.query(Conversation).filter_by(user_id=test_user.id).first()
            if not test_conversation:
                test_conversation = Conversation(
                    conversation_id="test-conversation-1",
                    user_id=test_user.id,
                    title="Test Conversation"
                )
                session.add(test_conversation)
                session.commit()
                logger.info("Test conversation created successfully.")

        except Exception as e:
            session.rollback()
            logger.error(f"Error during initialization: {e}")
            raise
        finally:
            session.close()

    logger.info("Database initialization completed.")

if __name__ == "__main__":
    init_db()