use enum names

pull/23762/head
Shane Smiskol 3 years ago
parent 0b01282b9b
commit 4018aff1e1
  1. 6
      selfdrive/car/car_helpers.py

@ -53,6 +53,7 @@ def load_interfaces(brand_names):
CarController = None
for model_name in brand_names[brand_name]:
# print(model_name)
ret[model_name] = (CarInterface, CarController, CarState)
return ret
@ -66,7 +67,10 @@ def _get_interface_names():
try:
brand_name = car_folder.split('/')[-1]
model_names = __import__(f'selfdrive.car.{brand_name}.values', fromlist=['CAR']).CAR
model_names = [getattr(model_names, c) for c in model_names.__dict__.keys() if not c.startswith("__")]
if brand_name == 'toyota': # TODO: remove exception
model_names = [c.name for c in model_names]
else:
model_names = [getattr(model_names, c) for c in model_names.__dict__.keys() if not c.startswith("__")]
brand_names[brand_name] = model_names
except (ImportError, OSError):
pass

Loading…
Cancel
Save