PedigreeDatabase/schemas.py at main · angelneychev/PedigreeDatabase · GitHub

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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
from typing import Optional, List
from pydantic import BaseModel, Field
from datetime import date, datetime

class HealthTestTypeBase(BaseModel):
    name: str
    description: Optional[str] = None
    valid_results: str  # JSON string

class HealthTestTypeCreate(HealthTestTypeBase):
    pass

class HealthTestType(HealthTestTypeBase):
    id: int

    class Config:
        from_attributes = True

class HealthTestBase(BaseModel):
    test_type_id: int
    test_date: date
    place: Optional[str] = None
    result: str
    notes: Optional[str] = None

class HealthTestCreate(HealthTestBase):
    pass

class HealthTest(HealthTestBase):
    id: int
    dog_id: int
    test_type: Optional[HealthTestType] = None
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True

class DogBase(BaseModel):
    name: str
    registration_number: Optional[str] = None
    sex: str = Field(..., pattern="^(Male|Female)$")
    date_of_birth: Optional[date] = None
    color: Optional[str] = None
    breed: str
    kennel_name: Optional[str] = None
    tatoo_no: Optional[str] = None
    microchip: Optional[str] = None
    breeder: Optional[str] = None
    sire_id: Optional[int] = None
    dam_id: Optional[int] = None

class DogCreate(DogBase):
    pass

class DogUpdate(BaseModel):
    name: Optional[str] = None
    registration_number: Optional[str] = None
    sex: Optional[str] = None
    date_of_birth: Optional[date] = None
    color: Optional[str] = None
    breed: Optional[str] = None
    kennel_name: Optional[str] = None
    tatoo_no: Optional[str] = None
    microchip: Optional[str] = None
    breeder: Optional[str] = None
    sire_id: Optional[int] = None
    dam_id: Optional[int] = None

class DogSimple(BaseModel):
    id: int
    name: str
    registration_number: Optional[str] = None
    sex: str
    date_of_birth: Optional[date] = None
    breed: str

    class Config:
        from_attributes = True

class Dog(DogBase):
    id: int
    sire: Optional[DogSimple] = None
    dam: Optional[DogSimple] = None
    health_tests: List[HealthTest] = []
    created_at: datetime
    updated_at: datetime

    class Config:
        from_attributes = True

class DogPedigree(DogSimple):
    sire: Optional['DogPedigree'] = None
    dam: Optional['DogPedigree'] = None

    class Config:
        from_attributes = True

# Enable forward references
DogPedigree.model_rebuild()