mirror of
https://github.com/kuhyx/wake-alarm.git
synced 2026-11-11 15:33:49 +01:00
Krzysztof kuhy Rudnicki
c1dbdcda22
`requires-python` and the CI pin already said 3.14, but ruff, mypy and pylint were still grading against 3.10/3.11 -- three tools judging an interpreter nothing runs any more, and ruff withholding the pyupgrade fixes that only become safe at the newer target. `pin_ci_python.py` rewrites these three knobs. What the stale target was hiding: `datetime.timezone.utc` where `datetime.UTC` is now correct (UP017), `socket.timeout` where `TimeoutError` is (UP041), and ruff's stable formatter applying PEP 758 -- `except (A, B):` becomes `except A, B:`. That makes the source 3.14-only at the syntax level, which is what `requires-python = ">=3.14"` already promised; kuhy chose it deliberately over holding ruff a release back (2026-08-29). ruff, ruff-format, mypy --strict and pylint all clean at the new target. Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01SfnJmxxsZ534AKk16rC2Gr
130 lines
3.4 KiB
Python
130 lines
3.4 KiB
Python
"""HMAC-signed state management for the weekend wake alarm."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import UTC, datetime
|
|
import json
|
|
import logging
|
|
|
|
from gatelock.log_integrity import (
|
|
compute_entry_hmac,
|
|
verify_entry_hmac,
|
|
)
|
|
|
|
from wake_alarm._constants import WAKE_STATE_FILE, WORKOUT_LOG_FILE
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
def _today_str() -> str:
|
|
"""Return today's date as YYYY-MM-DD in UTC."""
|
|
return datetime.now(tz=UTC).strftime("%Y-%m-%d")
|
|
|
|
|
|
def save_wake_state(
|
|
*,
|
|
dismissed_at: str | None,
|
|
skip_workout: bool,
|
|
) -> bool:
|
|
"""Write today's wake state with HMAC signature.
|
|
|
|
Args:
|
|
dismissed_at: ISO time when alarm was dismissed, or None.
|
|
skip_workout: Whether the user earned a workout skip.
|
|
|
|
Returns:
|
|
True if saved successfully, False otherwise.
|
|
"""
|
|
entry: dict[str, object] = {
|
|
"date": _today_str(),
|
|
"dismissed_at": dismissed_at,
|
|
"skip_workout": skip_workout,
|
|
}
|
|
signature = compute_entry_hmac(entry)
|
|
if signature is not None:
|
|
entry["hmac"] = signature
|
|
else:
|
|
_logger.warning("HMAC key unavailable — saving unsigned wake state")
|
|
|
|
try:
|
|
with WAKE_STATE_FILE.open("w") as f:
|
|
json.dump(entry, f, indent=2)
|
|
except OSError as exc:
|
|
_logger.warning("Failed to save wake state: %s", exc)
|
|
return False
|
|
|
|
_logger.info(
|
|
"Saved wake state: dismissed=%s skip=%s",
|
|
dismissed_at,
|
|
skip_workout,
|
|
)
|
|
return True
|
|
|
|
|
|
def load_wake_state() -> dict[str, object] | None:
|
|
"""Load and verify today's wake state.
|
|
|
|
Returns the state dict if it exists, is valid (HMAC OK), and is
|
|
for today. Returns None otherwise.
|
|
"""
|
|
if not WAKE_STATE_FILE.exists():
|
|
return None
|
|
|
|
try:
|
|
with WAKE_STATE_FILE.open() as f:
|
|
state = json.load(f)
|
|
except OSError, json.JSONDecodeError:
|
|
_logger.warning("Cannot read wake state file")
|
|
return None
|
|
|
|
if not isinstance(state, dict):
|
|
return None
|
|
|
|
if state.get("date") != _today_str():
|
|
return None
|
|
|
|
if not verify_entry_hmac(state):
|
|
_logger.warning("Wake state HMAC verification failed")
|
|
return None
|
|
|
|
return state
|
|
|
|
|
|
def has_workout_skip_today() -> bool:
|
|
"""Check if the user earned a workout skip for today."""
|
|
state = load_wake_state()
|
|
if state is None:
|
|
return False
|
|
return bool(state.get("skip_workout"))
|
|
|
|
|
|
def was_alarm_dismissed_today() -> bool:
|
|
"""Check if the alarm was already dismissed today."""
|
|
state = load_wake_state()
|
|
if state is None:
|
|
return False
|
|
return state.get("dismissed_at") is not None
|
|
|
|
|
|
def was_workout_logged_today() -> bool:
|
|
"""Check if the workout was already logged today via the screen locker.
|
|
|
|
Reads the companion screen_locker workout_log.json. The file is a
|
|
dict keyed by YYYY-MM-DD date strings; presence of today's key means
|
|
the workout was completed and the alarm is no longer needed.
|
|
|
|
Returns:
|
|
True if today's workout entry exists, False on any error or absence.
|
|
"""
|
|
if not WORKOUT_LOG_FILE.exists():
|
|
return False
|
|
try:
|
|
with WORKOUT_LOG_FILE.open() as f:
|
|
log = json.load(f)
|
|
except OSError, json.JSONDecodeError:
|
|
_logger.warning("Cannot read workout log file %s", WORKOUT_LOG_FILE)
|
|
return False
|
|
if not isinstance(log, dict):
|
|
return False
|
|
return _today_str() in log
|