我在 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。如果有另一个库可以处理这个问题,我很乐意改用它。
最佳答案
有一个真正支持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。
关于python - 如何使用需要用户定义类型表参数的 SQLAlchemy 调用存储过程,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50141058/