screen-locker/screen_locker/_early_bird.py

"""Early bird window detection and log helpers for ScreenLocker."""

from __future__ import annotations

from datetime import datetime, timezone
import json
import logging

from screen_locker._constants import (
    EARLY_BIRD_END_HOUR,
    EARLY_BIRD_END_MINUTE,
    EARLY_BIRD_START_HOUR,
    EXTRA_BENEFITS_FILE,
)
from screen_locker._extra_benefits import has_extended_early_bird

_logger = logging.getLogger(__name__)


class EarlyBirdMixin:
    """Mixin providing early-bird time window checks and log helpers."""

    def _get_local_time_minutes(self) -> int:
        """Return current local time as minutes from midnight."""
        now = datetime.now(tz=timezone.utc).astimezone()
        return now.hour * 60 + now.minute

    def _is_early_bird_time(self) -> bool:
        """Return True if current local time is in the early bird window.

        Normally the window closes at 08:30. When the current ISO week has an
        extended early-bird reward (earned by 5+ workouts the prior week) the
        window extends to 09:00.
        """
        minutes = self._get_local_time_minutes()
        start = EARLY_BIRD_START_HOUR * 60
        if has_extended_early_bird(EXTRA_BENEFITS_FILE):
            end = 9 * 60  # 09:00
        else:
            end = EARLY_BIRD_END_HOUR * 60 + EARLY_BIRD_END_MINUTE
        return start <= minutes < end

    def _is_early_bird_log(self) -> bool:
        """Check if today's workout log entry is an early_bird provisional entry."""
        if not self.log_file.exists():
            return False
        try:
            with self.log_file.open() as f:
                logs = json.load(f)
        except (OSError, json.JSONDecodeError):
            return False
        today = datetime.now(tz=timezone.utc).strftime("%Y-%m-%d")
        entry = logs.get(today)
        if entry is None:
            return False
        return entry.get("workout_data", {}).get("type") == "early_bird"

    def _save_early_bird_log(self) -> None:
        """Save an early_bird provisional entry to the workout log."""
        self.workout_data = {"type": "early_bird"}
        self.save_workout_log()