From 5b74ed562076d4ce074e70425763f6e4f8747cef Mon Sep 17 00:00:00 2001 From: Conrad Date: Thu, 24 Jul 2025 23:10:46 +0200 Subject: [PATCH] fix: added name for primary key constraint --- creyPY/fastapi/models/base.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/creyPY/fastapi/models/base.py b/creyPY/fastapi/models/base.py index 0a5109f..a6ee621 100644 --- a/creyPY/fastapi/models/base.py +++ b/creyPY/fastapi/models/base.py @@ -1,7 +1,7 @@ import uuid from datetime import datetime -from sqlalchemy import Column, DateTime, String +from sqlalchemy import Column, DateTime, PrimaryKeyConstraint, String from sqlalchemy.dialects.postgresql import UUID from sqlalchemy.ext.declarative import declared_attr from sqlalchemy.orm import as_declarative @@ -23,6 +23,11 @@ class Base(AutoAnnotateMixin, AutoInitMixin): # TODO: Add automated foreign key resolution + # Add name to primary key constraint to ensure alembic can pick it up later + @declared_attr + def __table_args__(cls): + return (PrimaryKeyConstraint("id", name=f"pk_{cls.__tablename__}"),) + # Generate __tablename__ automatically @declared_attr def __tablename__(cls) -> str: