firestar5683
67 lines
2.2 KiB
Python
67 lines
2.2 KiB
Python
#!/usr/bin/env python3
|
|
import argparse
|
|
import time
|
|
|
|
from cereal import log, messaging
|
|
from openpilot.system.hardware import HARDWARE
|
|
|
|
NETWORK_TYPES = {
|
|
"none": log.DeviceState.NetworkType.none,
|
|
"wifi": log.DeviceState.NetworkType.wifi,
|
|
"ethernet": log.DeviceState.NetworkType.ethernet,
|
|
"cell2G": log.DeviceState.NetworkType.cell2G,
|
|
"cell3G": log.DeviceState.NetworkType.cell3G,
|
|
"cell4G": log.DeviceState.NetworkType.cell4G,
|
|
"cell5G": log.DeviceState.NetworkType.cell5G,
|
|
}
|
|
|
|
NETWORK_STRENGTHS = {
|
|
"unknown": log.DeviceState.NetworkStrength.unknown,
|
|
"poor": log.DeviceState.NetworkStrength.poor,
|
|
"moderate": log.DeviceState.NetworkStrength.moderate,
|
|
"good": log.DeviceState.NetworkStrength.good,
|
|
"great": log.DeviceState.NetworkStrength.great,
|
|
}
|
|
|
|
|
|
def parse_args() -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="Publish fake deviceState network info for desktop UI demos.")
|
|
parser.add_argument("--network", choices=NETWORK_TYPES.keys(), default="wifi", help="network type to publish")
|
|
parser.add_argument("--strength", choices=NETWORK_STRENGTHS.keys(), default="great", help="network strength to publish")
|
|
parser.add_argument("--started", action="store_true", help="publish started=true (onroad)")
|
|
parser.add_argument("--metered", action="store_true", help="publish networkMetered=true")
|
|
parser.add_argument("--interval", type=float, default=0.2, help="publish interval in seconds")
|
|
return parser.parse_args()
|
|
|
|
|
|
def main() -> None:
|
|
args = parse_args()
|
|
|
|
pm = messaging.PubMaster(["deviceState"])
|
|
interval = max(args.interval, 0.05)
|
|
network_type = NETWORK_TYPES[args.network]
|
|
network_strength = NETWORK_STRENGTHS[args.strength]
|
|
|
|
while True:
|
|
msg = messaging.new_message("deviceState")
|
|
ds = msg.deviceState
|
|
|
|
ds.deviceType = HARDWARE.get_device_type()
|
|
ds.started = args.started
|
|
ds.networkType = network_type
|
|
ds.networkStrength = network_strength
|
|
ds.networkMetered = args.metered
|
|
ds.lastAthenaPingTime = time.monotonic_ns()
|
|
|
|
# Keep common sidebar values reasonable for desktop demos.
|
|
ds.freeSpacePercent = 80
|
|
ds.memoryUsagePercent = 35
|
|
ds.maxTempC = 45
|
|
|
|
pm.send("deviceState", msg)
|
|
time.sleep(interval)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|