sqlite：在自定义函数(sqlite3.Connection.create_function)中使用另一个sqlite命令时出错

goucqfw6 于 2023-03-23 发布在 SQLite

关注(0)|答案(1)|浏览(203)

我正在学习在Python中创建自定义sqlite3函数。这是我的代码：

import sqlite3

conn = sqlite3.connect("data.db")
c = conn.cursor()

query = "CREATE TABLE IF NOT EXISTS names (ID INTEGER UNIQUE PRIMARY KEY NOT NULL, name TEXT)"
c.execute(query)

query = "INSERT INTO names (name) VALUES ('foo')"
c.execute(query)

query = "INSERT INTO names (name) VALUES ('boo')"
c.execute(query)

conn.commit()

def get_ID(name):
    
    query = "SELECT ID FROM names WHERE name = '{}'".format(name)
    c.execute(query)
    id = c.fetchall()
    print(id)
    return id

print(get_ID("foo"))

conn.create_function("GETID", 1, get_ID)

query = "SELECT GETID(?)"
c.execute(query, ("foo",))
print(c.fetchall())

c.close()
conn.close()

这是最后的结果：

[(1,)]
[(1,)]
Traceback (most recent call last):
  File "D:\Sync1\Code\Python3\test.py", line 33, in <module>
    c.execute(query, ("foo",))
sqlite3.OperationalError: user-defined function raised exception

为什么我的函数给予出错？

sqlite

来源：https://stackoverflow.com/questions/75781507/sqlite-error-when-using-another-sqlite-command-inside-a-custom-function-sqlite

1条答案

按热度按时间

jecbmhm31#

在函数中使用另一个本地游标来执行SELECT语句。
此外，fetchall()返回子列表的列表（查询返回的所有行的列表），而不是标量值。
相反，取ID和fetchone()[0]。
最后，将参数传递给任何查询的方式保持一致，最好（我的意见）使用?占位符：

def get_ID(name):
    query = "SELECT ID FROM names WHERE name = ?"
    local_cursor = conn.cursor()
    local_cursor.execute(query, (name,))
    id = None
    if local_cursor.rowcount > 0:
        id = local_cursor.fetchone()[0]
    local_cursor.close()
    print(id)
    return id

赞(0）回复(0）举报 2023-03-23

我来回答

sqlite：在自定义函数(sqlite3.Connection.create_function)中使用另一个sqlite命令时出错

1条答案

相关问题

热门标签

最新问答