import uuid
from datetime import datetime

from pydantic import BaseModel, ConfigDict

from app.models.enums import ParentChildQualifier, RelationshipType


class RelationshipCreate(BaseModel):
    type: RelationshipType
    person_from_id: uuid.UUID
    person_to_id: uuid.UUID
    # Only meaningful for parent_child edges (from = parent, to = child).
    qualifier: ParentChildQualifier | None = None
    notes: str | None = None


class RelationshipRead(BaseModel):
    model_config = ConfigDict(from_attributes=True)

    id: uuid.UUID
    tree_id: uuid.UUID
    type: RelationshipType
    person_from_id: uuid.UUID
    person_to_id: uuid.UUID
    qualifier: ParentChildQualifier | None
    notes: str | None
    created_at: datetime