def import_document(self, values: Dict[str, Any]):
doc = Document()
doc.name = values['name']
doc.description = values['description']
doc.source = values['source']
doc.source_type = values['source_type']
doc.paragraphs = values['paragraphs']
doc.sentences = values['sentences']
doc.title = values['title']
doc.document_type_id = self.document_types[str(
values['document_type_id'])]
doc.project_id = self.project_ids[values['project_id']]
doc.status_id = str(values['status_id'])
doc.language = values['language']
doc.file_size = values['file_size']
if not pd.isnull(values['assign_date']):
doc.assign_date = values['assign_date']
doc.delete_pending = values['delete_pending'] == 't'
doc.processed = values['processed'] == 't'
doc.folder = values['folder']
doc.document_class = values['document_class']
doc.fields_dirty = values['fields_dirty']
if not pd.isnull(values['assignee_id']):
doc.assignee = self.target_user
doc.source_path = values['source_path']
doc.save()
self.document_ids[values['id']] = doc.pk
self.document_src_paths[doc.pk] = doc.source_path
self.initially_loaded_docs.append(doc.pk)
def make_documents(cls, doc_ids: List[int]):
texts = TEST_TEXTS
docs: List[Document] = []
text_index = 0
for id in doc_ids:
doc = Document()
doc.pk = id
doc.project_id = 1
doc.documenttext = DocumentText()
doc.documenttext.full_text = texts[text_index]
text_index += 1
docs.append(doc)
return docs
