4

以下是我的问题的概括：

考虑表格

    ID    A    B    C
r1  1     1    0    1
.   .     .    .    .
.   .     .    .    .
.   .     .    .    .
rN  N     1    1    0

列A,B,C包含0或的位置1。我正在尝试编写一个 python 函数，该函数采用0's 和1's 的排列列表，生成一个查询，该查询将传递给 SQLite3，然后计算A,B,C这些排列之一中的记录数。

例如，如果我将以下列表传递给我的函数permList = [[1,0,1],[1,0,0]]，那么它会将[A,B,C]组合为[1,0,1]or的所有记录计数[1,0,0]。

目前我正在这样做

def permCount(permList):
    SQLexpression = "SELECT Count(*) FROM Table WHERE "

    for i in range(len(permList)):
        perm = permList[i]
        SQLexpression += "(A=" + str(perm[0]) + " AND B=" + str(perm[1]) + 
                      " AND C=" + str(perm[2]) + ")"
        if i!=len(permList)-1:
            SQLexpression += " OR "

    *Execute SQLexpression and return answer*

现在这很好，但似乎有点麻烦。有没有更好的方法来动态生成输入长度permList未知的 SQL 查询？

4

2 回答 2

11 
def permCount(permList):
    condition = ' OR '.join(['(A=? AND B=? AND C=?)' 
                             for row in permList])
    sql = "SELECT Count(*) FROM Table WHERE {c}".format(
        c=condition)
    args = sum(permList, [])
    cursor.execute(sql, args)

使用参数化的 SQL。这意味着不要使用字符串格式插入值，而是使用地标（例如?），然后将参数作为第二个参数提供给cursor.execute.

这是更简单的代码并且可以防止SQL 注入。

于 2013-09-07T14:45:24.863 回答
1

在您的主 for 循环中尝试这些更改，以利用 python 生成器和列表理解功能。

def permCount(permList):

SQLexpression = "SELECT Count(*) FROM Table WHERE "

for perm in permList:    # if you need the i for other reason, you could write:
                         # for i, perm in enumerate(permList)

    a, b, c = [str(_) for _ in perm]

    SQLexpression += "(A=" + a + " AND B=" + b + \
                  " AND C=" + c + ") OR "

SQLexpression = SQLexpression[:-4] + ";"   # Trim the last " OR "
于 2013-09-07T15:25:06.743 回答