peewee model 使用枚举
时间: 2023-10-31 19:57:27 浏览: 164
枚举类型的使用
在 Peewee 模型中使用枚举可以通过 `EnumField` 字段来实现。下面是一个示例:
```python
from enum import Enum
from peewee import *
# 定义一个枚举类
class Gender(Enum):
MALE = 'Male'
FEMALE = 'Female'
# 定义 Peewee 模型
class Person(Model):
name = CharField()
gender = EnumField(choices=Gender)
class Meta:
database = db
# 创建数据库连接
db = SqliteDatabase('my_database.db')
# 创建表格
db.create_tables([Person])
# 插入数据
person = Person(name='John', gender=Gender.MALE)
person.save()
# 查询数据
male_persons = Person.select().where(Person.gender == Gender.MALE)
for person in male_persons:
print(person.name, person.gender)
```
在上述示例中,我们定义了一个名为 `Gender` 的枚举类,并在 `Person` 模型中使用了 `EnumField` 字段来存储性别信息。可以通过 `choices` 参数指定枚举类的选项。在插入和查询数据时,可以直接使用枚举类中的值进行操作。
阅读全文