Python树形函数逐行取数：将PHP树形层级选项生成代码转Python

我来帮你搞定这个需求——既要实现Python里树形结构的逐行取数，又要把那段PHP生成树形的代码转成Python版本。咱们一步步拆解：

一、核心逻辑分析 #

先看你的PHP代码：它通过递归查询数据库中parent_id匹配的子节点，逐行取出数据后拼接带层级缩进的<option>标签，最终生成完整的树形下拉选项。Python版本我们要复刻这个逻辑，同时优化字符串拼接的效率，还要实现真正的逐行取数。

二、Python实现方案 #

1. 完整递归生成树形函数 #

这里以常用的pymysql（MySQL驱动）为例，如果你用PostgreSQL可以换成psycopg2，逻辑是通用的：

def get_all_sectors(db_connection, parent_id=0, indent=""):
    # 定义SQL查询，和PHP的逻辑完全一致
    query = """
        SELECT id, name FROM sectors 
        WHERE parent_id = %s 
        ORDER BY parent_id, sort_order
    """
    # 创建游标，设置dictionary=True让返回结果是字典（和PHP的fetch()关联数组一致）
    cursor = db_connection.cursor(dictionary=True)
    cursor.execute(query, (parent_id,))
    
    # 用列表收集结果，比直接拼接字符串更高效（Python字符串不可变，频繁拼接会浪费内存）
    output = []
    
    # 逐行取数处理（这里用fetchone循环实现真正的逐行读取，适合大数据量场景）
    row = cursor.fetchone()
    while row:
        # 拼接带层级缩进的option标签，缩进用 实现视觉层级，也可以换成空格
        option_tag = f'<option value="{row["id"]}">{indent}{row["name"]}</option>'
        output.append(option_tag)
        
        # 递归处理当前节点的子节点，缩进增加一个层级
        child_indent = indent + "&emsp;"
        child_options = get_all_sectors(db_connection, row["id"], child_indent)
        output.extend(child_options)
        
        # 取下一行数据
        row = cursor.fetchone()
    
    cursor.close()
    # 把列表转为换行分隔的字符串，方便直接插入HTML
    return "\n".join(output)

2. 函数使用示例 #

先建立数据库连接，再调用函数生成选项：

import pymysql

# 替换成你的数据库配置
db_config = {
    "host": "localhost",
    "user": "your_username",
    "password": "your_password",
    "database": "your_database"
}

# 建立连接并生成选项
with pymysql.connect(**db_config) as db:
    sector_options = get_all_sectors(db)
    # 可以直接把这个结果插入到HTML的<select>标签中
    print(sector_options)

三、逐行取数的细节说明 #

如果你的数据量很大，不想一次性把所有节点加载到内存里，上面代码中的while row = cursor.fetchone()就是标准的逐行取数实现——每次只从数据库读取一行数据处理，处理完再取下一行，避免内存占用过高。如果数据量小，用for row in cursor.fetchall()循环也没问题，代码会更简洁。

内容的提问来源于stack exchange，提问作者Tarvo Mäesepp