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: