SQLAlchemy从查询结果访问列类型

问题描述

我正在使用sqlAlchemy（使用pymssql驱动程序）连接到sql Server数据库。

import sqlalchemy

conn_string = f'mssql+pymssql://{uid}:{pwd}@{instance}/?database={db};charset=utf8'
sql = 'SELECT * FROM FAKETABLE;'

engine = sqlalchemy.create_engine(conn_string)
connection = engine.connect()
result = connection.execute(sql)
result.cursor.description

结果为：

(('col_1',1,None,None),('col_2',('col_3',4,('col_4',3,('col_5',None))

根据PEP 249（光标的.description属性）：

前两项（名称和type_code）为必填项，其他五项为可选项，如果无法提供有意义的值，则将其设置为“无”。

我假设整数(1,3)是列类型。

我的两个问题：

如何将这些整数映射到数据类型（例如char，integer等）？
这些是sql数据类型吗？如果否，是否可以获取 sql数据类型？

FWIW，当使用raw_connection()而不是connect()时，我得到相同的结果。

以相似的方式出现在三个问题上（它们不能回答这个特定问题）。我需要使用connect() + execute()方法。

解决方法

如果否，是否可以获取SQL数据类型？

SQL Server函数 sys.dm_exec_describe_first_result_set 可用于直接为提供的查询获取SQL列的数据类型：

SELECT column_ordinal,name,system_type_name,*
FROM sys.dm_exec_describe_first_result_set('here goes query',NULL,0) ;

在您的示例中：

sql = """SELECT column_ordinal,system_type_name 
    FROM sys.dm_exec_describe_first_result_set('SELECT * FROM FAKETABLE',0) ;"""

针对：

CREATE TABLE FAKETABLE(id INT,d DATE,country NVARCHAR(10));

SELECT column_ordinal,system_type_name 
FROM sys.dm_exec_describe_first_result_set('SELECT * FROM FAKETABLE',0) ;

+-----------------+----------+------------------+
| column_ordinal  |  name    | system_type_name |
+-----------------+----------+------------------+
|              1  | id       | int              |
|              2  | d        | date             |
|              3  | country  | nvarchar(10)     |
+-----------------+----------+------------------+

db<>fiddle demo

对于PEP249：type_code在不同的数据库类型上看起来并不相同。

所以这个答案将集中在MS SQL Server上。

如何将这些整数映射到数据类型（例如char，integer等）？

您可以使用以下代码创建从type_code到type_object的字典：

import inspect
import pymssql

code_map = {
    type_obj.value: (type_name,type_obj)
    for type_name,type_obj
    in inspect.getmembers(
        pymssql,predicate=lambda x: isinstance(x,pymssql.DBAPIType),)
}

这将产生以下命令：

{2: ('BINARY',<DBAPIType 2>),4: ('DATETIME',<DBAPIType 4>),5: ('DECIMAL',<DBAPIType 5>),3: ('NUMBER',<DBAPIType 3>),1: ('STRING',<DBAPIType 1>)}

不幸的是，我无权访问正在运行的MS SQL Server实例。因此，我无法检查类型结果是否与您的示例匹配。

这些是SQL数据类型吗？如果否，是否可以获取SQL数据类型？

查看PEP并得到以下结果：该字段不是SQL数据类型。这是“类型对象”。

DB API似乎不提供检查查询结果元数据的方法/函数。该API仅提供了一种将数据类型从SQL绑定到python类型的方法。

如果需要获取确切的SQL数据类型，则必须编写服务器特定的SQL查询。

可能是驱动程序本身。下面，我仅使用AdventureWorks上的pyodbc驱动程序，就拥有与您几乎相同的代码。我选择了一个包含许多不同数据类型的表，它们都在显示。

import sqlalchemy
conn_string = conn_string = f'mssql+pyodbc://{username}:{pwd}@{instance}/AdventureWorksLT2017?driver=ODBC+Driver+17+for+SQL+Server'
sql = 'SELECT TOP 10 * FROM SalesLT.Product;'

engine = sqlalchemy.create_engine(conn_string)
connection = engine.connect()

result = connection.execute(sql)
print(result.cursor.description)

输出：

(('ProductID',<class 'int'>,None,10,False),('Name',<class 'str'>,50,('ProductNumber',25,('Color',15,True),('StandardCost',<class 'decimal.Decimal'>,19,4,('ListPrice',('Size',5,('Weight',8,2,('ProductCategoryID',('ProductModelID',('SellStartDate',<class 'datetime.datetime'>,23,3,('SellEndDate',('DiscontinuedDate',('ThumbNailPhoto',<class 'bytearray'>,('ThumbnailPhotoFileName',('rowguid',36,('ModifiedDate',False))

您是否可以尝试使用该驱动程序作为比较？

pymssql python sql-server sqlalchemy

SQLAlchemy从查询结果访问列类型

问题描述

解决方法

相关问答