mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-06-15 03:54:49 +08:00
71 lines
2.2 KiB
Python
Executable File
71 lines
2.2 KiB
Python
Executable File
import os
|
|
import subprocess
|
|
from openpilot.common.basedir import BASEDIR
|
|
|
|
|
|
class Spinner:
|
|
def __init__(self):
|
|
self.spinner_proc = None
|
|
|
|
# Prefer the legacy compiled spinner on device.
|
|
legacy_spinner = os.path.join(BASEDIR, "selfdrive", "ui", "spinner")
|
|
if os.path.isfile("/TICI") and os.path.isfile(legacy_spinner) and os.access(legacy_spinner, os.X_OK):
|
|
try:
|
|
self.spinner_proc = subprocess.Popen([legacy_spinner],
|
|
stdin=subprocess.PIPE,
|
|
cwd=os.path.join(BASEDIR, "selfdrive", "ui"),
|
|
close_fds=True)
|
|
return
|
|
except OSError:
|
|
self.spinner_proc = None
|
|
|
|
# Raylib spinner requires Python deps from the repo virtualenv.
|
|
spinner_cwd = os.path.join(BASEDIR, "system", "ui")
|
|
venv_python = os.path.join(BASEDIR, ".venv", "bin", "python")
|
|
python_exec = venv_python if os.path.isfile(venv_python) else "python3"
|
|
try:
|
|
self.spinner_proc = subprocess.Popen([python_exec, "./spinner.py"],
|
|
stdin=subprocess.PIPE,
|
|
cwd=spinner_cwd,
|
|
close_fds=True)
|
|
except OSError:
|
|
self.spinner_proc = None
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def update(self, spinner_text: str):
|
|
if self.spinner_proc is not None:
|
|
self.spinner_proc.stdin.write(spinner_text.encode('utf8') + b"\n")
|
|
try:
|
|
self.spinner_proc.stdin.flush()
|
|
except BrokenPipeError:
|
|
pass
|
|
|
|
def update_progress(self, cur: float, total: float):
|
|
self.update(str(round(100 * cur / total)))
|
|
|
|
def close(self):
|
|
if self.spinner_proc is not None:
|
|
self.spinner_proc.kill()
|
|
try:
|
|
self.spinner_proc.communicate(timeout=2.)
|
|
except subprocess.TimeoutExpired:
|
|
print("WARNING: failed to kill spinner")
|
|
self.spinner_proc = None
|
|
|
|
def __del__(self):
|
|
self.close()
|
|
|
|
def __exit__(self, exc_type, exc_value, traceback):
|
|
self.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import time
|
|
with Spinner() as s:
|
|
s.update("Spinner text")
|
|
time.sleep(5.0)
|
|
print("gone")
|
|
time.sleep(5.0)
|