mirror of
https://github.com/creyD/creyPY.git
synced 2026-04-12 19:30:30 +02:00
fix: added name for primary key constraint
This commit is contained in:
@@ -1,7 +1,7 @@
|
|||||||
import uuid
|
import uuid
|
||||||
from datetime import datetime
|
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.dialects.postgresql import UUID
|
||||||
from sqlalchemy.ext.declarative import declared_attr
|
from sqlalchemy.ext.declarative import declared_attr
|
||||||
from sqlalchemy.orm import as_declarative
|
from sqlalchemy.orm import as_declarative
|
||||||
@@ -23,6 +23,11 @@ class Base(AutoAnnotateMixin, AutoInitMixin):
|
|||||||
|
|
||||||
# TODO: Add automated foreign key resolution
|
# 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
|
# Generate __tablename__ automatically
|
||||||
@declared_attr
|
@declared_attr
|
||||||
def __tablename__(cls) -> str:
|
def __tablename__(cls) -> str:
|
||||||
|
|||||||
Reference in New Issue
Block a user