支持命令行任意参数的通用建表脚本，有比sys.argv更好的方案吗？

你想要实现一个通用脚本，通过命令行传参创建数据库表，期望传参格式为：

python create_table.py table_name=test name=TEXT city=TEXT

每次运行可创建包含不同字段（带对应类型）的新表。目前你用sys.argv实现的传参格式是：

python create_table.py test name TEXT city TEXT ...etc

其中第一个参数是数据库文件名，第二个是表名，后续参数为成对的字段名和类型，支持任意数量字段。当前代码存在参数顺序易出错、无校验、无提示等问题，下面推荐两种更优的实现方式：

一、使用标准库argparse（无需额外安装） #

argparse支持自定义参数格式、自动生成帮助信息，还能做参数校验，比直接操作sys.argv更健壮。以下是适配你期望传参格式的实现：

import sqlite3
import argparse

def parse_args():
    parser = argparse.ArgumentParser(description='创建SQLite数据库表')
    parser.add_argument('params', nargs='+', help='参数格式：table_name=表名 字段名=类型 ...，例如 table_name=test name=TEXT city=TEXT')
    args = parser.parse_args()
    
    # 解析key=value格式的参数
    param_dict = {}
    for param in args.params:
        key, value = param.split('=', 1)
        param_dict[key.strip()] = value.strip()
    
    # 校验必要参数
    if 'table_name' not in param_dict:
        parser.error('必须指定table_name参数，例如 table_name=test')
    if len(param_dict) < 2:
        parser.error('至少需要指定一个字段（格式：字段名=类型）')
    
    return param_dict

def main():
    params = parse_args()
    table_name = params.pop('table_name')
    # 若需自定义数据库文件名，可添加db_name=xxx参数，这里默认用表名作为数据库名
    db_file = f'db/{table_name}.db'
    
    conn = sqlite3.connect(db_file)
    print("数据库打开成功")
    
    # 构建CREATE TABLE语句
    fields = [f"{name} {type_}" for name, type_ in params.items()]
    create_query = f'CREATE TABLE {table_name} ({", ".join(fields)})'
    print(create_query)
    
    try:
        conn.execute(create_query)
        print("表创建成功")
    except sqlite3.Error as e:
        print(f"创建表失败：{e}")
    finally:
        conn.close()

if __name__ == '__main__':
    main()

优势说明 #

• 支持你期望的key=value传参格式，参数顺序不再受限
• 运行python create_table.py -h即可查看自动生成的帮助信息
• 自动校验必要参数，缺失时给出明确错误提示
• 代码结构清晰，易于维护和扩展

二、使用第三方库click（更简洁的语法） #

如果可以安装第三方库，click能让命令行参数的定义更直观，语法更简洁：

首先安装click：

pip install click

实现代码：

import sqlite3
import click

@click.command()
@click.argument('params', nargs=-1)
def create_table(params):
    # 解析参数
    param_dict = {}
    for param in params:
        key, value = param.split('=', 1)
        param_dict[key.strip()] = value.strip()
    
    if 'table_name' not in param_dict:
        click.echo('错误：必须指定table_name参数，例如 table_name=test', err=True)
        return
    if len(param_dict) < 2:
        click.echo('错误：至少需要指定一个字段（格式：字段名=类型）', err=True)
        return
    
    table_name = param_dict.pop('table_name')
    db_file = f'db/{table_name}.db'
    
    conn = sqlite3.connect(db_file)
    click.echo("数据库打开成功")
    
    fields = [f"{name} {type_}" for name, type_ in param_dict.items()]
    create_query = f'CREATE TABLE {table_name} ({", ".join(fields)})'
    click.echo(create_query)
    
    try:
        conn.execute(create_query)
        click.echo("表创建成功")
    except sqlite3.Error as e:
        click.echo(f"创建表失败：{e}", err=True)
    finally:
        conn.close()

if __name__ == '__main__':
    create_table()

额外提示 #

无论用哪种方式，都要注意SQL注入风险：虽然CREATE TABLE语句无法使用参数化查询，但可以对字段名和类型做简单校验（比如限制字段名仅含字母、下划线，类型仅允许SQLite支持的TEXT、INTEGER、REAL等），避免恶意参数破坏数据库。

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