Full Working static multi tenant

back/app/contract/schemas.py

@@ -6,11 +6,11 @@ from pydantic import BaseModel, Field
 from .models import ContractDraft, DraftProvision, DraftParty, Contract
 
 from ..entity.models import Entity
-from ..core.schemas import Writer
+from ..core.schemas import Writer, Reader
 from ..core.models import DictionaryEntry
 
 
-class ContractDraftRead(ContractDraft):
+class ContractDraftRead(Reader, ContractDraft):
     pass
 
 
@@ -28,12 +28,12 @@ class ContractDraftCreate(Writer):
         title='Variables'
     )
 
-    async def validate_foreign_key(self):
-        return
+    async def validate_foreign_key(self, db):
         for p in self.parties:
-            p.entity = await Entity.get(p.entity)
-            if p.entity is None:
-                raise ValueError
+            if p.entity_id:
+                p.entity = await Entity.get(db, p.entity_id)
+                if p.entity is None:
+                    raise ValueError
 
 
 class ContractDraftUpdate(ContractDraftCreate):
@@ -57,7 +57,7 @@ class PartyRead(BaseModel):
         title = "Partie"
 
 
-class ContractRead(Contract):
+class ContractRead(Reader, Contract):
     parties: List[PartyRead]
     lawyer: ForeignEntityRead
 
@@ -70,6 +70,14 @@ class ContractCreate(Writer):
     location: str
     draft_id: str
 
+class ContractInit(BaseModel):
+    date: datetime.date
+    location: str
+    lawyer: dict
+    name: str
+    title: str
+    parties: List[dict]
+    provisions: List[dict]
 
 class ContractUpdate(BaseModel):
     pass