python - RedGate SQL 数据生成器 Python 脚本检查日期时间列是否为空

Question

我正在使用 RedGate SQL 数据生成器在 SQL Server 2016 中生成测试数据。我有一个表，其中包含一个名为 DELETE_TS 的可为空的 DATETIME 列。另一列 IUD_CODE 是 CHAR(1)。在生产中，当 DELETE_TS 不为空时，该列为“D”。否则它是“I”或“U”，具体取决于其他逻辑。

为了创建“真实的”测试数据，我正在尝试使用 Python 脚本生成器来创建该逻辑。我目前正在使用：

def main(config):
    if DELETE_TS is None:
        return "U"
    else:
        return "D"

但是每个列都出现“D”，即使 DELETE_TS 为空。

我以前在 Python 方面做得不多，所以我确信这是我缺少的一些小东西。

编辑：这是表中相关列的 DDL：

  CREATE TABLE dbo.DIM_MERCHANT (
  DELETE_TS DATETIME NULL,
  IUD_CODE CHAR(1) NULL
  )

Answer 1

这适用于 python 生成器。

def main(config):
if DELETE_TS:
    return "D"
else:
    return "U"