mirror of https://github.com/commaai/panda.git
64 lines
1.8 KiB
Python
Executable File
64 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import sys
|
|
import time
|
|
import select
|
|
import codecs
|
|
|
|
sys.path.append(os.path.join(os.path.dirname(os.path.realpath(__file__)), ".."))
|
|
from panda import Panda # noqa: E402
|
|
|
|
setcolor = ["\033[1;32;40m", "\033[1;31;40m"]
|
|
unsetcolor = "\033[00m"
|
|
|
|
port_number = int(os.getenv("PORT", "0"))
|
|
claim = os.getenv("CLAIM") is not None
|
|
no_color = os.getenv("NO_COLOR") is not None
|
|
no_reconnect = os.getenv("NO_RECONNECT") is not None
|
|
|
|
if __name__ == "__main__":
|
|
while True:
|
|
try:
|
|
serials = Panda.list()
|
|
if os.getenv("SERIAL"):
|
|
serials = [x for x in serials if x == os.getenv("SERIAL")]
|
|
|
|
pandas = list([Panda(x, claim=claim) for x in serials])
|
|
decoders = [codecs.getincrementaldecoder('utf-8')() for _ in pandas]
|
|
|
|
if not len(pandas):
|
|
print("no pandas found")
|
|
if no_reconnect:
|
|
sys.exit(0)
|
|
time.sleep(1)
|
|
continue
|
|
|
|
if os.getenv("BAUD") is not None:
|
|
for panda in pandas:
|
|
panda.set_uart_baud(port_number, int(os.getenv("BAUD"))) # type: ignore
|
|
|
|
while True:
|
|
for i, panda in enumerate(pandas):
|
|
while True:
|
|
ret = panda.serial_read(port_number)
|
|
if len(ret) > 0:
|
|
decoded = decoders[i].decode(ret)
|
|
if no_color:
|
|
sys.stdout.write(decoded)
|
|
else:
|
|
sys.stdout.write(setcolor[i] + decoded + unsetcolor)
|
|
sys.stdout.flush()
|
|
else:
|
|
break
|
|
if select.select([sys.stdin], [], [], 0) == ([sys.stdin], [], []):
|
|
ln = sys.stdin.readline()
|
|
if claim:
|
|
panda.serial_write(port_number, ln)
|
|
time.sleep(0.01)
|
|
except KeyboardInterrupt:
|
|
break
|
|
except Exception:
|
|
print("panda disconnected!")
|
|
time.sleep(0.5)
|