Simplify cabana dbc dict (#32041)

* so needlessly complex!

* inside generate_dbc_json

* this is good

* clean up!

* spaces

* so much clean up
pull/32039/head
Shane Smiskol 1 year ago committed by GitHub
parent fa77d57e7c
commit d6d8e45931
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 14
      selfdrive/car/fingerprints.py
  2. 9
      tools/cabana/dbc/generate_dbc_json.py

@ -1,4 +1,3 @@
from typing import Any, Callable
from openpilot.selfdrive.car.interfaces import get_interface_attr from openpilot.selfdrive.car.interfaces import get_interface_attr
from openpilot.selfdrive.car.body.values import CAR as BODY from openpilot.selfdrive.car.body.values import CAR as BODY
from openpilot.selfdrive.car.chrysler.values import CAR as CHRYSLER from openpilot.selfdrive.car.chrysler.values import CAR as CHRYSLER
@ -11,7 +10,6 @@ from openpilot.selfdrive.car.nissan.values import CAR as NISSAN
from openpilot.selfdrive.car.subaru.values import CAR as SUBARU from openpilot.selfdrive.car.subaru.values import CAR as SUBARU
from openpilot.selfdrive.car.tesla.values import CAR as TESLA from openpilot.selfdrive.car.tesla.values import CAR as TESLA
from openpilot.selfdrive.car.toyota.values import CAR as TOYOTA from openpilot.selfdrive.car.toyota.values import CAR as TOYOTA
from openpilot.selfdrive.car.values import PLATFORMS, Platform
from openpilot.selfdrive.car.volkswagen.values import CAR as VW from openpilot.selfdrive.car.volkswagen.values import CAR as VW
FW_VERSIONS = get_interface_attr('FW_VERSIONS', combine_brands=True, ignore_none=True) FW_VERSIONS = get_interface_attr('FW_VERSIONS', combine_brands=True, ignore_none=True)
@ -338,15 +336,3 @@ MIGRATION = {
"SKODA SCALA 1ST GEN": VW.SKODA_SCALA_MK1, "SKODA SCALA 1ST GEN": VW.SKODA_SCALA_MK1,
"SKODA SUPERB 3RD GEN": VW.SKODA_SUPERB_MK3, "SKODA SUPERB 3RD GEN": VW.SKODA_SUPERB_MK3,
} }
MapFunc = Callable[[Platform], Any]
def create_platform_map(func: MapFunc):
ret = {str(platform): func(platform) for platform in PLATFORMS.values() if func(platform) is not None}
for m in MIGRATION:
ret[m] = ret[MIGRATION[m]]
return ret

@ -2,11 +2,16 @@
import argparse import argparse
import json import json
from openpilot.selfdrive.car.fingerprints import create_platform_map from openpilot.selfdrive.car.fingerprints import MIGRATION
from openpilot.selfdrive.car.values import PLATFORMS
def generate_dbc_json() -> str: def generate_dbc_json() -> str:
dbc_map = create_platform_map(lambda platform: platform.config.dbc_dict["pt"] if platform != "MOCK" else None) dbc_map = {platform.name: platform.config.dbc_dict['pt'] for platform in PLATFORMS.values() if platform != "MOCK"}
for m in MIGRATION:
dbc_map[m] = dbc_map[MIGRATION[m]]
return json.dumps(dict(sorted(dbc_map.items())), indent=2) return json.dumps(dict(sorted(dbc_map.items())), indent=2)

Loading…
Cancel
Save