用于获取SQL查询及其列名并将其转换为以下json格式的Python代码

import json
import pymysql

def get_sql_query_as_json(sql_query):
    # Connect to the database
    connection = pymysql.connect(host='localhost',
                                 user='username',
                                 password='password',
                                 db='database_name')
    
    try:
        # Create a cursor object
        cursor = connection.cursor()
        
        # Execute the SQL query
        cursor.execute(sql_query)
        
        # Fetch all the rows
        rows = cursor.fetchall()
        
        # Get the column names
        column_names = [desc[0] for desc in cursor.description]
        
        # Create a list to store the results
        results = []
        
        # Iterate over the rows
        for row in rows:
            # Create a dictionary for each row
            row_dict = {}
            
            # Iterate over the column names and row values
            for column, value in zip(column_names, row):
                # Add the column name and value to the dictionary
                row_dict[column] = value
            
            # Add the row dictionary to the results list
            results.append(row_dict)
        
        # Convert the results to JSON format
        json_data = json.dumps(results)
        
        return json_data
    
    finally:
        # Close the cursor and connection
        cursor.close()
        connection.close()

# Example usage
sql_query = "SELECT * FROM table_name"
json_data = get_sql_query_as_json(sql_query)
print(json_data)