def test_prepare_update_payload_accepts_a_custom_key(): obj = MockObj({}, {}) eng = MockEng() custom_key_prepare_update_payload = prepare_update_payload('custom_key') assert custom_key_prepare_update_payload(obj, eng) is None assert obj.extra_data['custom_key'] == {}
def test_prepare_update_payload_overwrites(): obj = MockObj({'bar': 'baz'}, {'foo': 'foo'}) eng = MockEng() foo_prepare_update_payload = prepare_update_payload('foo') assert foo_prepare_update_payload(obj, eng) is None assert obj.extra_data['foo'] == {'bar': 'baz'}
def test_prepare_update_payload_overwrites(): obj = MockObj({'bar': 'baz'}, {'foo': 'foo'}) eng = MockEng() foo_prepare_update_payload = prepare_update_payload('foo') assert foo_prepare_update_payload(obj, eng) is None assert obj.extra_data['foo'] == {'bar': 'baz'}
def test_prepare_update_payload(): obj = MockObj({}, {}) eng = MockEng() default_prepare_update_payload = prepare_update_payload() assert default_prepare_update_payload(obj, eng) is None assert obj.extra_data['update_payload'] == {}
def test_prepare_update_payload_accepts_a_custom_key(): obj = MockObj({}, {}) eng = MockEng() custom_key_prepare_update_payload = prepare_update_payload('custom_key') assert custom_key_prepare_update_payload(obj, eng) is None assert obj.extra_data['custom_key'] == {}
def test_prepare_update_payload(): obj = MockObj({}, {}) eng = MockEng() default_prepare_update_payload = prepare_update_payload() assert default_prepare_update_payload(obj, eng) is None assert obj.extra_data['update_payload'] == {}
class Article(object): """Article ingestion workflow for Literature collection.""" name = "HEP" data_type = "hep" workflow = [ # Make sure schema is set for proper indexing in Holding Pen set_schema, # Emit record signals to receive metadata enrichment emit_record_signals, # Query locally or via legacy search API to see if article # is already ingested and this is an update IF(article_exists, [ mark('match-found', True), ]), IF_ELSE( is_submission, [ # Article matching for submissions # ================================ IF(pending_in_holding_pen, [ mark('already-in-holding-pen', True), ]), # Special RT integration for submissions # ====================================== create_ticket( template="literaturesuggest/tickets/curator_submitted.html", queue="HEP_add_user", context_factory=new_ticket_context, ticket_id_key="ticket_id"), reply_ticket( template="literaturesuggest/tickets/user_submitted.html", context_factory=reply_ticket_context, keep_new=True), ], [ # Article matching for non-submissions # ==================================== # Query holding pen to see if we already have this article ingested # # NOTE on updates: # If the same article has been harvested before and the # ingestion has been completed, process is continued # to allow for updates. IF(pending_in_holding_pen, [ mark('already-in-holding-pen', True), mark('delete', True), ]), IF( is_arxiv_paper, [ # FIXME: This filtering step should be removed when this # workflow includes arXiv CORE harvesting IF(already_harvested, [ mark('already-ingested', True), mark('stop', True), ]), # FIXME: This filtering step should be removed when: # old previously rejected records are treated # differently e.g. good auto-reject heuristics or better # time based filtering (5 days is quite random now). IF(previously_rejected(), [ mark('already-ingested', True), mark('stop', True), ]), ]), IF(is_marked('delete'), [update_old_object, delete_self_and_stop_processing]), IF(is_marked('stop'), [stop_processing]), ]), # # Article Processing # ================== IF(is_arxiv_paper, [ arxiv_fulltext_download, arxiv_plot_extract, arxiv_refextract, arxiv_author_list("authorlist2marcxml.xsl"), ]), extract_journal_info, classify_paper( taxonomy="HEPont.rdf", only_core_tags=False, spires=True, with_author_keywords=True, ), filter_core_keywords, guess_categories, IF(is_experimental_paper, [ guess_experiments, ]), guess_keywords, # Predict action for a generic HEP paper based only on title # and abstract. guess_coreness, # ("arxiv_skip_astro_title_abstract.pickle) # Check if we shall halt or auto-reject # ===================================== # NOTE: User submissions are always relevant IF_ELSE(is_record_relevant, [ halt_record(action="hep_approval"), ], [reject_record("Article automatically rejected"), stop_processing]), IF_ELSE(is_record_accepted, [ IF(article_exists, [ IF_ELSE(is_submission, [ reject_record('Article was already found on INSPIRE'), stop_processing, reply_ticket( template= "literaturesuggest/tickets/user_rejected_exists.html", context_factory=reply_ticket_context), close_ticket(ticket_id_key="ticket_id"), ], [ halt_record(action="merge_approval"), ]), ]), add_core, add_note_entry, filter_keywords, user_pdf_get, IF_ELSE(shall_push_remotely, [ IF_ELSE(article_exists, [ prepare_update_payload(extra_data_key="update_payload"), send_robotupload(marcxml_processor=hep2marc, mode="correct", extra_data_key="update_payload"), ], [ send_robotupload(marcxml_processor=hep2marc, mode="insert"), ]) ], [store_record]), IF(is_submission, [ IF(curation_ticket_needed, [ create_ticket( template="literaturesuggest/tickets/curation_core.html", queue="HEP_curation", context_factory=curation_ticket_context, ticket_id_key="curation_ticket_id") ]), reply_ticket( template="literaturesuggest/tickets/user_accepted.html", context_factory=reply_ticket_context), ]), ], [ IF(is_submission, [reply_ticket(context_factory=reply_ticket_context)]) ]), close_ticket(ticket_id_key="ticket_id") ]
], ), ] POSTENHANCE_RECORD = [ add_core, add_note_entry, filter_keywords, prepare_keywords, remove_references, prepare_files, ] SEND_TO_LEGACY_AND_WAIT = [ IF_ELSE(article_exists, [ prepare_update_payload(extra_data_key="update_payload"), send_robotupload(marcxml_processor=hep2marc, mode="correct", extra_data_key="update_payload"), ], [ send_robotupload(marcxml_processor=hep2marc, mode="insert"), wait_webcoll, ]), ] CHECK_IF_MERGE_AND_STOP_IF_SO = [ IF( article_exists, [ IF_ELSE( is_submission,
POSTENHANCE_RECORD = [ add_core, add_note_entry, filter_keywords, prepare_keywords, user_pdf_get, remove_references, prepare_files, ] SEND_TO_LEGACY_AND_WAIT = [ IF_ELSE( article_exists, [ prepare_update_payload(extra_data_key="update_payload"), send_robotupload( marcxml_processor=hep2marc, mode="correct", extra_data_key="update_payload" ), ], [ send_robotupload( marcxml_processor=hep2marc, mode="insert" ), wait_webcoll, ] ), ]