feat: 自启动任务添加条件

This commit is contained in:
caoqianming 2026-01-27 17:35:38 +08:00
parent d1f32bd9b0
commit 5b6e4ee591
1 changed files with 6 additions and 4 deletions

View File

@ -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'