forked from IQ.Lvbs/IQ.Pilot
114 lines
4.6 KiB
Python
Executable File
114 lines
4.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import re
|
|
import os
|
|
import jinja2
|
|
import argparse
|
|
import unicodedata
|
|
from typing import get_args
|
|
|
|
from enum import Enum
|
|
from collections import defaultdict
|
|
|
|
from iqdbc.car.common.basedir import BASEDIR
|
|
from iqdbc.car import gen_empty_fingerprint
|
|
from iqdbc.car.structs import CarParams, IQCarParams
|
|
from iqdbc.car.docs_definitions import BaseCarHarness, CarDocs, Device, ExtraCarDocs, Column, ExtraCarsColumn, CommonFootnote, PartType, SupportType
|
|
from iqdbc.car.car_helpers import interfaces
|
|
from iqdbc.car.interfaces import get_interface_attr
|
|
from iqdbc.car.values import Platform
|
|
from iqdbc.car.mock.values import CAR as MOCK
|
|
from iqdbc.car.extra_cars import CAR as EXTRA
|
|
|
|
|
|
EXTRA_CARS_MD_OUT = os.path.join(BASEDIR, "../", "../", "docs", "CARS.md")
|
|
|
|
# TODO: merge these platforms into normal car ports with SupportType flag
|
|
ExtraPlatform = Platform | EXTRA
|
|
EXTRA_BRANDS = get_args(ExtraPlatform)
|
|
EXTRA_PLATFORMS: dict[str, ExtraPlatform] = {str(platform): platform for brand in EXTRA_BRANDS for platform in brand}
|
|
|
|
|
|
def get_params_for_docs(platform) -> tuple[CarParams, IQCarParams]:
|
|
cp_platform = platform if platform in interfaces else MOCK.MOCK
|
|
CP: CarParams = interfaces[cp_platform].get_params(cp_platform, fingerprint=gen_empty_fingerprint(),
|
|
car_fw=[CarParams.CarFw(ecu=CarParams.Ecu.unknown)],
|
|
alpha_long=True, is_release=True, docs=True)
|
|
|
|
CP_IQ: IQCarParams = interfaces[cp_platform].get_params_iq(CP, cp_platform, fingerprint=gen_empty_fingerprint(),
|
|
car_fw=[CarParams.CarFw(ecu=CarParams.Ecu.unknown)],
|
|
alpha_long=True, is_release_iq=True, docs=True)
|
|
return CP, CP_IQ
|
|
|
|
|
|
def get_all_footnotes() -> dict[Enum, int]:
|
|
all_footnotes = list(CommonFootnote)
|
|
for footnotes in get_interface_attr("Footnote", ignore_none=True).values():
|
|
all_footnotes.extend(footnotes)
|
|
return {fn: idx + 1 for idx, fn in enumerate(all_footnotes)}
|
|
|
|
|
|
def _natural_sort_key(s):
|
|
# NFKD normalization ensures accented characters sort with their base letter (e.g., Š sorts with S)
|
|
normalized = unicodedata.normalize('NFKD', s)
|
|
return [int(t) if t.isdigit() else t.lower() for t in re.split(r'(\d+)', normalized) if t]
|
|
|
|
|
|
def build_sorted_car_docs_list(platforms, footnotes=None):
|
|
collected_car_docs: list[CarDocs | ExtraCarDocs] = []
|
|
for platform in platforms.values():
|
|
car_docs = platform.config.car_docs
|
|
CP, CP_IQ = get_params_for_docs(platform)
|
|
|
|
if not len(car_docs):
|
|
continue
|
|
|
|
# A platform can include multiple car models
|
|
for _car_docs in car_docs:
|
|
if not hasattr(_car_docs, "row"):
|
|
_car_docs.init_make(CP)
|
|
_car_docs.init(CP, footnotes)
|
|
collected_car_docs.append(_car_docs)
|
|
|
|
# Sort cars by make and model + year
|
|
sorted_cars = sorted(collected_car_docs, key=lambda car: _natural_sort_key(car.name))
|
|
return sorted_cars
|
|
|
|
|
|
# CAUTION: This function is imported by shop.comma.ai and comma.ai/vehicles, test changes carefully
|
|
def get_all_car_docs() -> list[CarDocs]:
|
|
collected_footnotes = get_all_footnotes()
|
|
sorted_list: list[CarDocs] = build_sorted_car_docs_list(EXTRA_PLATFORMS, footnotes=collected_footnotes)
|
|
return sorted_list
|
|
|
|
|
|
def group_by_make(all_car_docs: list[CarDocs]) -> dict[str, list[CarDocs]]:
|
|
sorted_car_docs = defaultdict(list)
|
|
for car_docs in all_car_docs:
|
|
sorted_car_docs[car_docs.make].append(car_docs)
|
|
return dict(sorted_car_docs)
|
|
|
|
|
|
def generate_cars_md(all_car_docs: list[CarDocs], template_fn: str, **kwargs) -> str:
|
|
with open(template_fn) as f:
|
|
template = jinja2.Template(f.read(), trim_blocks=True, lstrip_blocks=True)
|
|
|
|
footnotes = [fn.value.text for fn in get_all_footnotes()]
|
|
cars_md: str = template.render(all_car_docs=all_car_docs, PartType=PartType,
|
|
group_by_make=group_by_make, footnotes=footnotes,
|
|
Device=Device, Column=Column, ExtraCarsColumn=ExtraCarsColumn,
|
|
BaseCarHarness=BaseCarHarness, SupportType=SupportType,
|
|
**kwargs)
|
|
return cars_md
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Auto generates supportability info docs for all known cars",
|
|
formatter_class=argparse.ArgumentDefaultsHelpFormatter)
|
|
|
|
parser.add_argument("--out", default=EXTRA_CARS_MD_OUT, help="Override default generated filename")
|
|
args = parser.parse_args()
|
|
|
|
with open(args.out, 'w') as f:
|
|
f.write(generate_cars_md(get_all_car_docs(), args.template))
|
|
print(f"Generated and written to {args.out}")
|