Implementing Contract creation on draft page

api/rpk-api/firm/contract/schemas.py

@@ -1,13 +1,14 @@
 import datetime
 from typing import List
 
+from beanie import PydanticObjectId
 from pydantic import BaseModel, Field
 
 from firm.contract.models import ContractDraft, DraftProvision, DraftParty, Contract
 
 from firm.entity.models import Entity
 from firm.core.schemas import Writer, Reader
-from firm.core.models import DictionaryEntry
+from firm.core.models import DictionaryEntry, ForeignKey
 
 
 class ContractDraftRead(Reader, ContractDraft):
@@ -68,7 +69,7 @@ class ContractRead(Reader, Contract):
 class ContractCreate(Writer):
     date: datetime.date
     location: str
-    draft_id: str
+    draft_id: PydanticObjectId = ForeignKey(resource="contracts/drafts", schema="ContractDraft")
 
 class ContractInit(BaseModel):
     date: datetime.date