Update app2.py

app2.py

@@ -34,6 +34,56 @@ class Settings(BaseSettings):
 
 settings = Settings()
 
+# --- Database Connection ---
+def get_db_url(settings: Settings) -> str:
+    if settings.DATABASE_TYPE == "mysql":
+        return f"mysql+aiomysql://{settings.DATABASE_USER}:{settings.DATABASE_PASSWORD}@{settings.DATABASE_HOST}:{settings.DATABASE_PORT}/{settings.DATABASE_NAME}"
+    elif settings.DATABASE_TYPE == "postgresql":
+        return f"postgresql+asyncpg://{settings.DATABASE_USER}:{settings.DATABASE_PASSWORD}@{settings.DATABASE_HOST}:{settings.DATABASE_PORT}/{settings.DATABASE_NAME}"
+    else:
+        return "sqlite+aiosqlite:///default.db"
+
+async def set_db_connection(
+    db_type: str = None,
+    db_host: str = None,
+    db_port: int = None,
+    db_user: str = None,
+    db_password: str = None,
+    db_name: str = None
+):
+    global db_session, engine, settings
+    try:
+        # Update settings if new values provided
+        if db_type:
+            settings.DATABASE_TYPE = db_type
+        if db_host:
+            settings.DATABASE_HOST = db_host
+        if db_port:
+            settings.DATABASE_PORT = db_port
+        if db_user:
+            settings.DATABASE_USER = db_user
+        if db_password:
+            settings.DATABASE_PASSWORD = db_password
+        if db_name:
+            settings.DATABASE_NAME = db_name
+
+        # Close existing connection if any
+        if db_session:
+            await db_session.close()
+        if engine:
+            await engine.dispose()
+
+        # Create new connection
+        db_url = get_db_url(settings)
+        engine = create_async_engine(db_url, echo=False)
+        async_session_maker = sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)
+        db_session = async_session_maker()
+        logger.info("Database connection established.")
+        return "Database connection established."
+    except Exception as e:
+        logger.error(f"Failed to establish database connection: {e}")
+        return f"Failed to connect to database: {e}"
+
 # --- Database Model (Example) ---
 class Article:
     def __init__(self, title, url, content, timestamp):