如何用Python檢查SQLite中是否存在行？

Question

我有遊標與查詢語句如下：

cursor.execute("select rowid from components where name = ?", (name,)) 

我要檢查組件的存在：名稱，並返回到Python變量。 我該怎麼做？

Answer 1

由於name s爲獨一無二的，我真的很贊成使用fetchone或使用SELECT count(*)在我使用fetchall的初始建議的亞歷克斯·馬爾泰利的方法，你（的OP的）方法。

fetchall將結果（通常是多行數據）包裝在列表中。由於name是唯一的，因此fetchall返回列表中只有一個元組的列表（例如[(rowid,),]或空列表[]。如果您想知道rowid，則使用fetchall需要您在列表和元組中挖掘以獲取到 的rowid。

使用fetchone在這種情況下更好，因爲你得到的只是一排，(rowid,)或None。爲了獲得在rowid（前提是有一個），你只需要摘掉元組的第一個元素

如果你不關心特定rowid，你只是想知道有一個命中， 然後你可以使用亞歷克斯Martelli的建議，SELECT count(*)，這將返回(1,)或(0,)。

下面是一些示例代碼：

首先一些鍋爐板代碼設置一個玩具SQLite表：

import sqlite3 
connection = sqlite3.connect(':memory:') 
cursor=connection.cursor() 
cursor.execute('create table components (rowid int,name varchar(50))')  
cursor.execute('insert into components values(?,?)', (1,'foo',)) 

使用fetchall：

for name in ('bar','foo'): 
    cursor.execute("SELECT rowid FROM components WHERE name = ?", (name,)) 
    data=cursor.fetchall() 
    if len(data)==0: 
     print('There is no component named %s'%name) 
    else: 
     print('Component %s found with rowids %s'%(name,','.join(map(str, next(zip(*data)))))) 

收率：

There is no component named bar 
Component foo found with rowids 1 

使用fetchone：

for name in ('bar','foo'): 
    cursor.execute("SELECT rowid FROM components WHERE name = ?", (name,)) 
    data=cursor.fetchone() 
    if data is None: 
     print('There is no component named %s'%name) 
    else: 
     print('Component %s found with rowid %s'%(name,data[0])) 

收率：

There is no component named bar 
Component foo found with rowid 1 

使用SELECT count(*)：

for name in ('bar','foo'): 
    cursor.execute("SELECT count(*) FROM components WHERE name = ?", (name,)) 
    data=cursor.fetchone()[0] 
    if data==0: 
     print('There is no component named %s'%name) 
    else: 
     print('Component %s found in %s row(s)'%(name,data)) 

收率：

There is no component named bar 
Component foo found in 1 row(s) 

Answer 2

我找到了答案。

exist = cursor.fetchone() 
if exist is None: 
    ... # does not exist 
else: 
    ... # exists 

Answer 3

由於兩個現有的回答（你自己和@ unutbu的）所指出的，關鍵是你需要做一些排序取的，在執行SELECT，後檢查是否已經有任何結果到選擇與否（無論你是通過單個獲取並檢查沒有，或檢查所有並檢查一個空列表，這是一個邊際區別 - 考慮到你提到了約束條件，它們基本上是等價的方法）。

對於一個非常直接的答案，你可以select count(*) from components where name = ?，而不是選擇rowid，如果你只關心的是給定的名稱值是否存在（而不是關心它是什麼行ID，如果有的話;-)。執行此選擇，以及獲取的結果，爲您提供了0如果該值不存在，1如果它的存在（沒有其他的結果是可能給你繞UNIQUE約束上name列;-)的評論中提及。

Answer 4

爲了使它更短...：

row = cursor.fetchone() 

if row: 
    print "row is present" 
else: 
    print "row is not present"