zcdsj
/
paper_server
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
apps/resm/tasks.py
View File

@ -158,7 +158,8 @@ def get_abstract_from_elsevier(publication_year: int = None, number_of_task:int
elif res.status_code == 404: elif res.status_code == 404:
paper.fail_reason = "elsevier_doi_not_found" paper.fail_reason = "elsevier_doi_not_found"
paper.save(update_fields=["fail_reason", "update_time"]) 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( current_app.send_task(
"apps.resm.tasks.get_abstract_from_elsevier", "apps.resm.tasks.get_abstract_from_elsevier",
kwargs={ kwargs={
@ -167,7 +168,7 @@ def get_abstract_from_elsevier(publication_year: int = None, number_of_task:int
}, },
countdown=5, countdown=5,
) )
return f'{err_msg}, remaining {qs.count()} papers' return f'{err_msg}, remaining {qs_count} papers'
def is_elsevier_pdf_task_enabled(): 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.save_file_pdf(res.content)
paper.has_fulltext_pdf = True paper.has_fulltext_pdf = True
paper.save(update_fields=["has_fulltext_pdf", "update_time"]) 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( current_app.send_task(
"apps.resm.tasks.get_pdf_from_elsevier", "apps.resm.tasks.get_pdf_from_elsevier",
kwargs={ kwargs={
@ -212,5 +214,5 @@ def get_pdf_from_elsevier(number_of_task=100):
}, },
countdown=5, countdown=5,
) )
return f'{err_msg}, remaining {qs.count()} papers' return f'{err_msg}, remaining {qs_count} papers'