Peewee：使用指定随机种子获取随机记录

我明白你想在测试里复现Peewee随机查询结果的需求，之前直接用fn.setseed()报错是因为不同数据库的随机种子设置方式不一样，而且你之前的写法也没正确执行种子设置的操作，下面给你几个靠谱的解决方案：

1. 针对不同数据库执行原生种子设置SQL #

Peewee的fn.setseed()不是通用的，因为不同数据库的种子函数/语句差异很大，你需要先单独执行对应数据库的种子设置命令，再执行随机查询：

对于SQLite #

SQLite没有setseed函数，而是用PRAGMA random_seed来设置，注意它接受的是字节数据：

# 设置随机种子（这里用字节串作为种子，可替换成你需要的固定值）
db.execute_sql("PRAGMA random_seed = ?", (b'fixed_seed_123',))
# 之后执行随机查询就能复现结果
random_records = Table.select().order_by(fn.Random()).limit(10)

对于PostgreSQL #

PostgreSQL支持setseed()函数，参数是0到1之间的浮点数，先执行种子设置再查询：

# 设置种子值（固定值保证复现）
db.execute_sql("SELECT setseed(%s)", (0.75,))
# 执行随机查询
random_records = Table.select().order_by(fn.Random()).limit(10)

对于MySQL #

MySQL需要设置rand_seed1和rand_seed2两个系统变量：

# 设置固定种子
db.execute_sql("SET @@rand_seed1 = 1234; SET @@rand_seed2 = 5678;")
# 用rand()排序获取随机记录
random_records = Table.select().order_by(fn.Rand()).limit(10)

2. 跨数据库兼容的Python原生随机方案（优化版） #

你之前想到的用Pythonrandom模块选ID的思路很稳妥，这里给你优化一下，让它更高效：

import random

# 固定种子保证复现
random.seed(42)

# 先只查询所有ID，避免加载全表数据
all_ids = [row.id for row in Table.select(Table.id)]
# 随机选择需要的ID数量
selected_ids = random.sample(all_ids, 10)
# 根据选中的ID查询完整记录
random_records = Table.select().where(Table.id.in_(selected_ids))

这个方法不依赖数据库特性，在任何数据库上都能稳定复现结果，而且只查ID的开销比全表查询小很多。

为什么你之前的写法会报错？ #

你用Table.select(fn.setseed(1)).order_by(fn.Random())的时候，Peewee会把setseed(1)作为查询的一个字段来处理，而不是先执行这个函数设置种子。另外，如果你的数据库（比如SQLite）本身没有setseed这个函数，自然会抛出no such function的错误。正确的做法是先单独执行种子设置的SQL，再执行随机查询。

内容的提问来源于stack exchange，提问作者Bub Espinja