数据库连接对象还提供了一个 create_function(name, num_params, func) 方法,该方法用于注册一个自定义函数,接下来程序就可以在 SQL 语句中使用该自定义函数:
create_function 方法包含 3 个参数:
name 参数:指定注册的自定义函数的名字。
num_params:指定自定义函数所需参数的个数。
func:指定自定义函数对应的函数。
下面程序使用 create_function() 方法为 SQL 语句注册一个自定义函数,然后程序就可以在 SQL 语句中使用该自定义函数:
# 导入访问SQLite的模块 import sqlite3 # 先定义一个普通函数,准备注册为SQL中的自定义函数 def reverse_ext(st): # 对字符串反转,前后加方括号 return '[' + st[::-1] + ']' # ①、打开或创建数据库 # 也可以使用特殊名::memory:代表创建内存中的数据库 conn = sqlite3.connect('first.db') # 调用create_function注册自定义函数:enc conn.create_function('enc', 1, reverse_ext) # ②、获取游标 c = conn.cursor() # ③、在SQL语句中使用enc自定义函数 c.execute('insert into user_tb values(null, ?, enc(?), ?)', ('贾宝玉', '123456', 'male')) conn.commit() # ④、关闭游标 c.close() # ⑤、关闭连接 conn.close()
上面程序中,第 12 行代码将 reverse_ext() 函数注册为自定义函数 enc,该函数用于模拟一个简单的加密功能:程序会对字符串反转,并在字符串前后添加方括号。
提示:此时使用的加密功能只是一个简单的模拟,如果需要真正对密码进行加密,则建议使用更高强度的加密算法,比如加盐 MD5 加密。
程序中第 16 行代码在执行的 SQL 语句中使用了 enc 自定义函数,该自定义函数用于对插入的密码进行加密。因此,当使用上面程序插入数据时,程序会自动对插入的密码进行加密;程序会对密码进行反转,并在密码前后添加方括号。