python psycopg2_Python psycopg2检查行存在

In Python psycopg2 how can I check if a row exists?

def track_exists(self, track_id):

cur = self.conn.cursor()

cur.execute("SELECT fma_track_id FROM tracks WHERE fma_track_id = %s", (track_id,))

if cur.fetchall() > 0:

return true

else:

return false

Currently I am getting

Traceback (most recent call last):

File "", line 1, in

File "mumu.py", line 38, in track_exists

if cur.fetchall() > 0:

TypeError: 'NoneType' object has no attribute '__getitem__'

解决方案

Don't use fetchall() (which returns a list, which is never 'larger than 0'), use fetchone():

def track_exists(self, track_id):

cur = self.conn.cursor()

cur.execute("SELECT fma_track_id FROM tracks WHERE fma_track_id = %s", (track_id,))

return cur.fetchone() is not None

fetchone() returns None if there is nothing to fetch, and testing against is not None gives you a handy boolean value to return directly.