33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
"""Diaper change database model."""
|
|
|
|
from typing import TYPE_CHECKING
|
|
from sqlalchemy import Column, Integer, DateTime, ForeignKey, String
|
|
from datetime import datetime, UTC
|
|
|
|
if TYPE_CHECKING:
|
|
from sqlalchemy.orm import DeclarativeBase
|
|
|
|
Base = DeclarativeBase
|
|
else:
|
|
from baby_monitor.repositories.dependencies.get_database import Base
|
|
|
|
|
|
class DiaperChange(Base):
|
|
"""Diaper change log database model."""
|
|
|
|
__tablename__ = "diaper_changes"
|
|
|
|
id = Column(Integer, primary_key=True, index=True)
|
|
child_id = Column(Integer, ForeignKey("children.id"), nullable=False)
|
|
change_time = Column(DateTime(timezone=True), nullable=False)
|
|
poop_amount = Column(String, nullable=True) # stores PoopAmount enum
|
|
poop_color = Column(String, nullable=True) # stores PoopColor enum
|
|
pee_amount = Column(String, nullable=True) # stores PeeAmount enum
|
|
pee_color = Column(String, nullable=True) # stores PeeColor enum
|
|
created_at = Column(
|
|
DateTime(timezone=True), default=datetime.now(UTC), nullable=False
|
|
)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"<DiaperChange(id={self.id}, child_id={self.child_id})>"
|