def insert_data(table_name, column_values): query = "INSERT INTO %s (%s) VALUES (%s)" columns = ', '.join(column_values.keys()) values = ', '.join(['%s'] * len(column_values)) query = query % (table_name, columns, values) execute_operation(query, tuple(column_values.values()))根据这段代码，如果每次插入一条数据就需要关联另外一张表的id，应该怎么写

如果需要在插入数据时关联另一张表的id，您可以在INSERT INTO语句中添加一个子查询来获取该id。例如，假设您需要关联的表名为"related_table"，id列名为"related_id"，则可以按照以下方式修改insert_data()函数：

def insert_data(table_name, column_values, related_id):
    query = "INSERT INTO %s (%s, related_id) VALUES (%s, (SELECT id FROM related_table WHERE id=%s))"
    columns = ', '.join(column_values.keys())
    values = ', '.join(['%s'] * len(column_values))
    query = query % (table_name, columns, values, related_id)
    execute_operation(query, tuple(column_values.values()))

在这个例子中，我们把原来的INSERT INTO语句修改为了"INSERT INTO %s (%s, related_id) VALUES (%s, (SELECT id FROM related_table WHERE id=%s))"，其中"related_id"是要插入的关联id的列名，"(SELECT id FROM related_table WHERE id=%s)"是一个子查询，用于获取相关的id。注意，我们在execute_operation()函数中添加了一个额外的参数"related_id"，该参数将在子查询中使用。