Question

我想编写一个生成update语句的方法，而不需要硬编码列和值。该语句将包含可选的where子句，并将传递给executemany，该子句仅包含列和值，其中没有select。例如：

update TABLE
set
    Col1 = 'a',
    Col2 = 'b',
    Col3 = 'c'
where
    Col4 = 'd'
        and Col5 = 'e'
        and Col1 is null;

到目前为止我写的内容：

def update(self, table_name, update_columns, values, where_columns=None, where=True):
    update_columns_and_values = self.generator.generateColumnsAndPlaceholders(update_columns)
    if where:
        where_clause = self.generator.generateWhereClause(where_columns)
    else:
        where_clause = ''
    query = '''
            update {t}
            set
                {cv}
            {w}
        '''.format(t=table_name, cv=update_columns_and_values, w=where_clause)
    self.cursor.executemany(query, values)
    self.connection.commit()

def generateColumnsAndPlaceholders(columns):
    if type(columns) is str:
        columns = columns.split(', ')
    return ', \n'.join([str(c) + ' = ' + "'%s'" for c in columns])

现在，我应该如何编写一个函数generateWhereClause，它接受任意数量的列，并返回一个where子句，其中占位符针对非空值进行了调整（用=表示）和空值（用is null表示）？另外，我认为generateColumnsAndPlaceholders返回的字符串没有为null准备，因为占位符周围有单引号。如果是这样，我应该如何改变它？

一般情况下，如何在没有硬编码特定语句的情况下在update语句中处理null？

Answer 1

生成查询的函数 - 它采用表名，列{column: value}的值字典和约束字典，将None视为约束{column: constraint}

def update_query(table_name, values, constraints):
    v_list = [k + '=' + '"' + v + '"' for k, v in values.iteritems()]
    v_query = ', '.join(v_list)
    c_list = [k + (' IS NULL' if c is None else '=' + '"' + c + '"') for k, c in constraints.iteritems()]
    c_query = ' AND '.join(c_list)
    return 'UPDATE ' + table_name + ' SET ' + v_query + ' WHERE ' + c_query

测试代码：

tn = "table"
vl = {"Col1":"a","Col2":"b","Col3":"c"}
cn = {"Col4":"d","Col5":"e","Col":None}

结果：

UPDATE table SET Col2="b", Col3="c", Col1="a" WHERE Col6 IS NULL AND Col4="d" AND Col5="e"

我希望订单不是您的问题。

在update where子句中处理null而不对语句进行硬编码

1 个答案: