sqlmodel定义主键为uuid

在 SQLModel 中,如果你想将主键定义为 UUID 类型,你可以使用 Field 函数与 sqlalchemyColumn 函数来实现。首先,你需要从 uuid 模块导入 UUID,以及从 sqlalchemy 导入 Column 和其他必要的组件。

以下是一个定义 UUID 作为主键的 SQLModel 模型示例:

python 复制代码
from sqlmodel import SQLModel, Field
from sqlalchemy import Column, String, UUID
import uuid

class MyModel(SQLModel, table=True):
    # 使用 UUID 作为主键
    id: UUID = Field(default=lambda: uuid.uuid4(), primary_key=True)
    some_field: str = Field(sa_column=Column(String))

# 如果需要在没有默认值的情况下创建一个 UUID 主键的模型
class MyOtherModel(SQLModel, table=True):
    id: UUID = Field(primary_key=True)
    some_field: str = Field(sa_column=Column(String))

    def __init__(self, id: uuid.UUID = None, **data):
        if id is None:
            self.id = uuid.uuid4()
        super().__init__(id=id, **data)

在这个例子中:

  • MyModel 类定义了一个使用 UUID 作为主键的模型。Field 函数的 default 参数设置为 lambda: uuid.uuid4(),这意味着每次创建新实例时,如果没有提供 id,将自动生成一个新的 UUID。
  • MyOtherModel 类展示了如何在没有默认生成函数的情况下定义一个 UUID 主键。在这个模型中,你需要在创建实例时手动提供一个 UUID,或者在构造函数中设置一个 UUID。

请注意,SQLModel 建立在 Pydantic 模型之上,Pydantic 要求模型的字段必须有默认值或者通过构造函数显式设置。这就是为什么 MyOtherModel 中需要重写构造函数来处理 UUID 的初始化。

确保你的数据库支持 UUID 类型,并且 SQLAlchemy 的 dialect 正确配置以使用 UUID。大多数现代数据库(如 PostgreSQL)都支持 UUID 类型。

要使用这些模型,你需要创建一个 SQLAlchemy 引擎并连接到数据库,然后使用 SQLModel 的元数据来创建表:

python 复制代码
from sqlmodel import SQLModel, create_engine

SQLModel.metadata.create_all(create_engine('sqlite:///example.db'))

这将创建一个 SQLite 数据库(或你选择的其他数据库类型),并根据模型定义创建相应的表。

相关推荐
敖云岚30 分钟前
【Redis】分布式锁的介绍与演进之路
数据库·redis·分布式
LUCIAZZZ1 小时前
HikariCP数据库连接池原理解析
java·jvm·数据库·spring·springboot·线程池·连接池
我在北京coding1 小时前
300道GaussDB(WMS)题目及答案。
数据库·gaussdb
小Tomkk1 小时前
阿里云 RDS mysql 5.7 怎么 添加白名单 并链接数据库
数据库·mysql·阿里云
明月醉窗台2 小时前
qt使用笔记二:main.cpp详解
数据库·笔记·qt
沉到海底去吧Go3 小时前
【图片自动识别改名】识别图片中的文字并批量改名的工具,根据文字对图片批量改名,基于QT和腾讯OCR识别的实现方案
数据库·qt·ocr·图片识别自动改名·图片区域识别改名·pdf识别改名
老纪的技术唠嗑局3 小时前
重剑无锋,大巧不工 —— OceanBase 中的 Nest Loop Join 使用技巧分享
数据库·sql
未来之窗软件服务4 小时前
JAVASCRIPT 前端数据库-V6--仙盟数据库架构-—-—仙盟创梦IDE
数据库·数据库架构·仙盟创梦ide·东方仙盟·东方仙盟数据库
一只爱撸猫的程序猿5 小时前
构建一个简单的智能文档问答系统实例
数据库·spring boot·aigc
nanzhuhe5 小时前
sql中group by使用场景
数据库·sql·数据挖掘