如何实现支持自定义参数的SQLite交互式表创建程序

我帮你完善了这个交互式的SQLite表创建工具，解决了原代码中列数固定、输入提示有误的问题，让它能支持任意数量的列定义，完全匹配你需要的交互流程：

import sqlite3

# 连接/创建目标数据库
db_name = input("请输入要创建的数据库名称（建议带.db后缀，比如mydb.db）：")
connection = sqlite3.connect(db_name)
cursor = connection.cursor()

# 获取要创建的表名称
table_name = input("请输入要创建的表名称：")

# 获取表需要的列数量
column_count = int(input("请输入表应包含的列数："))
columns = []

# 逐个输入列的完整定义
for i in range(column_count):
    # 提示用户输入格式，支持添加约束（比如主键、自增）
    column_def = input(f"请输入第{i+1}列的定义（格式：列名 数据类型，例：username TEXT 或 id INTEGER PRIMARY KEY AUTOINCREMENT）：")
    columns.append(column_def)

# 构建创建表的SQL语句
create_table_sql = f"CREATE TABLE IF NOT EXISTS {table_name} ({', '.join(columns)})"

# 先删除已存在的同名表（可选操作，根据需求保留）
cursor.execute(f"DROP TABLE IF EXISTS {table_name}")
# 执行创建表的SQL
cursor.execute(create_table_sql)

# 提交更改并关闭数据库连接
connection.commit()
connection.close()

print(f"数据库{db_name}和表{table_name}已成功创建！")

关键改进说明： #

• 灵活适配任意列数：不再像原代码那样硬编码列数量，用户输入多少列就生成多少列的表结构
• 支持完整列约束：允许用户输入带约束的列定义，比如自增主键、非空限制等，满足更多场景需求
• 友好的输入提示：优化了输入引导文字，避免拼写错误，让用户更清楚该输入什么格式
• 规范的数据库操作：最后主动提交更改并关闭连接，确保操作生效且资源正常释放

如果需要默认把第一列设为自增主键，可以在循环输入列之后添加这行代码：

# 自动插入自增主键作为第一列
columns.insert(0, "id INTEGER PRIMARY KEY AUTOINCREMENT")

内容的提问来源于stack exchange，提问作者Giorgi