我正在嘗試撰寫一個 pytest 固定裝置,它回傳一些型別為 的測驗資料List[sa.engine.row.LegacyRow]。
問題是 sqllite 將日期時間物件轉換為 str。
import sqlalchemy as sa
from datetime import datetime
# test data
data = {'id': '1',
'date': datetime(2019, 12, 9, 14, 11, 2, 442076)}
sql = "SELECT " ", ".join([f":{col} as {col}" for col in data]) # SELECT :col1 as col1, :col2 as col2
engine = sa.create_engine("sqlite:///:memory:")
with engine.begin() as conn:
row = conn.execute(sa.text(sql), data).fetchall()
print(row)
print(type(row[0][1])) # str
如何保留日期時間物件?
uj5u.com熱心網友回復:
您可以使用TextClause.columns方法來指定回傳列的型別。
例如,這段代碼
with engine.begin() as conn:
q = sa.text("""SELECT id, date FROM test WHERE id = :id AND date = :date""")
q = q.columns(date=sa.DateTime)
res = conn.execute(q, data)
for row in res:
for col in row:
print(type(col), col)
產生這個輸出:
<class 'int'> 1
<class 'datetime.datetime'> 2019-12-09 14:11:02.442076
轉載請註明出處,本文鏈接:https://www.uj5u.com/caozuo/377838.html
標籤:sqlite sqlalchemy
