python下向sqlite插入记录,是否存在过记录的判断不起作用

每次都是返回 False

import sqlite3

conn = sqlite3.connect('urls.db')
cursor = conn.cursor()
cursor.execute('CREATE TABLE IF NOT EXISTS  urls (id integer primary key autoincrement, link text)') 

def url_exists(url):
    sql_cmd = '''select * from urls where link = '%s' ''' % url
    cursor.execute(sql_cmd)
    res = cursor.fetchall()
    if len(res) > 0 :
        print '%s exists' % url
        return True
    else:
        sql_cmd_2 = '''insert into urls(link) values('%s') ''' % url
        cursor.execute(sql_cmd_2)
        print '%s added' % url
        return False
    cursor.close()
    conn.commit()
    conn.close()

url_status = url_exists('example.com')
print url_status
阅读 7.9k
1 个回答

因为添加之后你直接return True了,数据库没有commit。。。

import sqlite3

conn = sqlite3.connect('/tmp/urls.db')
cursor = conn.cursor()
cursor.execute('CREATE TABLE IF NOT EXISTS  urls (id integer primary key autoincrement, link text)') 

def url_exists(url):
    sql_cmd = '''select * from urls where link = '%s' ''' % url
    cursor.execute(sql_cmd)
    res = cursor.fetchall()
    suc = True
    if len(res) > 0 :
        print '%s exists' % url
        suc = True
    else:
        sql_cmd_2 = '''insert into urls(link) values('%s') ''' % url
        cursor.execute(sql_cmd_2)
        print '%s added' % url
        suc = False
    cursor.close()
    conn.commit()
    conn.close()
    return suc

url_status = url_exists('example.com')
print url_status
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题