Updated heartbeat quiet hours

1 files changed, 6 insertions, 3 deletions

diff --git a/scripts/heartbeat.py b/scripts/heartbeat.py
index 062cecb..5fe321b 100644
--- a/scripts/heartbeat.py
+++ b/scripts/heartbeat.py
@@ -12,7 +12,7 @@ import subprocess
 import sys
 import urllib.request
 from email.header import Header
-from datetime import datetime, timedelta, timezone
+from datetime import time, datetime, timedelta, timezone
 from pathlib import Path
 
 # ---------- config ----------
@@ -22,7 +22,8 @@ STATE_DIR = Path("/etc/senia/data/heartbeat")
 LAST_WAKE_FILE = STATE_DIR / "last_wake_at"
 WORKSPACE = Path("/workspace")
 
-QUIET_MAX_HOURS = 4
+QUIET_MIN_HOURS = 2.5
+QUIET_MAX_HOURS = QUIET_MIN_HOURS * 2
 
 NTFY_URL = os.environ["NTFY_URL"]       # e.g. https://ntfy.example.tld/senia
 NTFY_TOKEN = os.environ["NTFY_TOKEN"]
@@ -63,7 +64,9 @@ def should_wake(tasks, last_wake, now) -> tuple[bool, str]:
     if last_wake is None:
         return True, "First wake-up since the script was set up."
     elapsed = now - last_wake
-    if elapsed >= timedelta(hours=QUIET_MAX_HOURS):
+    now = datetime.now().time()
+    quiet_hours = QUIET_MAX_HOURS if (now >= time(22) or now < time(7)) else QUIET_MIN_HOURS
+    if elapsed >= timedelta(hours=quiet_hours):
         hrs = elapsed.total_seconds() / 3600
         return True, f"It's been {hrs:.1f} hours since the last check-in."
     return False, ""