diff --git a/selfprivacy_api/jobs/nix_collect_garbage.py b/selfprivacy_api/jobs/nix_collect_garbage.py index 850764d..8fa40bf 100644 --- a/selfprivacy_api/jobs/nix_collect_garbage.py +++ b/selfprivacy_api/jobs/nix_collect_garbage.py @@ -1,12 +1,11 @@ import re import subprocess +from selfprivacy_api.utils.huey import huey + from selfprivacy_api.jobs import JobStatus, Jobs, Job -# from selfprivacy_api.utils.huey import huey - - COMPLETED_WITH_ERROR = "Completed with an error" RESULT_WAS_NOT_FOUND_ERROR = "We are sorry, result was not found :(" CLEAR_COMPLETED = "Сleaning completed." @@ -83,6 +82,7 @@ def get_dead_packages(output): return dead, percent +@huey.task() def calculate_and_clear_dead_packages(job: Jobs): Jobs.update( job=job, diff --git a/selfprivacy_api/task_registry.py b/selfprivacy_api/task_registry.py index f9f92e1..a4eb212 100644 --- a/selfprivacy_api/task_registry.py +++ b/selfprivacy_api/task_registry.py @@ -2,4 +2,4 @@ from selfprivacy_api.utils.huey import huey from selfprivacy_api.jobs.test import test_job from selfprivacy_api.restic_controller.tasks import * from selfprivacy_api.services.generic_service_mover import move_service -from selfprivacy_api.jobs.nix_collect_garbage import nix_collect_garbage +from selfprivacy_api.jobs.nix_collect_garbage import calculate_and_clear_dead_packages