python - 如何使用需要用户定义类型表参数的 SQLAlchemy 调用存储过程

Question

我在 MSSQL 服务器上有一个存储过程“prc_add_names”，它采用一个表值参数。参数本身是自定义类型“StringTable”，定义如下:

CREATE TYPE [dbo].[StringTable] AS TABLE([strValue] [nvarchar](max) NULL)

我不知道如何使用 SQLAlchemy 执行此过程。我习惯于使用 session.execute 调用带参数的过程，如下所示:

result = session.execute('prc_do_something :pArg', {pArg:'foo'})

但是，如果我只是传递一个字符串列表作为参数，这将不起作用:

result = session.execute('prc_add_names :pArg', {pArg: ['Name One', 'Name Two']})

这导致:

sqlalchemy.exc.ProgrammingError: (pymssql.ProgrammingError) (102, "Incorrect syntax near 'Name One'.DB-Lib error message 20018, severity 15:
General SQL Server error: Check messages from the SQL Server
") [SQL: 'prc_add_names %(pArg)s'] [parameters: {'pArg': ['Name One', 'Name Two']}] (Background on this error at: http://sqlalche.me/e/f405)

显然，SQLAlchemy 无法将我的字符串列表理解为尝试创建我的 StringTable 类型的参数，但在谷歌搜索和阅读文档几个小时后，我还没有弄清楚我应该如何处理它。

仅供引用，我无法控制此数据库，因此无法修改存储过程或任何其他选项。

编辑:我没有嫁给 SQLAlchemy。如果有另一个库可以处理这个问题，我很乐意改用它。

Answer 1

有一个真正支持TVPs的驱动:Pytds .它不受官方支持，但有一个第 3 方方言实现:sqlalchemy-pytds .使用它们，您可以像这样调用您的存储过程:

In [1]: engine.execute(DDL("CREATE TYPE [dbo].[StringTable] AS TABLE([strValue] [nvarchar](max) NULL)"))
Out[1]: <sqlalchemy.engine.result.ResultProxy at 0x7f235809ae48>

In [2]: engine.execute(DDL("CREATE PROC test_proc (@pArg [StringTable] READONLY) AS BEGIN SELECT * FROM @pArg END"))
Out[2]: <sqlalchemy.engine.result.ResultProxy at 0x7f2358027b70>

In [3]: arg = ['Name One', 'Name Two']

In [4]: import pytds

In [5]: tvp = pytds.TableValuedParam(type_name='StringTable',
   ...:                              rows=((x,) for x in arg))

In [6]: engine.execute('EXEC test_proc %s', (tvp,))
Out[6]: <sqlalchemy.engine.result.ResultProxy at 0x7f294e699e10>

In [7]: _.fetchall()
Out[7]: [('Name One',), ('Name Two',)]

通过这种方式，您可以将潜在的大量数据作为参数传递:

In [21]: tvp = pytds.TableValuedParam(type_name='StringTable',
    ...:                              rows=((str(x),) for x in range(100000)))

In [22]: engine.execute('EXEC test_proc %s', (tvp,))
Out[22]: <sqlalchemy.engine.result.ResultProxy at 0x7f294c6e9f98>

In [23]: _.fetchall()[-1]
Out[23]: ('99999',)

另一方面，如果您使用的驱动程序不支持 TVP，您可以 declare a table variable , 插入值，和 pass that as the argument到您的程序:

In [12]: engine.execute(
    ...:     """
    ...:     DECLARE @pArg AS [StringTable];
    ...:     INSERT INTO @pArg VALUES {placeholders};
    ...:     EXEC test_proc @pArg;
    ...:     """.format(placeholders=",".join(["(%s)"] * len(arg))),
    ...:     tuple(arg))
    ...:     
Out[12]: <sqlalchemy.engine.result.ResultProxy at 0x7f23580f2908>

In [15]: _.fetchall()
Out[15]: [('Name One',), ('Name Two',)]

请注意，您不能使用任何 executemany 方法，否则您最终将分别为每个表值调用过程。这就是为什么占位符是手动构造的，而表值作为单独的参数传递的原因。必须注意不要将任何参数直接格式化到查询中，而是要为 DB-API 使用正确数量的占位符。行值限制为 maximum of 1000 .

如果底层 DB-API 驱动程序为表值参数提供适当的支持，那当然很好，但至少我找不到使用 FreeTDS 的 pymssql 的方法。 reference to TVPs on the mailing list明确表示不支持它们。情况是not much better for PyODBC .

_{免责声明:我之前并没有真正使用过 MS SQL Server。}