def _init_resources(self):
self.time = Time()
self.odo = Odometer(
self.setup_file.get_value("odo").get("value"),
self.setup_file.get_value("odo").get("unit"),
)
# if we are here because a reset don't re-init the ecu
try:
self.ecu
except AttributeError:
self.ecu = Ecu()
def test__get_value_from_ecu(self, vendor_id, product_id, value, result):
def found_device(idVendor, idProduct):
if idVendor == vendor_id and idProduct == product_id:
return MagicMock()
else:
return None
with mock.patch("usb.core.find") as m_find, mock.patch(
"threading.Thread.start"
):
m_find.side_effect = found_device
ecu = Ecu()
assert ecu._get_value_from_ecu(value, fallback=666) == result
def test___init__(self, vendor_id, product_id, result):
def found_device(idVendor, idProduct):
if idVendor == vendor_id and idProduct == product_id:
return MagicMock()
else:
return None
with mock.patch("usb.core.find") as m_find, mock.patch(
"threading.Thread.start"
):
m_find.side_effect = found_device
ecu = Ecu()
assert isinstance(ecu.ecu, result)
class Backend:
def __init__(self):
self._load_user_preferences()
self._init_resources()
self._init_websocket()
def stop(self):
self.websocket.stop()
def _init_websocket(self):
self.websocket = Websocket(self)
def _init_resources(self):
self.time = Time()
self.odo = Odometer(
self.setup_file.get_value("odo").get("value"),
self.setup_file.get_value("odo").get("unit"),
)
# if we are here because a reset don't re-init the ecu
try:
self.ecu
except AttributeError:
self.ecu = Ecu()
def _load_user_preferences(self):
"""
In order to read only once from the setup file
we will load in memory some user preferences that we are gonna use later on.
"""
self.setup_file = SetupFile()
self.style = Style(
self.setup_file.json.get("style").get("tpsLowerThreshold"),
self.setup_file.json.get("style").get("tpsUpperThreshold"),
self.setup_file.json.get("style").get("elapsedSeconds"),
)
self.iat_unit = self.setup_file.json.get("iat",
{}).get("unit", "celsius")
self.ect_unit = self.setup_file.json.get("ect",
{}).get("unit", "celsius")
self.vss_unit = self.setup_file.json.get("vss", {}).get("unit", "kmh")
self.o2_unit = self.setup_file.json.get("o2", {}).get("unit", "afr")
self.odo_unit = self.setup_file.json.get("odo", {}).get("unit", "km")
self.map_unit = self.setup_file.json.get("map", {}).get("unit", "bar")
self.an0_unit = self.setup_file.json.get("an0",
{}).get("unit", "volts")
self.an1_unit = self.setup_file.json.get("an1",
{}).get("unit", "volts")
self.an2_unit = self.setup_file.json.get("an2",
{}).get("unit", "volts")
self.an3_unit = self.setup_file.json.get("an3",
{}).get("unit", "volts")
self.an4_unit = self.setup_file.json.get("an4",
{}).get("unit", "volts")
self.an5_unit = self.setup_file.json.get("an5",
{}).get("unit", "volts")
self.an6_unit = self.setup_file.json.get("an6",
{}).get("unit", "volts")
self.an7_unit = self.setup_file.json.get("an7",
{}).get("unit", "volts")
self.an0_formula, self.an0_extra_params = self.setup_file.get_formula(
"an0")
self.an1_formula, self.an1_extra_params = self.setup_file.get_formula(
"an1")
self.an2_formula, self.an2_extra_params = self.setup_file.get_formula(
"an2")
self.an3_formula, self.an3_extra_params = self.setup_file.get_formula(
"an3")
self.an4_formula, self.an4_extra_params = self.setup_file.get_formula(
"an4")
self.an5_formula, self.an5_extra_params = self.setup_file.get_formula(
"an5")
self.an6_formula, self.an6_extra_params = self.setup_file.get_formula(
"an6")
self.an7_formula, self.an7_extra_params = self.setup_file.get_formula(
"an7")
def _call_analog_input(self, port):
voltage = self.ecu.analog_input(port)
extra_params = getattr(self, f"an{port}_extra_params")
formula = getattr(self, f"an{port}_formula")
if extra_params is None:
return formula(**{"voltage": voltage})[getattr(
self, f"an{port}_unit")]
else:
args = {"voltage": voltage}
args.update(extra_params)
return formula(**args)
def update(self):
""" load the websocket with updated info """
if self.odo.save(self.ecu.vss["kmh"]):
self.setup_file.update_key("odo",
{"value": self.odo.preferred_mileage})
self.style.update(self.ecu.tps)
return {
"bat": self.ecu.bat,
"gear": self.ecu.gear,
"iat": self.ecu.iat[self.iat_unit],
"tps": self.ecu.tps,
"ect": self.ecu.ect[self.ect_unit],
"rpm": self.ecu.rpm,
"vss": self.ecu.vss[self.vss_unit],
"o2": self.ecu.o2[self.o2_unit],
"cam": self.ecu.cam,
"mil": self.ecu.mil,
"fan": self.ecu.fanc,
"bksw": self.ecu.bksw,
"flr": self.ecu.flr,
"eth": self.ecu.eth,
"scs": self.ecu.scs,
"fmw": self.ecu.firmware,
"map": self.ecu.map[self.map_unit],
"an0": self._call_analog_input(0),
"an1": self._call_analog_input(1),
"an2": self._call_analog_input(2),
"an3": self._call_analog_input(3),
"an4": self._call_analog_input(4),
"an5": self._call_analog_input(5),
"an6": self._call_analog_input(6),
"an7": self._call_analog_input(7),
"time": self.time.get_time(),
"odo": self.odo.preferred_mileage,
"style": self.style.status,
"name": self.ecu.name,
"ver": __version__,
}
def setup(self):
"""Return the current setup"""
return self.setup_file.load_setup()
def save(self, new_setup):
"""Save a new setup"""
SetupValidator().validate(new_setup)
self.setup_file.save_setup(new_setup)
self.setup_file.rotate_screen(new_setup["screen"]["rotate"])
# refresh the backend too
self._load_user_preferences()
self._init_resources()
def reset(self):
"""Reset to the default setup"""
self.setup_file.reset_setup()
# refresh the backend too
self._load_user_preferences()
self._init_resources()
# PYTHONPATH=src python src/backend/bench/ecu.py
from __future__ import print_function
from reprint import output
from backend.devices.ecu import Ecu
ecu = Ecu()
with output(output_type="dict", initial_len=1, interval=0) as output_list:
while True:
output_list["ETH"] = str(ecu.eth)
output_list["FLT"] = str(ecu.flt)
output_list["BAT"] = str(ecu.bat)
output_list["CAM"] = str(ecu.cam)
output_list["O2"] = str(ecu.o2)
output_list["IAT"] = str(ecu.iat)
output_list["RPM"] = str(ecu.rpm)
output_list["TPS"] = str(ecu.tps)
output_list["VSS"] = str(ecu.vss)
output_list["ECT"] = str(ecu.ect)
output_list["GEAR"] = str(ecu.gear)
output_list["EPS"] = str(ecu.eps)
output_list["SCS"] = str(ecu.scs)
output_list["RVSLCK"] = str(ecu.rvslck)
output_list["BKSW"] = str(ecu.bksw)
output_list["ACSW"] = str(ecu.acsw)
output_list["ACCL"] = str(ecu.accl)
output_list["FLR"] = str(ecu.flr)
output_list["FANC"] = str(ecu.fanc)
output_list["MAP"] = str(ecu.map)
output_list["AN0"] = str(ecu.analog_input(0))