schedule.py

from functools import lru_cache
import tzlocal
from datetime import datetime, timedelta
from dateutil import parser

class ScheduleManager(object):
    def __init__(self, config, timezone=None):
        self.config = config
        if timezone == None:
            self.timezone = tzlocal.get_localzone()
        else:
            self.timezone = timezone

    @staticmethod
    @lru_cache(maxsize=8, typed=True)
    def _calc_sched(day_sched):
        # If day isn't defined, 9-5. If defined and empty, not on that day
        if not day_sched:
            return None
        day_sched = [parser.parse(x).time() for x in day_sched.split('-')]
        return day_sched

    def _getsched(self):
        if 'schedule' in self.config:
            # Missing days have no active hours
            schedule = self.config['schedule'] or {}
            schedule = [schedule.get(("mon", "tue", "wed", "thu", "fri", "sat", "sun")[x]) for x in range(7)]
        else:
            # Default schedule is 9-5 weekdays
            schedule = ['9-17']*5 + ['']*2

        return [self._calc_sched(d) for d in schedule]

    def get_next_sleep(self):
        # If schedule key exists and is empty, then there is no schedule
        schedule = self._getsched()
        if schedule == [None]*7:
            return 0
        sleep_day = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0)
        sched = None
        while True:
            sched = schedule[sleep_day.weekday()]
            if sched is None:
                sleep_day -= timedelta(1)
            else:
                break
        return self.timezone.localize(sleep_day.combine(sleep_day, sched[1]))

    def get_next_wake(self):
        # If schedule key exists and is empty, then there is no schedule
        schedule = self._getsched()
        if schedule == [None]*7:
            return 0
        wake_day = datetime.now().replace(hour=0, minute=0, second=0, microsecond=0) + timedelta(1)
        sched = None
        while True:
            sched = schedule[wake_day.weekday()]
            if sched is None:
                wake_day += timedelta(1)
            else:
                break
        return self.timezone.localize(wake_day.combine(wake_day, sched[0]))

    def live_now(self, now=None):
        if now == None:
            now = datetime.now(tz=self.timezone)
        schedule = self._getsched()
        if schedule == [None]*7:
            return True
        sched = schedule[now.weekday()]
        now = now.time()
        if sched is None:
            return False
        return sched[0] <= now <= sched[1]