Sqlserver
 sql >> Database >  >> RDS >> Sqlserver

Come chiamare una stored procedure con SQLAlchemy che richiede un parametro di tabella di tipo definito dall'utente

Esiste un driver che supporta davvero i TVP:Pytds . Non è ufficialmente supportato, ma esiste un'implementazione dialettale di terze parti:sqlalchemy-pytds . Usandoli potresti chiamare la tua procedura memorizzata in questo modo:

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 questo modo puoi passare grandi quantità di dati come parametri:

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',)

Se invece stai utilizzando un driver che non supporta i TVP, potresti dichiara una variabile di tabella , inserisci i valori e passalo come argomento alla tua procedura:

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',)]

Nota che non puoi usare alcun metodo executemany, o finirai per chiamare la procedura per ogni valore di tabella separatamente. Ecco perché i segnaposto vengono costruiti manualmente ei valori della tabella passati come argomenti individuali. È necessario prestare attenzione a non formattare alcun argomento direttamente nella query, ma la quantità corretta di segnaposto per l'API DB. I valori di riga sono limitati a un massimo 1000 .

Ovviamente sarebbe bello se il driver DB-API sottostante fornisse un supporto adeguato per i parametri con valori di tabella, ma almeno non riuscissi a trovare un modo per pymssql, che utilizza FreeTDS. Un riferimento ai TVP nella mailing list chiarisce che non sono supportati. La situazione è non molto migliore per PyODBC .

Disclaimer:non ho mai usato MS SQL Server prima.