33

app.py

@@ -1,7 +1,30 @@
 from fastapi import FastAPI
+import mariadb
+import os
 
 app = FastAPI()
 
+# MariaDB connection setup
+def get_db_connection():
+    conn = mariadb.connect(
+        user=os.getenv("MARIADB_USER", "root"),
+        password=os.getenv("MARIADB_PASSWORD", ""),
+        host=os.getenv("MARIADB_HOST", "localhost"),
+        port=int(os.getenv("MARIADB_PORT", 3306)),
+        database=os.getenv("MARIADB_DATABASE", "test")
+    )
+    return conn
+
 @app.get("/")
 def greet_json():
-    return {"Hello": "World!"}
+    # Example: test DB connection
+    try:
+        conn = get_db_connection()
+        cur = conn.cursor()
+        cur.execute("SELECT 1")
+        result = cur.fetchone()
+        cur.close()
+        conn.close()
+        return {"Hello": "World!", "db_test": result[0]}
+    except Exception as e:
+        return {"Hello": "World!", "db_error": str(e)}