"""Child database model."""

from typing import TYPE_CHECKING
from sqlalchemy import Column, Integer, String, DateTime, Float
from datetime import datetime

if TYPE_CHECKING:
    from sqlalchemy.orm import DeclarativeBase

    Base = DeclarativeBase
else:
    from baby_monitor.repositories.dependencies.get_database import Base


class Child(Base):
    """Child database model."""

    __tablename__ = "children"

    id = Column(Integer, primary_key=True, index=True)
    name = Column(String, nullable=False)
    birth_time = Column(DateTime(timezone=True), nullable=False)
    birth_weight = Column(Float, nullable=False)  # in grams
    created_at = Column(
        DateTime(timezone=True), default=datetime.utcnow, nullable=False
    )

    def __repr__(self) -> str:
        return f"<Child(id={self.id}, name='{self.name}')>"