将具有“null”值的 MS SQL 表导出到 CSV

Question

我试图弄清楚如何创建一个包含我在 MS SQL 数据库表中的 null 值的 csv 文件。现在我使用的脚本用''（空字符串）填充null值。我应该如何指示 csv Writer 保留空值？

源表示例

ID,Date,Entitled Key
10000002,NULL,805
10000003,2020-11-22 00:00:00,805

export_sql_to_csv.py

import csv
import os
import pyodbc

filePath = os.getcwd() + '/'
fileName = 'rigs_latest.csv'

server = 'ip-address'
database = 'db-name'
username = 'admin'
password = 'password'

# Database connection variable.
connect = pyodbc.connect('DRIVER={ODBC Driver 17 for SQL Server};SERVER=' +
                                 server+';DATABASE='+database+';UID='+username+';PWD=' + password)
cursor = connect.cursor()

sqlSelect = "SELECT * FROM my_table"


cursor.execute(sqlSelect)
results = cursor.fetchall()

# Extract the table headers.
headers = [i[0] for i in cursor.description]

# Open CSV file for writing.
csvFile = csv.writer(open(filePath + fileName, 'w', newline=''),
                     delimiter=',', lineterminator='\r\n',
                     quoting=csv.QUOTE_NONE, escapechar='\\')

# Add the headers and data to the CSV file.
csvFile.writerow(headers)
csvFile.writerows(results)

运行上述脚本后的结果示例：

ID,Date,Entitled Key
10000002,,805
10000003,2020-11-22 00:00:00,805

我想保留null 值的主要原因是我想将该csv 文件转换为一系列插入SQL 语句，并针对Aurora Serverless PostgreSQL 数据库执行这些语句。数据库不接受 date 类型的空字符串并导致该错误：ERROR: invalid input syntax for type date: ""

Answer 1

如docs for the csv module 中所述，None 值按设计以''（空字符串）形式写入 CSV。所有其他非字符串值首先调用str。

因此，如果您希望您的 CSV 包含字符串 null 而不是 ''，那么您必须在值到达 CSV 写入器之前对其进行修改。也许：

results = [
['null' if val is None else val for val in row] for row in results
]