medical card create base

src/medical_info/db/models.py

@@ -6,7 +6,8 @@ import sqlalchemy as sa
 from sqlalchemy.dialects.postgresql import UUID, ARRAY, JSONB
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import relationship, DeclarativeBase
-from sqlalchemy.sql.schema import ForeignKey
+from sqlalchemy.sql.schema import ForeignKey, UniqueConstraint
+
 
 class Base(DeclarativeBase):
   pass
@@ -26,12 +27,14 @@ class MedicalCard(BaseModel):
   health_maps = relationship('HealthMap', back_populates='medical_card')
   confidant_persons = sa.Column(ARRAY(UUID(as_uuid=True)))
   person_id = sa.Column(UUID(as_uuid=True), nullable=False)
-  
+
+  __table_args__ = (UniqueConstraint("person_id", name="person_id"),)
+    
 
 class Checkup(BaseModel):
   __tablename__ = 'checkups'
 
-  medical_card_id = sa.Column(UUID(as_uuid=True), sa.ForeignKey("medical_cards.id"), nullable=False)
+  medical_card_id = sa.Column(UUID(as_uuid=True), ForeignKey("medical_cards.id"), nullable=False)
   medical_card = relationship('MedicalCard', back_populates='checkups')
   medic_id = sa.Column(UUID(as_uuid=True))
   claims = sa.Column(ARRAY(sa.String)) # жалобы
@@ -49,5 +52,5 @@ class HealthMap(BaseModel):
 
   comment = sa.Column(sa.Text)
   attachments = sa.Column(ARRAY(sa.String))
-  medical_card_id = sa.Column(UUID(as_uuid=True), sa.ForeignKey("medical_cards.id"), nullable=False)
+  medical_card_id = sa.Column(UUID(as_uuid=True), ForeignKey("medical_cards.id"), nullable=False)
   medical_card = relationship('MedicalCard', back_populates='health_maps')

src/medical_info/db/schemas/medical_card.py

@@ -1,11 +1,14 @@
 from pydantic import BaseModel
+from pydantic.types import UUID4
 
 class MedicalCard(BaseModel):
   number: str
-  person_id: str
+  person_id: UUID4
 
   class Config:
     orm_mode = True
 
 class MedicalCardCreate(BaseModel):
-  person_id: str
+  number: str
+  person_id: UUID4
+  

src/medical_info/routers/medical_cards.py

@@ -3,7 +3,7 @@ from fastapi.routing import APIRouter
 from typing import List
 
 from medical_info.db import models
-from medical_info.db.schemas import MedicalCard
+from medical_info.db.schemas import MedicalCard, MedicalCardCreate
 from medical_info.services import MedicalCardService, get_medical_cards_service
 
 router = APIRouter(prefix='/medical_cards')
@@ -14,3 +14,8 @@ def list_medical_cards(
 ) -> List[models.MedicalCard]:
   return medical_cards_service.list()
 
+@router.post('/', status_code=201, response_model=MedicalCard)
+async def create_medical_card(
+  medical_card: MedicalCardCreate, medical_cards_service: MedicalCardService = Depends(get_medical_cards_service)
+) -> models.MedicalCard:
+  return medical_cards_service.create(medical_card)