Adding a label to signature read entities

back/app/contract/__init__.py

@@ -7,7 +7,7 @@ from .routes_draft import draft_router
 from .print import print_router
 
 from .models import Contract, ContractDraft, ContractDraftStatus, Party, replace_variables_in_value
-from .schemas import ContractCreate, ContractRead, ContractUpdate
+from .schemas import ContractCreate, ContractRead, ContractUpdate, SignatureRead
 
 from ..entity.models import Entity
 from ..template.models import ProvisionTemplate
@@ -75,10 +75,12 @@ async def update(id: str, contract_form: ContractUpdate, user=Depends(get_curren
 
 
 @contract_router.get("/signature/{signature_id}", response_description="")
-async def get_signature(signature_id: str) -> Party:
+async def get_signature(signature_id: str) -> SignatureRead:
     contract = await Contract.find_by_signature_id(signature_id)
     signature = contract.get_signature(signature_id)
-    return signature
+    signature_dict = signature.dict()
+    signature_dict['contract_label'] = contract.label
+    return signature_dict
 
 
 @contract_router.post("/signature/{signature_id}", response_description="")

back/app/contract/schemas.py

@@ -73,3 +73,12 @@ class ContractCreate(Writer):
 
 class ContractUpdate(BaseModel):
     pass
+
+
+class SignatureRead(BaseModel):
+    signature_uuid: str
+    entity: Entity
+    part: str
+    representative: Entity = None
+    signature_affixed: bool
+    contract_label: str