83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
import uuid
|
|
from fastapi import Depends, HTTPException
|
|
|
|
from ..core.routes import get_crud_router
|
|
from .routes_draft import draft_router
|
|
from .print import print_router
|
|
|
|
from .models import Contract, ContractDraft
|
|
from .schemas import ContractCreate, ContractRead, ContractUpdate
|
|
|
|
from ..entity.models import Entity
|
|
from ..template.models import ProvisionTemplate
|
|
|
|
contract_router = get_crud_router(Contract, ContractCreate, ContractRead, ContractUpdate)
|
|
del(contract_router.routes[0])
|
|
del(contract_router.routes[2])
|
|
del(contract_router.routes[2])
|
|
|
|
contract_router.include_router(draft_router, prefix="/draft", tags=["draft"], )
|
|
contract_router.include_router(print_router, prefix="/print", tags=["print"], )
|
|
|
|
|
|
def can_create_contract():
|
|
class User:
|
|
entity_id = '63d127bcf355de8e65a193e1'
|
|
return User()
|
|
|
|
|
|
@contract_router.post("/", response_description="Contract Successfully created")
|
|
async def create(item: ContractCreate, user=Depends(can_create_contract)) -> dict:
|
|
await item.validate_foreign_key()
|
|
|
|
draft = await ContractDraft.get(item.draft_id)
|
|
|
|
for v in draft.variables:
|
|
if not v.key or not v.value:
|
|
raise HTTPException(status_code=400, detail="Variable {} is invalid".format(v))
|
|
|
|
contract_dict = item.dict()
|
|
del(contract_dict['draft_id'])
|
|
|
|
contract_dict['lawyer'] = await Entity.get(user.entity_id)
|
|
|
|
contract_dict['name'] = draft.name
|
|
contract_dict['title'] = draft.title
|
|
parties = []
|
|
for p in draft.parties:
|
|
parties.append({
|
|
'entity': await Entity.get(p.entity_id),
|
|
'part': p.part,
|
|
'representative': await Entity.get(p.representative_id) if p.representative_id else None,
|
|
'signature_uuid': str(uuid.uuid4())
|
|
})
|
|
|
|
contract_dict['parties'] = parties
|
|
|
|
provisions = []
|
|
for p in draft.provisions:
|
|
p = p.provision
|
|
provision = await ProvisionTemplate.get(p.provision_template_id) if p.type == 'template' \
|
|
else p
|
|
|
|
provisions.append({
|
|
'title': replace_variables_in_value(draft.variables, provision.title),
|
|
'body': replace_variables_in_value(draft.variables, provision.body)
|
|
})
|
|
|
|
contract_dict['provisions'] = provisions
|
|
|
|
o = await Contract(**contract_dict).create()
|
|
return {"message": "Contract Successfully created", "id": o.id}
|
|
|
|
|
|
@contract_router.put("/{id}", response_description="")
|
|
async def update(id: str, req: ContractUpdate) -> ContractRead:
|
|
raise HTTPException(status_code=400, detail="No modification on contract")
|
|
|
|
|
|
def replace_variables_in_value(variables, value: str):
|
|
for v in variables:
|
|
value = value.replace('%{}%'.format(v.key), v.value)
|
|
return value
|