From 5b6e4ee59125b9e5634437f70aa788b138c633b5 Mon Sep 17 00:00:00 2001 From: caoqianming Date: Tue, 27 Jan 2026 17:35:38 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E8=87=AA=E5=90=AF=E5=8A=A8=E4=BB=BB?= =?UTF-8?q?=E5=8A=A1=E6=B7=BB=E5=8A=A0=E6=9D=A1=E4=BB=B6?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- apps/resm/tasks.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) 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