Python SQLite create_function方法:注册自定义函数

广告位

数据库连接对象还提供了一个 create_function(name, num_params, func) 方…

数据库连接对象还提供了一个 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 自定义函数,该自定义函数用于对插入的密码进行加密。因此,当使用上面程序插入数据时,程序会自动对插入的密码进行加密;程序会对密码进行反转,并在密码前后添加方括号。

关于作者: Python基础入门教程

为您推荐

广告位

发表评论