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