Update app.py

app.py

@@ -1,16 +1,130 @@
-# app.py
+import os
+from flask import Flask, jsonify, request, g
+import psycopg2
+from dotenv import load_dotenv
 
-from flask import Flask
+# Tải các biến môi trường từ file .env (chỉ cho môi trường phát triển cục bộ)
+load_dotenv()
 
-# Tạo một instance của ứng dụng Flask
-# __name__ là tên của module hiện tại, Flask sử dụng nó để biết
-# nơi tìm các tài nguyên như template và tệp tĩnh.
 app = Flask(__name__)
 
-# Định nghĩa một route (đường dẫn)
-# Decorator @app.route('/') nói với Flask rằng khi người dùng truy cập
-# địa chỉ gốc của trang web (ví dụ: http://127.0.0.1:5000/),
-# hàm bên dưới sẽ được thực thi.
+# Cấu hình kết nối cơ sở dữ liệu từ biến môi trường
+DB_HOST = os.getenv('DB_HOST')
+DB_NAME = os.getenv('DB_NAME')
+DB_USER = os.getenv('DB_USER')
+DB_PASSWORD = os.getenv('DB_PASSWORD')
+DB_PORT = os.getenv('DB_PORT')
+
+# Hàm để thiết lập kết nối cơ sở dữ liệu
+def get_db_connection():
+    if 'db' not in g:
+        try:
+            g.db = psycopg2.connect(
+                host=DB_HOST,
+                database=DB_NAME,
+                user=DB_USER,
+                password=DB_PASSWORD,
+                port=DB_PORT
+            )
+            print("Kết nối đến Supabase PostgreSQL thành công!")
+        except Exception as e:
+            print(f"Lỗi khi kết nối đến cơ sở dữ liệu: {e}")
+            g.db = None # Đảm bảo g.db là None nếu kết nối thất bại
+    return g.db
+
+# Hàm đóng kết nối cơ sở dữ liệu khi request kết thúc
+@app.teardown_appcontext
+def close_db_connection(exception):
+    db = g.pop('db', None)
+    if db is not None:
+        db.close()
+        print("Đã đóng kết nối Supabase PostgreSQL.")
+
+@app.route('/test_db')
+def test_db():
+    conn = get_db_connection()
+    if conn is None:
+        return jsonify({"message": "Không thể kết nối đến cơ sở dữ liệu.", "status": "error"}), 500
+
+    try:
+        cur = conn.cursor()
+        cur.execute("SELECT version();") # Lấy thông tin phiên bản PostgreSQL
+        db_version = cur.fetchone()[0]
+        cur.close()
+        return jsonify({
+            "message": "Kết nối cơ sở dữ liệu thành công!",
+            "db_version": db_version,
+            "status": "success"
+        })
+    except Exception as e:
+        return jsonify({"message": f"Lỗi truy vấn cơ sở dữ liệu: {e}", "status": "error"}), 500
+
+@app.route('/create_table')
+def create_table():
+    conn = get_db_connection()
+    if conn is None:
+        return jsonify({"message": "Không thể kết nối đến cơ sở dữ liệu.", "status": "error"}), 500
+
+    try:
+        cur = conn.cursor()
+        cur.execute("""
+            CREATE TABLE IF NOT EXISTS users (
+                id SERIAL PRIMARY KEY,
+                name VARCHAR(100) NOT NULL,
+                email VARCHAR(100) UNIQUE NOT NULL
+            );
+        """)
+        conn.commit() # Commit thay đổi
+        cur.close()
+        return jsonify({"message": "Bảng 'users' đã được tạo (nếu chưa tồn tại).", "status": "success"})
+    except Exception as e:
+        conn.rollback() # Hoàn tác nếu có lỗi
+        return jsonify({"message": f"Lỗi khi tạo bảng: {e}", "status": "error"}), 500
+
+@app.route('/add_user', methods=['POST'])
+def add_user():
+    conn = get_db_connection()
+    if conn is None:
+        return jsonify({"message": "Không thể kết nối đến cơ sở dữ liệu.", "status": "error"}), 500
+
+    data = request.json
+    if not data or 'name' not in data or 'email' not in data:
+        return jsonify({"message": "Yêu cầu cần 'name' và 'email' trong body JSON.", "status": "error"}), 400
+
+    name = data['name']
+    email = data['email']
+
+    try:
+        cur = conn.cursor()
+        cur.execute("INSERT INTO users (name, email) VALUES (%s, %s) RETURNING id;", (name, email))
+        user_id = cur.fetchone()[0]
+        conn.commit()
+        cur.close()
+        return jsonify({"message": "Người dùng đã được thêm thành công!", "user_id": user_id, "status": "success"}), 201
+    except Exception as e:
+        conn.rollback()
+        return jsonify({"message": f"Lỗi khi thêm người dùng: {e}", "status": "error"}), 500
+
+@app.route('/users')
+def get_users():
+    conn = get_db_connection()
+    if conn is None:
+        return jsonify({"message": "Không thể kết nối đến cơ sở dữ liệu.", "status": "error"}), 500
+
+    try:
+        cur = conn.cursor()
+        cur.execute("SELECT id, name, email FROM users;")
+        users_data = cur.fetchall()
+        cur.close()
+
+        users_list = []
+        for user in users_data:
+            users_list.append({"id": user[0], "name": user[1], "email": user[2]})
+
+        return jsonify({"users": users_list, "status": "success"})
+    except Exception as e:
+        return jsonify({"message": f"Lỗi khi lấy danh sách người dùng: {e}", "status": "error"}), 500
+
 @app.route('/')
 def welcome_page():
     """
@@ -56,9 +170,5 @@ def welcome_page():
     </html>
     """
 
-# Phần này đảm bảo rằng máy chủ phát triển chỉ chạy khi bạn thực thi tệp này trực tiếp,
-# không phải khi nó được import như một module ở nơi khác.
-# debug=True cung cấp các thông báo lỗi hữu ích và tự động tải lại máy chủ
-# khi bạn thay đổi mã, rất tiện lợi cho việc phát triển.
 if __name__ == '__main__':
     app.run(debug=True, host='0.0.0.0', port=7860)