def documents(self, project: Union[Project, int]) -> List[Document]:
project_id = self._get_object_id(project)
response = self.client.get(f'/projects/{project_id}/documents',
allowed_statuses=(200, ))
document_list = DocumentSchema().load(response.json()['body'],
many=True)
for document in document_list:
document.project_id = project_id
return document_list
def create_document(self, project_id: int, document_name: str, content: IO,
document_format: str = InceptionFormat.DEFAULT, document_state: str = DocumentState.DEFAULT):
response = self.client.post(f"/projects/{project_id}/documents", form_data={"name": document_name,
"format": document_format,
"state": document_state},
files={"content": ('test/path', content)},
allowed_statuses=(201, 200))
document = DocumentSchema().load(response.json()['body'], many=False)
document.project_id = project_id
return document
def document_schema():
return DocumentSchema()