scheduler.py

from apscheduler.schedulers.blocking import BlockingScheduler
from apscheduler.jobstores.redis import RedisJobStore
from apscheduler.executors.pool import ThreadPoolExecutor

from datetime import datetime, timedelta
import os
import sys

from logger import log


REDIS_HOST = os.environ['REDIS_HOST']
REDIS_PORT = os.environ['REDIS_PORT']
REDIS_PASSWORD = os.environ['REDIS_PASSWORD']

jobstores = {
  'default': RedisJobStore(host=REDIS_HOST, port=REDIS_PORT, password=REDIS_PASSWORD,
                           ssl_cert_reqs='optional',
                           ssl=True,
                           jobs_key='apscheduler.jobs',
                           run_times_key='apscheduler.run_times')
}

executors = {
  'default': ThreadPoolExecutor(20)
}

scheduler_timezone = 'UTC'

sched = BlockingScheduler(jobstores=jobstores, executors=executors, timezone=scheduler_timezone)


def schedule_function_1():
  scheduled_job = sched.add_job(func='functions:function_1',
                                trigger='date',
                                run_date=datetime.utcnow() + timedelta(minutes=1),
                                timezone=scheduler_timezone,
                                misfire_grace_time=None
                                )
  log.info(f'scheduled_job: {scheduled_job}')

def schedule_function_2():
  scheduled_job = sched.add_job(func='functions:function_2',
                                trigger='date',
                                run_date=datetime.utcnow() + timedelta(minutes=1),
                                timezone=scheduler_timezone,
                                misfire_grace_time=None
                                )
  log.info(f'scheduled_job: {scheduled_job}')
  log.info(f'Scheduler running? {sched.running}')


if __name__ == '__main__':
    schedule_function_1()

    try:
      log.info(f'Starting main scheduling loop')
      sched.start()
    except (KeyboardInterrupt, SystemExit):
      log.warning(f'Interrupted, shutting down')
      sched.shutdown()