Spaces:
Running
Running
File size: 1,393 Bytes
09801ca | 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 43 44 45 46 47 48 49 50 51 52 53 54 55 56 | """
Database Model Base Classes
Provides the declarative base and reusable mixins for all ORM models.
"""
import uuid
from datetime import datetime
from typing import Optional
from sqlalchemy import DateTime, Boolean, func
from sqlalchemy.dialects.postgresql import UUID
from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
class Base(DeclarativeBase):
"""Declarative base for all ORM models."""
pass
class TimestampMixin:
"""Adds created_at and updated_at columns to any model."""
created_at: Mapped[datetime] = mapped_column(
DateTime(timezone=True),
server_default=func.now(),
nullable=False,
)
updated_at: Mapped[datetime] = mapped_column(
DateTime(timezone=True),
server_default=func.now(),
onupdate=func.now(),
nullable=False,
)
class SoftDeleteMixin:
"""Adds soft delete capability to any model."""
is_deleted: Mapped[bool] = mapped_column(
Boolean, default=False, index=True, nullable=False
)
deleted_at: Mapped[Optional[datetime]] = mapped_column(
DateTime(timezone=True), nullable=True
)
class UUIDPrimaryKeyMixin:
"""UUID primary key mixin — prevents enumeration attacks."""
id: Mapped[uuid.UUID] = mapped_column(
UUID(as_uuid=True),
primary_key=True,
default=uuid.uuid4,
nullable=False,
)
|