57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import random
|
|
from time import sleep
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from loguru import logger
|
|
|
|
from common.models import BaseJob, JobManager
|
|
|
|
# @JobManager.register
|
|
# class DummyJob(BaseJob):
|
|
# interval = timedelta(seconds=10)
|
|
|
|
# def run(self):
|
|
# logger.info("Dummy job started")
|
|
# if random.choice([0, 1]) == 0:
|
|
# raise Exception("Dummy job randomly failed")
|
|
# sleep(3)
|
|
# logger.info("Dummy job stopped")
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = "Schedule timed jobs"
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
"--cancel",
|
|
action="store_true",
|
|
)
|
|
parser.add_argument(
|
|
"--schedule",
|
|
action="store_true",
|
|
)
|
|
parser.add_argument(
|
|
"--list",
|
|
action="store_true",
|
|
)
|
|
parser.add_argument(
|
|
"--runonce",
|
|
action="append",
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
if options["cancel"]:
|
|
JobManager.cancel_all()
|
|
if options["schedule"]:
|
|
JobManager.cancel_all() # cancel previously scheduled jobs if any
|
|
JobManager.schedule_all()
|
|
if options["runonce"]:
|
|
for job_id in options["runonce"]:
|
|
run = JobManager.run(job_id)
|
|
if not run:
|
|
logger.error(f"job not found: {job_id}")
|
|
if options["list"]:
|
|
all_jobs = [j.__name__ for j in JobManager.registry]
|
|
logger.info(f"{len(all_jobs)} available jobs: {' '.join(all_jobs)}")
|
|
jobs = JobManager.get_scheduled_job_ids()
|
|
logger.info(f"{len(jobs)} scheduled jobs: {' '.join(jobs)}")
|