py-ailbl-user-profile/apps/crud.py

import io
from flask import Response
from helpers import init_db_connection, CORS_HEADERS


def get_profile(id):
    try:
        conn = init_db_connection()
        cursor = conn.cursor()

        # Truy van thong tin nguoi dung tu bang ailbl_user_profiles
        query = "SELECT * FROM ailbl_user_profile WHERE id = %s"
        cursor.execute(query, (id,))
        profile = cursor.fetchone()  # fetchone la gi ?

        if profile:
            return {
                "id": profile[0],
                "first_name": profile[1],
                "last_name": profile[2],
                "dob": profile[3],
                "gender": profile[4],
                "created": profile[5],
                "modified": profile[6]
            }, 200, CORS_HEADERS
        else:
            return {"error": "Profile not found"}, 404, CORS_HEADERS
    except Exception as e:
        return {"error": str(e)}, 500, CORS_HEADERS
    finally:
        if conn:
            conn.close()


def update_profile(id, data):
    try:
        conn = init_db_connection()
        cursor = conn.cursor()
        # Cap nhat thong tin nguoi dung trong bang ailbl_user_profiles
        query = """
        UPDATE ailbl_user_profile
        SET first_name = %s, last_name = %s, dob = %s, gender = %s, modified = NOW()
        WHERE id = %s
        """
        cursor.execute(query, (
            data.get("first_name"),
            data.get("last_name"),
            data.get("dob"),
            data.get("gender"),
            id
        ))
        conn.commit()
        return {"message": "Profile updated successfully"}, 200, CORS_HEADERS

    except Exception as e:
        return {"error": str(e)}, 500, CORS_HEADERS
    finally:
        if conn:
            conn.close()


def insert_profile(id, data):
    try:
        conn = init_db_connection()
        cursor = conn.cursor()
        # Tao moi thong tin nguoi dung trong bang ailbl_user_profile
        query = """
        INSERT INTO ailbl_user_profile (id, first_name, last_name, dob, gender, created, modified)
        VALUES (%s, %s, %s, %s, %s, NOW(), NOW())
        """
        cursor.execute(query, (
            id,
            data.get("first_name"),
            data.get("last_name"),
            data.get("dob"),
            data.get("gender"),
        ))
        conn.commit()
        return {"message": "Profile created successfully"}, 201, CORS_HEADERS
    except Exception as e:
        return {"error": str(e)}, 500, CORS_HEADERS
    finally:
        if conn:
            conn.close()


def delete_profile(id):
    try:
        conn = init_db_connection()
        cursor = conn.cursor()

        # Xoa thong tin nguoi dung trong bang ailbl_user_profiles
        query = "DELETE FROM ailbl_user_profile WHERE id = %s"
        cursor.execute(query, (id,))

        conn.commit()
        return {"message": "Profile deleted successfully"}, 200, CORS_HEADERS
    except Exception as e:
        return {"error": str(e)}, 500, CORS_HEADERS
    finally:
        if conn:
            conn.close()