mirror of
https://github.com/commaai/agnos-builder.git
synced 2026-07-11 01:32:09 +08:00
Adeeb Shihadeh
65 lines
1.7 KiB
Python
Executable File
65 lines
1.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import time
|
|
import fcntl
|
|
import struct
|
|
import threading
|
|
import subprocess
|
|
from datetime import timedelta
|
|
|
|
FN = "/var/tmp/power_watchdog"
|
|
THRESHOLD = timedelta(hours=1.0)
|
|
timestamps = {}
|
|
|
|
|
|
def read(path: str, num: bool = False):
|
|
try:
|
|
with open(path) as f:
|
|
if num:
|
|
return float(f.read())
|
|
return f.read()
|
|
except Exception:
|
|
return 0 if num else ""
|
|
|
|
|
|
def check_touches():
|
|
global timestamps
|
|
|
|
event_format = "llHHi"
|
|
event_size = struct.calcsize(event_format)
|
|
|
|
with open("/dev/input/by-path/platform-894000.i2c-event", "rb") as event_file:
|
|
fcntl.fcntl(event_file, fcntl.F_SETFL, os.O_NONBLOCK)
|
|
while True:
|
|
while (event := event_file.read(event_size)):
|
|
(sec, usec, etype, code, value) = struct.unpack(event_format, event)
|
|
if etype != 0 or code != 0 or value != 0:
|
|
timestamps['touch'] = time.monotonic()
|
|
time.sleep(60)
|
|
|
|
def ssh_active():
|
|
p = subprocess.run("ss | grep ssh", shell=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
return p.returncode == 0
|
|
|
|
if __name__ == "__main__":
|
|
# we limit worst-case power usage when openpilot isn't managing it,
|
|
# e.g. while building or during setup.
|
|
|
|
threading.Thread(target=check_touches, daemon=True).start()
|
|
|
|
timestamps['startup'] = time.monotonic()
|
|
while True:
|
|
timestamps['watchdog'] = read(FN, True)
|
|
if ssh_active():
|
|
timestamps['ssh'] = time.monotonic()
|
|
if read("/data/params/d/IsEngaged").startswith("1"):
|
|
timestamps['engaged'] = time.monotonic()
|
|
|
|
# time to shutoff?
|
|
dt = timedelta(seconds=time.monotonic() - max(timestamps.values()))
|
|
if dt > THRESHOLD:
|
|
os.system("sudo poweroff")
|
|
|
|
print((THRESHOLD - dt), "until shutdown", f"/ {timestamps=}")
|
|
time.sleep(60)
|