Spaces:
Sleeping
Sleeping
File size: 1,053 Bytes
3a7eb07 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 | """
Base Model Classes
Mixins and base classes for all SQLAlchemy models
"""
import uuid
from datetime import datetime
from sqlalchemy import Column, DateTime
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.ext.declarative import declarative_base, declared_attr
Base = declarative_base()
class UUIDMixin:
"""Mixin that adds a UUID primary key"""
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4, index=True)
class TimestampMixin:
"""Mixin that adds created_at and updated_at timestamps"""
created_at = Column(DateTime, default=datetime.utcnow, nullable=False)
updated_at = Column(
DateTime, default=datetime.utcnow, onupdate=datetime.utcnow, nullable=True
)
class TableNameMixin:
"""Mixin that auto-generates table name from class name"""
@declared_attr
def __tablename__(cls):
# Convert CamelCase to snake_case
name = cls.__name__
return "".join(["_" + c.lower() if c.isupper() else c for c in name]).lstrip(
"_"
)
|