### 获取数据库表列名的 Python 代码示例 以下是通过不同数据库驱动程序获取数据库表列名的具体实现方法。 #### 使用 `sqlite3` 驱动获取 SQLite 数据库表列名 当使用 `sqlite3` 连接 SQLite 数据库时，可以通过执行 SQL 查询语句 `"PRAGMA table_info(table_name)"` 来获取指定表的列信息。每条记录返回的结果中包含了列名以及其他元数据。 ```python import sqlite3 def get_columns_sqlite(db_path, table_name): conn = sqlite3.connect(db_path) cursor = conn.cursor() # 执行 PRAGMA table_info 查询 cursor.execute(f"PRAGMA table_info({table_name})") columns = [column[1] for column in cursor.fetchall()] conn.close() return columns # 示例调用 db_path = r"D:\pyWork\test.db" table_name = "staff" columns = get_columns_sqlite(db_path, table_name) print("SQLite 表列名:", columns) ``` 这种方法适用于 SQLite 数据库，并能有效提取表结构中的列名[^1]^。 --- #### 使用 `pymysql` 驱动获取 MySQL 数据库表列名 对于 MySQL 数据库，可以利用 `pymysql` 库建立连接后，通过查询 `information_schema.columns` 或者直接运行 `SHOW COLUMNS FROM table_name` 获得所需列名。 ```python import pymysql def get_columns_mysql(host, port, user, password, database, table_name): connection = pymysql.connect( host=host, port=port, user=user, passwd=password, db=database, charset='utf8' ) cursor = connection.cursor() # 方法一：通过 SHOW COLUMNS 查询 cursor.execute(f"SHOW COLUMNS FROM {table_name}") columns = [column[0] for column in cursor.fetchall()] # 方法二：通过 information_schema 查询 (可选) """ query = f""" SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.COLUMNS WHERE TABLE_SCHEMA='{database}' AND TABLE_NAME='{table_name}' ORDER BY ORDINAL_POSITION; """ cursor.execute(query) columns = [col[0] for col in cursor.fetchall()] """ connection.close() return columns # 示例调用 mysql_config = { 'host': 'localhost', 'port': 3306, 'user': 'root', 'password': '123456', 'database': 'stats', } table_name = "t_tel_user_stat" columns = get_columns_mysql(**mysql_config, table_name=table_name) print("MySQL 表列名:", columns) ``` 该方案展示了如何基于 `pymysql` 提取 MySQL 中特定表的所有列名称[^3]^。 --- #### 处理复杂 SQL 的通用解决方案 如果面对的是复杂的 SQL 查询而非简单的表定义，则需要借助专门工具来解析 SQL 并提取其中涉及的表名和字段名。推荐使用 ANTLR4 工具完成这一任务，它能够精确识别嵌套子查询及其他高级特性下的目标实体。 由于此部分较为深入且依赖外部资源构建语法树模型，这里仅提供概念指导而不展开具体编码细节[^4]^。 --- ### 总结 以上分别介绍了针对 SQLite 和 MySQL 数据库的操作技巧以及应对更广泛场景时采用的技术路线图。实际应用过程中应根据项目需求选取合适的策略组合实施开发工作。