Prisma-Client-Py 中如何导入枚举值

在Python中使用Prisma ORM时，处理数据库模型中的枚举类型是一个常见的需求。与Prisma官方TypeScript版本不同，prisma-client-py对枚举值的导入方式进行了特定的设计，开发者需要了解其生成规则才能正确使用。

枚举类型在Prisma中的定义

在Prisma的schema文件中，我们可以定义枚举类型：

enum Role {
  ADMIN
  GUEST
}

model Member {
  id   String @id @default(uuid())
  role Role   @default(GUEST)
  name String
}

Python中的导入方式

prisma-client-py会将所有枚举类型生成到专门的prisma.enums模块中。要使用这些枚举值，正确的导入方式是：

from prisma.enums import Role

这样导入后，就可以像普通Python枚举一样使用这些值：

print(Role.ADMIN)  # 输出: Role.ADMIN
print(Role.GUEST)  # 输出: Role.GUEST

使用场景示例

在实际应用中，这些枚举值通常用于创建或查询记录：

from prisma import Prisma
from prisma.enums import Role

prisma = Prisma()
await prisma.connect()

# 创建使用枚举值的记录
await prisma.member.create(
    data={
        'name': 'Alice',
        'role': Role.ADMIN
    }
)

# 查询使用枚举值的记录
admins = await prisma.member.find_many(
    where={
        'role': Role.ADMIN
    }
)

注意事项

1. 枚举值的比较应该使用枚举成员本身，而不是字符串
2. 所有枚举类型都会被集中生成到prisma.enums模块中
3. 每次修改schema后需要重新生成客户端代码，枚举值才会更新

通过这种方式，prisma-client-py提供了类型安全的枚举使用体验，既保持了Python的简洁性，又确保了类型系统的严谨性。