Coverage for api/services/documents.py: 62%

1import config 

2from exceptions import DocGenerationFailed 

3from services.microservice_connector import DocumentsConnector 

4from sotrans_models.models.orders.order import OrderDBModel 

5 

6 

7class DocumentsQuerier: 

8 API_URL = config.DocumentsServiceConfig.DOCUMENTS_SERVICE_URL 

9 

10 def __init__(self, connector: DocumentsConnector): 

11 self.connector = connector 

12 

13 async def upload_document(self, order: OrderDBModel): 

14 file_path = await self.connector.make_request( 

15 self.API_URL, order.model_dump(format_ids=False, mode="json") 

16 ) 

17 if file_path: 

18 return file_path 

19 raise DocGenerationFailed 

20 

21 

22documents_querier = DocumentsQuerier(DocumentsConnector())