vorta/src/vorta/scheduler.py

from apscheduler.schedulers.qt import QtScheduler
from apscheduler.triggers import cron

from vorta.borg.create import BorgCreateThread
from .models import BackupProfileMixin


class VortaScheduler(QtScheduler, BackupProfileMixin):
    def __init__(self, parent):
        super().__init__()
        self.app = parent
        self.start()
        self.reload()

    def reload(self):
        self.remove_all_jobs()
        trigger = None
        if self.profile().schedule_mode == 'interval':
            trigger = cron.CronTrigger(hour=f'*/{self.profile().schedule_interval_hours}',
                                       minute=self.profile().schedule_interval_minutes)
        elif self.profile().schedule_mode == 'fixed':
            trigger = cron.CronTrigger(hour=self.profile().schedule_fixed_hour,
                                       minute=self.profile().schedule_fixed_minute)

        if trigger is not None:
            self.add_job(self.create_backup, trigger, id='create-backup', misfire_grace_time=180)

    @property
    def next_job(self):
        job = self.get_job('create-backup')
        if job is None:
            return 'Manual Backups'
        else:
            return job.next_run_time.strftime('%Y-%m-%d %H:%M')

    def create_backup(self):
        msg = BorgCreateThread.prepare()
        if msg['ok']:
            thread = BorgCreateThread(msg['cmd'], msg['params'])
            thread.start()
            thread.wait()