def test_get_document_by_id(session): """Verify that the get_document_by_id returns a document provided an id.""" docs = [ Documents(id=1110, label='shp doc', type=1, required=True), Documents(id=2110, label='ath doc', type=1, required=True) ] session.bulk_save_objects(docs) session.commit() assert Documents.get_document_by_id(1110) assert Documents.get_document_by_id(2110) assert Documents.get_document_by_id(79897777879) is None
def get_document_label(self, data): """Returns a document label.""" dereg_details = DeRegDetails.get_by_id(data.dereg_id) upload_dir_path = GLOBAL_CONF['upload_directory'] document = Documents.get_document_by_id(data.document_id) data.label = document.label data.required = Documents.required data.link = '{server_dir}/{local_dir}/{file_name}'.format( server_dir=upload_dir_path, local_dir=dereg_details.tracking_id, file_name=data.filename)