Skip to content

Commit 3f384cf

Browse files
committed
chore: move class for better git diff
1 parent 73689ba commit 3f384cf

File tree

1 file changed

+47
-47
lines changed

1 file changed

+47
-47
lines changed

ami/jobs/models.py

Lines changed: 47 additions & 47 deletions
Original file line numberDiff line numberDiff line change
@@ -411,6 +411,53 @@ def run(cls, job: "Job"):
411411
job.save()
412412

413413

414+
class DataStorageSyncJob(JobType):
415+
name = "Data storage sync"
416+
key = "data_storage_sync"
417+
418+
@classmethod
419+
def run(cls, job: "Job"):
420+
"""
421+
Run the data storage sync job.
422+
423+
This is meant to be called by an async task, not directly.
424+
"""
425+
426+
job.progress.add_stage(cls.name)
427+
job.progress.add_stage_param(cls.key, "Total files", "")
428+
job.update_status(JobState.STARTED)
429+
job.started_at = datetime.datetime.now()
430+
job.finished_at = None
431+
job.save()
432+
433+
if job.deployment:
434+
job.logger.info(f"Syncing captures for deployment {job.deployment}")
435+
job.progress.update_stage(
436+
cls.key,
437+
status=JobState.STARTED,
438+
progress=0,
439+
total_files=0,
440+
)
441+
job.save()
442+
443+
job.deployment.sync_captures(job=job)
444+
445+
job.logger.info(f"Finished syncing captures for deployment {job.deployment}")
446+
job.progress.update_stage(
447+
cls.key,
448+
status=JobState.SUCCESS,
449+
progress=1,
450+
)
451+
job.update_status(JobState.SUCCESS)
452+
job.save()
453+
else:
454+
job.update_status(JobState.FAILURE)
455+
456+
job.update_progress()
457+
job.finished_at = datetime.datetime.now()
458+
job.save()
459+
460+
414461
class SourceImageCollectionPopulateJob(JobType):
415462
name = "Populate Captures Collection"
416463
key = "populate_captures_collection"
@@ -467,53 +514,6 @@ def run(cls, job: "Job"):
467514
job.save()
468515

469516

470-
class DataStorageSyncJob(JobType):
471-
name = "Data storage sync"
472-
key = "data_storage_sync"
473-
474-
@classmethod
475-
def run(cls, job: "Job"):
476-
"""
477-
Run the data storage sync job.
478-
479-
This is meant to be called by an async task, not directly.
480-
"""
481-
482-
job.progress.add_stage(cls.name)
483-
job.progress.add_stage_param(cls.key, "Total files", "")
484-
job.update_status(JobState.STARTED)
485-
job.started_at = datetime.datetime.now()
486-
job.finished_at = None
487-
job.save()
488-
489-
if job.deployment:
490-
job.logger.info(f"Syncing captures for deployment {job.deployment}")
491-
job.progress.update_stage(
492-
cls.key,
493-
status=JobState.STARTED,
494-
progress=0,
495-
total_files=0,
496-
)
497-
job.save()
498-
499-
job.deployment.sync_captures(job=job)
500-
501-
job.logger.info(f"Finished syncing captures for deployment {job.deployment}")
502-
job.progress.update_stage(
503-
cls.key,
504-
status=JobState.SUCCESS,
505-
progress=1,
506-
)
507-
job.update_status(JobState.SUCCESS)
508-
job.save()
509-
else:
510-
job.update_status(JobState.FAILURE)
511-
512-
job.update_progress()
513-
job.finished_at = datetime.datetime.now()
514-
job.save()
515-
516-
517517
class UnknownJobType(JobType):
518518
name = "Unknown"
519519
key = "unknown"

0 commit comments

Comments
 (0)