mirror of
https://github.com/kuhyx/screen-locker.git
synced 2026-11-10 20:47:14 +01:00
Krzysztof kuhy Rudnicki
02a93f1322
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
123 lines
4.3 KiB
Python
123 lines
4.3 KiB
Python
"""Shutdown projection and the one-line i3blocks summary.
|
|
|
|
Split out of :mod:`screen_locker._status_data` to keep every file under the
|
|
250-line cap. ``format_summary_line`` is re-exported from there, so
|
|
``status_view`` and the i3blocks CLI are unaffected.
|
|
|
|
Projects what time the PC will shut down on each remaining day of the week,
|
|
and renders the compact status line.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from datetime import datetime
|
|
from typing import TYPE_CHECKING
|
|
|
|
from screen_locker._extra_benefits import (
|
|
current_streak,
|
|
preview_bonus_if_week_ended_now,
|
|
weekly_shutdown_bonus_hours,
|
|
)
|
|
from screen_locker._shutdown import read_shutdown_config
|
|
from screen_locker._shutdown_base import BASE_HOUR
|
|
from screen_locker._status_types import ShutdownProjection, ShutdownProjectionDay
|
|
from screen_locker._weekly_check import count_weekly_workouts
|
|
|
|
if TYPE_CHECKING:
|
|
from pathlib import Path
|
|
|
|
from screen_locker._status_types import StatusSnapshot
|
|
|
|
_RELAXED_DAY_EXPLANATION = (
|
|
"Shutdown time and the screen lock are two separate systems: the lock "
|
|
"decides whether you must log a workout right now; shutdown time only "
|
|
"controls how late the PC stays on tonight."
|
|
)
|
|
_WEEKDAY_LABELS = ("Mon", "Tue", "Wed", "Thu", "Fri", "Sat", "Sun")
|
|
_MON_WED_WEEKDAYS = frozenset({0, 1, 2})
|
|
|
|
|
|
def _week_rows(
|
|
mon_wed_hour: int, thu_sun_hour: int, *, speculative: bool
|
|
) -> tuple[ShutdownProjectionDay, ...]:
|
|
"""Build 7 labeled rows, one per weekday, from a Mon-Wed/Thu-Sun band pair."""
|
|
return tuple(
|
|
ShutdownProjectionDay(
|
|
label=label,
|
|
hour=mon_wed_hour if weekday in _MON_WED_WEEKDAYS else thu_sun_hour,
|
|
speculative=speculative,
|
|
)
|
|
for weekday, label in enumerate(_WEEKDAY_LABELS)
|
|
)
|
|
|
|
|
|
def _shutdown_projection(
|
|
*,
|
|
shutdown_config_file: Path,
|
|
extra_benefits_file: Path,
|
|
log_file: Path,
|
|
today_local: datetime,
|
|
) -> ShutdownProjection:
|
|
"""Build the shutdown-time projection: tonight, rest of week, next week."""
|
|
tonight = read_shutdown_config(shutdown_config_file)
|
|
bonus = weekly_shutdown_bonus_hours(extra_benefits_file, today=today_local)
|
|
rest_of_week = _week_rows(BASE_HOUR + bonus, BASE_HOUR + bonus, speculative=False)
|
|
|
|
this_week_count = count_weekly_workouts(log_file, today=today_local)
|
|
streak = current_streak(extra_benefits_file)
|
|
_would_be_streak, would_be_bonus = preview_bonus_if_week_ended_now(
|
|
this_week_count, streak
|
|
)
|
|
next_week_preview = _week_rows(
|
|
BASE_HOUR + would_be_bonus, BASE_HOUR + would_be_bonus, speculative=True
|
|
)
|
|
|
|
return ShutdownProjection(
|
|
tonight=tonight,
|
|
rest_of_week=rest_of_week,
|
|
next_week_preview=next_week_preview,
|
|
explanation=_RELAXED_DAY_EXPLANATION,
|
|
)
|
|
|
|
|
|
def _tonight_hour(snapshot: StatusSnapshot) -> int | None:
|
|
"""Pick the shutdown hour band (Mon-Wed vs Thu-Sun) matching today.
|
|
|
|
Uses ``snapshot.generated_at`` (not a fresh ``datetime.now()``) so the
|
|
summary line is a pure function of the snapshot it was handed.
|
|
"""
|
|
if snapshot.shutdown.tonight is None:
|
|
return None
|
|
mon_wed_hour, thu_sun_hour, _morning = snapshot.shutdown.tonight
|
|
generated_weekday = (
|
|
datetime.fromisoformat(snapshot.generated_at).astimezone().weekday()
|
|
)
|
|
return mon_wed_hour if generated_weekday in _MON_WED_WEEKDAYS else thu_sun_hour
|
|
|
|
|
|
def format_summary_line(snapshot: StatusSnapshot) -> str:
|
|
"""Cheap one-liner for the i3blocks summary (e.g. before/after a click)."""
|
|
week = snapshot.week
|
|
mark = "✓" if week.remaining == 0 else "…"
|
|
tonight_hour = _tonight_hour(snapshot)
|
|
tonight_str = f"{tonight_hour:02d}:00" if tonight_hour is not None else "?"
|
|
sick = snapshot.sick_budget
|
|
return (
|
|
f"{mark} {week.counted_count}/{week.minimum} workouts · "
|
|
f"{tonight_str} tonight · sick {sick.used_7d}/{sick.budget_7d}"
|
|
)
|
|
|
|
|
|
def compliance_state_word(snapshot: StatusSnapshot) -> str:
|
|
"""One-word compliance state for the tray icon: ``ok`` / ``warn`` / ``lock``.
|
|
|
|
``lock`` — the lock would fire right now (no skip condition applies).
|
|
``warn`` — not locked, but this week's minimum isn't met yet.
|
|
``ok`` — not locked and the weekly minimum is already met.
|
|
"""
|
|
if snapshot.lock_explanation.fired:
|
|
return "lock"
|
|
if snapshot.week.remaining > 0:
|
|
return "warn"
|
|
return "ok"
|