如何使用FastAPI、SQLalchemy和SQLModel生成UUID字段

Question

在FastAPI应用程序中创建模型时，我很难获得创建UUID字段的语法。我使用的是SQLModel。

所以基本上，我的models.py文件看起来是这样的：

from datetime import datetime
from typing import Optional
import uuid
from sqlalchemy import Column, DateTime
from sqlalchemy.dialects import postgresql as psql
from sqlmodel import SQLModel, Field

class ModelBase(SQLModel):
"""
Base class for database models.
"""
id: Optional[int] = Field(default=None, primary_key=True)
created_at: datetime = Field(sa_column=Column(DateTime(timezone=True), default=datetime.utcnow))
updated_at: datetime = Field(sa_column=Column(DateTime(timezone=True),
onupdate=datetime.utcnow, default=datetime.utcnow))

class UUIDModelBase(ModelBase, table=True):
"""
Base class for UUID-based models.
"""
uuid: uuid.UUID = Field(sa_column=Column(psql.UUID(as_uuid=True)), default=uuid.uuid4)

出现上述错误

AttributeError: 'FieldInfo' object has no attribute 'UUID'

我也试过

id: uuid.UUID = Column(psql.UUID(as_uuid=True), default=uuid.uuid4)
TypeError: Boolean value of this clause is not defined

还有

uuid: uuid.UUID = Column(psql.UUID(as_uuid=True), default=uuid.uuid4)
AttributeError: Neither 'Column' object nor 'Comparator' object has an attribute 'UUID'

和

uuid: uuid.UUID = Field(default_factory=uuid.uuid4, index=True, nullable=False)
AttributeError: 'FieldInfo' object has no attribute 'UUID'

你明白了。这些错误对我没有帮助，我只需要正确的语法。

在这种情况下，我实际上并不想使用UUID作为主键。从导入中可以看出，我使用的是postgreSQL。该数据库基于postgres:12docker图像。

Answer 1

解释器可能使用实际字段uuid的UUID，而不是导入的包。因此，您可以按如下方式更改代码。

import uuid as uuid_pkg
from sqlalchemy import Field
from sqlmodel import Field
class UUIDModelBase(ModelBase):
"""
Base class for UUID-based models.
"""
uuid: uuid_pkg.UUID = Field(
default_factory=uuid_pkg.uuid4,
primary_key=True,
index=True,
nullable=False,
)

参考：https://github.com/tiangolo/sqlmodel/issues/140