是否可以限制wtf-peewee中SelectModelField中的選擇

Question

我將燒瓶與長頸鹿皮和wtfpeewee一起使用。
所以，我有這樣的模型：

class Category(Model):
    name = CharField()
    user = ForeignKeyField(User, null=True)

class Record(Model):
    value = DecimalField()
    category = ForeignKeyField(Category)
    user = ForeignKeyField(User)

當我創建供用戶添加記錄的表單時，我會這樣做：

RecordForm = model_form(Record)

數據庫的所有類別均可在此表單的“類別”字段中選擇，但我需要將“類別”字段的可用選項限制為用戶字段等於“無”或當前（已登錄）用戶的類別的子集。 我知道如何通過查詢來限制它，但是對於表單字段應該怎么做呢？

Answer 1

很抱歉剛才看到這個

您可以在類定義時執行此操作：

from wtfpeewee.fields import SelectQueryField

class MyForm(Form):
    category = SelectQueryField(query=Category.filter(some_val=other_val)

另外，我相信您可以在運行時執行此操作：

my_form = MyForm()
my_form.category.query = Category.filter(user=some_user)