mirror of
https://git.selfprivacy.org/SelfPrivacy/selfprivacy-rest-api.git
synced 2024-09-11 20:27:19 +00:00
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
|
from selfprivacy_api.migrations.migration import Migration
|
||
|
from selfprivacy_api.jobs import JobStatus, Jobs
|
||
|
|
||
|
|
||
|
class CheckForSystemRebuildJobs(Migration):
|
||
|
"""Check if there are unfinished system rebuild jobs and finish them"""
|
||
|
|
||
|
def get_migration_name(self):
|
||
|
return "check_for_system_rebuild_jobs"
|
||
|
|
||
|
def get_migration_description(self):
|
||
|
return "Check if there are unfinished system rebuild jobs and finish them"
|
||
|
|
||
|
def is_migration_needed(self):
|
||
|
# Check if there are any unfinished system rebuild jobs
|
||
|
for job in Jobs.get_jobs():
|
||
|
if (
|
||
|
job.type_id
|
||
|
in [
|
||
|
"system.nixos.rebuild",
|
||
|
"system.nixos.upgrade",
|
||
|
]
|
||
|
) and job.status in [
|
||
|
JobStatus.CREATED,
|
||
|
JobStatus.RUNNING,
|
||
|
]:
|
||
|
return True
|
||
|
|
||
|
def migrate(self):
|
||
|
# As the API is restarted, we assume that the jobs are finished
|
||
|
for job in Jobs.get_jobs():
|
||
|
if (
|
||
|
job.type_id
|
||
|
in [
|
||
|
"system.nixos.rebuild",
|
||
|
"system.nixos.upgrade",
|
||
|
]
|
||
|
) and job.status in [
|
||
|
JobStatus.CREATED,
|
||
|
JobStatus.RUNNING,
|
||
|
]:
|
||
|
Jobs.update(
|
||
|
job=job,
|
||
|
status=JobStatus.FINISHED,
|
||
|
result="System rebuilt.",
|
||
|
progress=100,
|
||
|
)
|