def create_dummy_documents(files, request_type, request, app=None): """Helper method to create dummy documents for a request. """ if request_type == 'Registration': current_time = datetime.datetime.now().strftime('%Y%m%d%H%M%S') for file in files: document = Documents.get_document_by_name(file.get('label'), 1) reg_doc = RegDocuments(filename='{0}_{1}'.format(current_time, file.get('file_name'))) reg_doc.reg_details_id = request.id reg_doc.document_id = document.id reg_doc.save() file_path = '{0}/{1}/{2}'.format(app.config['DRS_UPLOADS'], request.tracking_id, file.get('file_name')) if not os.path.exists(os.path.dirname(file_path)): os.makedirs(os.path.dirname(file_path)) with open(file_path, 'wb') as f: f.seek(1073741824-1) f.write(b"\0") else: current_time = datetime.datetime.now().strftime('%Y%m%d%H%M%S') for file in files: document = Documents.get_document_by_name(file.get('label'), 2) dereg_doc = DeRegDocuments(filename='{0}_{1}'.format(current_time, file.get('file_name'))) dereg_doc.dereg_id = request.id dereg_doc.document_id = document.id dereg_doc.save() file_path = '{0}/{1}/{2}'.format(app.config['DRS_UPLOADS'], request.tracking_id, file.get('file_name')) if not os.path.exists(os.path.dirname(file_path)): os.makedirs(os.path.dirname(file_path)) with open(file_path, 'wb') as f: f.seek(1073741824-1) f.write(b"\0") return request
def test_get_document_by_name(session): """Verify that the get_document_by_name() returns a document provided the name.""" docs = [ Documents(id=11101, label='shp doc', type=1, required=True), Documents(id=21102, label='ath doc', type=2, required=True) ] session.bulk_save_objects(docs) session.commit() assert Documents.get_document_by_name('shp doc', 1) assert Documents.get_document_by_name('ath doc', 2) assert Documents.get_document_by_name('88668', 1) is None
def bulk_create(cls, documents, dereg_details, time): """Create/save documents in bulk.""" created_documents = [] for name in documents: document = Documents.get_document_by_name(name, 2) dereg_document = DeRegDocuments( filename='{0}_{1}'.format(time, documents[name].filename)) dereg_document.dereg_id = dereg_details.id dereg_document.document_id = document.id dereg_document.save() created_documents.append(dereg_document) return created_documents
def bulk_create(cls, documents, reg_details, time): """Create documents for a request in bulk.""" try: created_documents = [] for name in documents: document = Documents.get_document_by_name(name, 1) reg_document = cls( filename='{0}_{1}'.format(time, documents[name].filename)) reg_document.reg_details_id = reg_details.id reg_document.document_id = document.id reg_document.save() created_documents.append(reg_document) return created_documents except Exception: raise Exception
def bulk_update(cls, documents, dereg_details, time): """Update the documents in bulk.""" try: updated_documents = [] for name in documents: document = Documents.get_document_by_name(name, 2) cls.trash_document(document.id, dereg_details) dereg_document = cls( filename='{0}_{1}'.format(time, documents[name].filename)) dereg_document.dereg_id = dereg_details.id dereg_document.document_id = document.id dereg_document.save() updated_documents.append(dereg_document) return updated_documents except Exception: raise Exception
def bulk_update(cls, documents, reg_details, time): """Update documents of a request in bulk.""" try: updated_documents = [] for name in documents: document = Documents.get_document_by_name(name, 1) cls.trash_document(document.id, reg_details) reg_document = cls( filename='{0}_{1}'.format(time, documents[name].filename)) reg_document.reg_details_id = reg_details.id reg_document.document_id = document.id reg_document.save() updated_documents.append(reg_document) return updated_documents except Exception: raise Exception