initial commit

back/app/contract/schemas.py

@@ -0,0 +1,36 @@
+import uuid
+from datetime import datetime
+from typing import List
+
+from pydantic import BaseModel, validator
+
+from beanie import PydanticObjectId
+
+from .models import Contract, ContractType, Clause
+from ..entity.models import Entity
+from ..core.schemas import Writer
+
+
+class ContractRead(Contract):
+    pass
+
+
+class PartyCreate(BaseModel):
+    entity: PydanticObjectId
+    part: str
+
+
+class ContractCreate(Writer):
+    type: ContractType
+    parties: List[PartyCreate]
+    clauses: List[Clause]
+
+    async def validate_foreign_key(self):
+        for p in self.parties:
+            p.entity = await Entity.get(p.entity)
+            if p.entity is None:
+                raise ValueError
+
+
+class ContractUpdate(BaseModel):
+    status: str