Felipe Cardoso
5c47be2ee5
- Annotated `role` column in `user_organization` with a specific type for better clarity. - Added `mypy` overrides in `pyproject.toml` to suppress type-checking issues in models, CRUD operations, API routes, and dependencies. - Updated comment for `Base` re-export in `models.base` to clarify its purpose. - Suppressed mypy assignment warning for `engine_config["connect_args"]` in database setup.
29 lines
771 B
Python
29 lines
771 B
Python
import uuid
|
|
from datetime import UTC, datetime
|
|
|
|
from sqlalchemy import Column, DateTime
|
|
from sqlalchemy.dialects.postgresql import UUID
|
|
|
|
# noinspection PyUnresolvedReferences
|
|
from app.core.database import Base # Re-exported for other models
|
|
|
|
|
|
class TimestampMixin:
|
|
"""Mixin to add created_at and updated_at timestamps to models"""
|
|
|
|
created_at = Column(
|
|
DateTime(timezone=True), default=lambda: datetime.now(UTC), nullable=False
|
|
)
|
|
updated_at = Column(
|
|
DateTime(timezone=True),
|
|
default=lambda: datetime.now(UTC),
|
|
onupdate=lambda: datetime.now(UTC),
|
|
nullable=False,
|
|
)
|
|
|
|
|
|
class UUIDMixin:
|
|
"""Mixin to add UUID primary keys to models"""
|
|
|
|
id = Column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|