database.py

from config import database_config
from pymysql import connect


class Database:
    def __init__(self):
        self.config = database_config

    def execute(self, sql):
        """执行 SQL 并返回 {"columns": [列名...], "rows": [结果行...]}。"""
        conn = connect(**self.config)
        try:
            with conn.cursor() as cursor:
                cursor.execute(sql)
                columns = [desc[0] for desc in cursor.description] if cursor.description else []
                result = cursor.fetchall()

        finally:
            conn.close()

        return {"columns": columns, "rows": result}


if __name__ == "__main__":
    from result_formatter import format_result

    db = Database()
    sql = "SELECT * FROM t_user LIMIT 5;"  # Replace with your actual table name
    result = db.execute(sql)
    print(format_result(result))