mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-07-04 15:03:15 +02:00
Krzysztof kuhy Rudnicki
74a8bd7529
- Refactor RunnerUp verification: extract RunnerUpDbMixin (_runnerup_db.py), split _scan_and_fill_week_runnerup into a helper _try_fill_runnerup_for_date to keep cyclomatic complexity ≤10 - Generalise TCX lookup to any date in the ISO week (was today-only); all gap days Mon→today auto-filled on every startup and 08:30 timer firing - Add _adjust_shutdown_time_by(): +1h per extra workout beyond the 4-workout minimum, capped at midnight (hour=24) - Add _shutdown_base.py: daily reset of shutdown config to a stored base so the bonus doesn't silently accumulate across days - Add _extra_benefits.py: streak tracking, skip credits (earn (n-4) credits for 5+ workout weeks), early-bird extension to 09:00 for eligible weeks - Add --status mode (_status.py): non-locking CLI view showing per-day breakdown (✓/✗), RunnerUp auto-scan, bonus status, shutdown time, streak, skip credits, and early-bird status - Hook carrot into _check_non_verify_exits: bonus applied whenever auto-fill pushes weekly count above the minimum - Pass all pre-commit hooks (ruff, mypy, pylint, bandit, shellcheck, codespell, max-file-length); 508 tests at 100% branch coverage Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_017auyHmf2ZwQcDAwXaSo7KX
136 lines
6.5 KiB
Python
136 lines
6.5 KiB
Python
"""Mixin: auto-upgrade early_bird/sick_day log entries via phone or RunnerUp."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime, timezone
|
|
import json
|
|
import logging
|
|
import sys
|
|
|
|
from screen_locker._wake_state import has_workout_skip_today
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class AutoUpgradeMixin:
|
|
"""Handles today-state detection and silent log-entry upgrading.
|
|
|
|
Relies on methods from EarlyBirdMixin, PhoneVerificationMixin,
|
|
RunnerUpVerificationMixin, LogMixin, and ShutdownMixin via MRO.
|
|
"""
|
|
|
|
def _is_sick_day_log(self) -> bool:
|
|
"""Check if today's workout log is a sick day (not yet verified)."""
|
|
if not self.log_file.exists(): # type: ignore[attr-defined]
|
|
return False
|
|
try:
|
|
with self.log_file.open() as f: # type: ignore[attr-defined]
|
|
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") == "sick_day"
|
|
|
|
def _check_early_exits(self, *, verify_only: bool) -> None:
|
|
"""Check startup conditions and exit early when appropriate."""
|
|
if verify_only:
|
|
if not self._is_sick_day_log():
|
|
_logger.info("No sick day logged today. Nothing to verify.")
|
|
sys.exit(0)
|
|
return
|
|
self._check_non_verify_exits() # type: ignore[attr-defined]
|
|
|
|
def _check_today_state_exits(self) -> bool:
|
|
"""Handle early-bird and today's log states. Return True to stop startup."""
|
|
if (
|
|
self._is_early_bird_log() # type: ignore[attr-defined]
|
|
and not self._is_early_bird_time() # type: ignore[attr-defined]
|
|
):
|
|
if self._try_auto_upgrade_early_bird():
|
|
_logger.info("Auto-upgraded early_bird entry to phone_verified.")
|
|
sys.exit(0)
|
|
return True
|
|
return False # Expired early bird, upgrade unavailable — full lock.
|
|
if self._is_early_bird_log(): # type: ignore[attr-defined]
|
|
_logger.info("Early bird window still active — skipping lock.")
|
|
elif self._is_sick_day_log() and self._try_auto_upgrade_sick_day():
|
|
_logger.info("Auto-upgraded today's sick_day entry to phone_verified.")
|
|
elif self.has_logged_today(): # type: ignore[attr-defined]
|
|
_logger.info("Workout already logged today. Skipping screen lock.")
|
|
elif has_workout_skip_today():
|
|
_logger.info("Wake alarm earned workout skip. Skipping screen lock.")
|
|
elif self._is_early_bird_time(): # type: ignore[attr-defined]
|
|
self._save_early_bird_log() # type: ignore[attr-defined]
|
|
_logger.info("Early bird time — skipping lock, will re-check at 08:30.")
|
|
else:
|
|
return False
|
|
sys.exit(0)
|
|
return True
|
|
|
|
def _try_auto_upgrade_sick_day(self) -> bool:
|
|
"""Upgrade sick_day entry when phone or RunnerUp detects a valid workout."""
|
|
try:
|
|
status, message = self._verify_phone_workout() # type: ignore[attr-defined]
|
|
except (OSError, RuntimeError) as exc:
|
|
_logger.info("Auto-upgrade phone check failed: %s", exc)
|
|
status, message = "error", str(exc)
|
|
if status == "verified":
|
|
self.workout_data["type"] = "phone_verified" # type: ignore[attr-defined]
|
|
self.workout_data["source"] = message # type: ignore[attr-defined]
|
|
self.workout_data["after_sick_day"] = "true" # type: ignore[attr-defined]
|
|
self._adjust_shutdown_time_later() # type: ignore[attr-defined]
|
|
self.save_workout_log() # type: ignore[attr-defined]
|
|
return True
|
|
_logger.info("Auto-upgrade phone skipped (%s), trying RunnerUp...", status)
|
|
try:
|
|
runnerup_status, runnerup_msg = self._verify_runnerup_workout() # type: ignore[attr-defined]
|
|
except (OSError, RuntimeError) as exc:
|
|
_logger.info("Auto-upgrade RunnerUp check failed: %s", exc)
|
|
return False
|
|
if runnerup_status != "verified":
|
|
_logger.info(
|
|
"Auto-upgrade RunnerUp skipped (%s): %s", runnerup_status, runnerup_msg
|
|
)
|
|
return False
|
|
self.workout_data["type"] = "runnerup_verified" # type: ignore[attr-defined]
|
|
self.workout_data["source"] = runnerup_msg # type: ignore[attr-defined]
|
|
self.workout_data["after_sick_day"] = "true" # type: ignore[attr-defined]
|
|
self._adjust_shutdown_time_later() # type: ignore[attr-defined]
|
|
self.save_workout_log() # type: ignore[attr-defined]
|
|
return True
|
|
|
|
def _try_auto_upgrade_early_bird(self) -> bool:
|
|
"""Try phone then RunnerUp to upgrade an early_bird log entry."""
|
|
try:
|
|
status, message = self._verify_phone_workout() # type: ignore[attr-defined]
|
|
except (OSError, RuntimeError) as exc:
|
|
_logger.info("Early bird upgrade phone check failed: %s", exc)
|
|
status, message = "error", str(exc)
|
|
if status == "verified":
|
|
self.workout_data["type"] = "phone_verified" # type: ignore[attr-defined]
|
|
self.workout_data["source"] = message # type: ignore[attr-defined]
|
|
self.workout_data["after_early_bird"] = "true" # type: ignore[attr-defined]
|
|
self._adjust_shutdown_time_later() # type: ignore[attr-defined]
|
|
self.save_workout_log() # type: ignore[attr-defined]
|
|
return True
|
|
_logger.info("Early bird phone skipped (%s), trying RunnerUp...", status)
|
|
try:
|
|
runnerup_status, runnerup_msg = self._verify_runnerup_workout() # type: ignore[attr-defined]
|
|
except (OSError, RuntimeError) as exc:
|
|
_logger.info("Early bird RunnerUp check failed: %s", exc)
|
|
return False
|
|
if runnerup_status != "verified":
|
|
_logger.info(
|
|
"Early bird RunnerUp skipped (%s): %s", runnerup_status, runnerup_msg
|
|
)
|
|
return False
|
|
self.workout_data["type"] = "runnerup_verified" # type: ignore[attr-defined]
|
|
self.workout_data["source"] = runnerup_msg # type: ignore[attr-defined]
|
|
self.workout_data["after_early_bird"] = "true" # type: ignore[attr-defined]
|
|
self._adjust_shutdown_time_later() # type: ignore[attr-defined]
|
|
self.save_workout_log() # type: ignore[attr-defined]
|
|
return True
|