gear-orders/scheduling.py

import logging
import datetime
import pytz

from scheduler.asyncio import Scheduler

from settings import TIMEZONE
from orders import order_issue, order_check
from db.queries import order_status_outstanding, skip_day_contains
from util import order_time

logger = logging.getLogger(__name__)

SATURDAY = 5
SUNDAY = 6

GRACE_PERIOD = datetime.timedelta(seconds=10)

class OrderScheduler():
  def __init__(self, loop):
    self.tz = pytz.timezone(TIMEZONE)

    self.scheduler = Scheduler(loop=loop, tzinfo=self.tz)

    # Regularly scheduled orders
    order_time_dt = order_time()
    self.scheduler.daily(order_time_dt, self.scheduled_order)

    # Schedule any outstanding orders
    outstanding_orders = order_status_outstanding()
    for oo in outstanding_orders:
      self.scheduler.once(
        datetime.datetime.fromisoformat(oo.due_at) + GRACE_PERIOD,
        self.scheduled_check
        )
    
    logger.info(self.scheduler)

  async def scheduled_order(self):
    # Skip weekends
    day_of_week = datetime.datetime.now(tz=self.tz).weekday()
    if (day_of_week in [SATURDAY, SUNDAY]):
      logger.info('Today is a weekend')
      return

    # Skip stored dates
    today = datetime.datetime.now(tz=self.tz).strftime("%Y-%m-%d")
    logger.info('Today %s' % today)
    if (skip_day_contains(today)):
      logger.info('Today is a skip day')
      return

    due_at = await order_issue()

    if due_at is not None:
      # Schedule check
      self.scheduler.once(
        due_at + GRACE_PERIOD,
        self.scheduled_check
      )

    logger.info(self.scheduler)

  async def scheduled_check(self):
    await order_check()