如何在Python中为Postgres自动生成UUID？

Question

我正在尝试在Postgres db中创建对象。

我正在使用这种方法https://websauna.org/docs/narrative/modelling/models.html#uuid-primary-keys

class Role(Base):
    __tablename__ = 'role'

    # Pass `binary=False` to fallback to CHAR instead of BINARY
    id = sa.Column(UUIDType(binary=False), primary_key=True)

但是当我创建对象时

user_role = Role(name='User')
db.session.add(user_role)
db.session.commit()

我有以下错误：

sqlalchemy.exc.IntegrityError: (psycopg2.IntegrityError) null value in column "id" violates not-null constraint

好像我没有提供任何ID。因此，如何使数据库自动生成或自行生成？

Answer 1

您似乎正在使用此代码。它缺少该列的默认值。您正在模拟此SQL：

id UUID PRIMARY KEY DEFAULT uuid_generate_v4()

但是您已经链接到正确的代码。

id = Column(UUID(as_uuid=True),
    primary_key=True,
    server_default=sqlalchemy.text("uuid_generate_v4()"),)

另外，如果您不想加载Postgres UUID扩展名，则可以在Python中创建UUID。

from uuid import uuid4

id = Column(UUID(as_uuid=True),
    primary_key=True,
    default=uuid4,)

Answer 2

您可以使用该uuid模块并只设置一个列默认值。例如：

from uuid import uuid4
from sqlalchemy import Column, String

class Role(Base):
    __tablename__ = 'role'

    id = Column(String, primary_key=True, default=uuid4)