diff --git a/apps/resm/tasks.py b/apps/resm/tasks.py index 9a39e8a..83d2b51 100644 --- a/apps/resm/tasks.py +++ b/apps/resm/tasks.py @@ -158,7 +158,8 @@ def get_abstract_from_elsevier(publication_year: int = None, number_of_task:int elif res.status_code == 404: paper.fail_reason = "elsevier_doi_not_found" paper.save(update_fields=["fail_reason", "update_time"]) - if is_elsevier_abstract_task_enabled(): + qs_count = qs.count() + if is_elsevier_abstract_task_enabled() and qs_count > 0: current_app.send_task( "apps.resm.tasks.get_abstract_from_elsevier", kwargs={ @@ -167,7 +168,7 @@ def get_abstract_from_elsevier(publication_year: int = None, number_of_task:int }, countdown=5, ) - return f'{err_msg}, remaining {qs.count()} papers' + return f'{err_msg}, remaining {qs_count} papers' def is_elsevier_pdf_task_enabled(): @@ -204,7 +205,8 @@ def get_pdf_from_elsevier(number_of_task=100): paper.save_file_pdf(res.content) paper.has_fulltext_pdf = True paper.save(update_fields=["has_fulltext_pdf", "update_time"]) - if is_elsevier_pdf_task_enabled(): + qs_count = qs.count() + if is_elsevier_pdf_task_enabled() and qs_count > 0: current_app.send_task( "apps.resm.tasks.get_pdf_from_elsevier", kwargs={ @@ -212,5 +214,5 @@ def get_pdf_from_elsevier(number_of_task=100): }, countdown=5, ) - return f'{err_msg}, remaining {qs.count()} papers' + return f'{err_msg}, remaining {qs_count} papers' \ No newline at end of file