From 72c414f925dff21cb39af4052c401fad71de4242 Mon Sep 17 00:00:00 2001
From: Lukas <61192133+lukasloetkolben@users.noreply.github.com>
Date: Tue, 14 Jan 2025 23:26:39 +0100
Subject: [PATCH] Tesla: add model years and TeslaCarDocs (#1636)

* add model years and TeslaCarDocs

* update years

* clean up

---------

Co-authored-by: Shane Smiskol <shane@smiskol.com>
---
 opendbc/car/tesla/values.py | 38 ++++++++++++++++++++++---------------
 1 file changed, 23 insertions(+), 15 deletions(-)

diff --git a/opendbc/car/tesla/values.py b/opendbc/car/tesla/values.py
index db9880a177..948989c78d 100644
--- a/opendbc/car/tesla/values.py
+++ b/opendbc/car/tesla/values.py
@@ -1,23 +1,31 @@
+from dataclasses import dataclass, field
 from enum import IntFlag
-from opendbc.car.structs import CarParams
-from opendbc.car import Bus, structs
-from opendbc.car import AngleRateLimit, CarSpecs, PlatformConfig, Platforms
+from opendbc.car import Bus, CarSpecs, DbcDict,  PlatformConfig, Platforms, AngleRateLimit
+from opendbc.car.structs import CarParams, CarState
 from opendbc.car.docs_definitions import CarDocs
 from opendbc.car.fw_query_definitions import FwQueryConfig, Request, StdQueries
 
 Ecu = CarParams.Ecu
 
 
+@dataclass
+class TeslaCarDocs(CarDocs):
+  package: str = "Traffic Aware Cruise Control"
+
+
+@dataclass
+class TeslaPlatformConfig(PlatformConfig):
+  dbc_dict: DbcDict = field(default_factory=lambda: {Bus.party: 'tesla_model3_party'})
+
+
 class CAR(Platforms):
-  TESLA_MODEL_3 = PlatformConfig(
-    [CarDocs("Tesla Model 3", "All")],
+  TESLA_MODEL_3 = TeslaPlatformConfig(
+    [TeslaCarDocs("Tesla Model 3 2019-24")],
     CarSpecs(mass=1899., wheelbase=2.875, steerRatio=12.0),
-    {Bus.party: 'tesla_model3_party'},
   )
-  TESLA_MODEL_Y = PlatformConfig(
-    [CarDocs("Tesla Model Y", "All")],
+  TESLA_MODEL_Y = TeslaPlatformConfig(
+    [TeslaCarDocs("Tesla Model Y 2020-24")],
     CarSpecs(mass=2072., wheelbase=2.890, steerRatio=12.0),
-    {Bus.party: 'tesla_model3_party'},
   )
 
 
@@ -40,12 +48,12 @@ class CANBUS:
 
 
 GEAR_MAP = {
-  "DI_GEAR_INVALID": structs.CarState.GearShifter.unknown,
-  "DI_GEAR_P": structs.CarState.GearShifter.park,
-  "DI_GEAR_R": structs.CarState.GearShifter.reverse,
-  "DI_GEAR_N": structs.CarState.GearShifter.neutral,
-  "DI_GEAR_D": structs.CarState.GearShifter.drive,
-  "DI_GEAR_SNA": structs.CarState.GearShifter.unknown,
+  "DI_GEAR_INVALID": CarState.GearShifter.unknown,
+  "DI_GEAR_P": CarState.GearShifter.park,
+  "DI_GEAR_R": CarState.GearShifter.reverse,
+  "DI_GEAR_N": CarState.GearShifter.neutral,
+  "DI_GEAR_D": CarState.GearShifter.drive,
+  "DI_GEAR_SNA": CarState.GearShifter.unknown,
 }
 
 class CarControllerParams: