Files
StarPilot/scripts/lint/check_added_large_files.py
T
Danny c777a9ec8a iOS: Add the Galaxy companion with live device pages and CarPlay
Add LAN and Bluetooth connections, secure pairing, device-matched Galaxy page
caching, phone location and CarPlay navigation. Include connection recovery,
diagnostics, regression tests and a simulator CI build. Add shared companion
docs and host tests, and remove obsolete bundled-web lint exceptions.
2026-10-22 15:37:20 -07:00

77 lines
2.6 KiB
Python
Executable File

#!/usr/bin/env python3
import argparse
import math
import os
import subprocess
LARGE_TEXT_FILES = frozenset((
# shortcut: these existing modules bypass 120 KiB; split when their responsibilities warrant it.
'openpilot/selfdrive/selfdrived/selfdrived.py',
'openpilot/starpilot/galaxy/server.py',
'openpilot/starpilot/galaxy/web/js/onroad-layout.js',
'openpilot/cereal/gen/cpp/car.capnp.c++',
'openpilot/cereal/gen/cpp/car.capnp.h',
'openpilot/cereal/gen/cpp/custom.capnp.c++',
'openpilot/cereal/gen/cpp/custom.capnp.h',
'openpilot/cereal/gen/cpp/deprecated.capnp.c++',
'openpilot/cereal/gen/cpp/deprecated.capnp.h',
'openpilot/cereal/gen/cpp/log.capnp.c++',
'openpilot/cereal/gen/cpp/log.capnp.h',
'openpilot/selfdrive/controls/lib/longitudinal_mpc_lib/c_generated_code/acados_ocp_solver_pyx.c',
'openpilot/starpilot/lateral/tests/fixtures/corolla_tss2_ba901b5f.json',
'openpilot/starpilot/lateral/tests/fixtures/genesis_g70_ba901b5f.json',
'openpilot/starpilot/lateral/tests/fixtures/oct2_turns.json',
))
def binary_files(filenames: list[str]) -> set[str]:
if not filenames:
return set()
result = subprocess.run(
("git", "check-attr", "text", "-z", "--stdin"),
input="\0".join(filenames),
check=True,
capture_output=True,
text=True,
)
fields = result.stdout.rstrip("\0").split("\0") if result.stdout else []
return {fields[i] for i in range(0, len(fields), 3) if fields[i + 2] == "unset"}
def check_added_large_files(filenames: list[str], max_kb: int) -> int:
failed = False
ignored = binary_files(filenames)
for filename in filenames:
with open(filename, 'rb') as file:
if file.read(43).startswith(b'version https://git-lfs.github.com/spec/v1'):
print(f'{filename}: placeholder pointer payload.')
failed = True
continue
if os.stat(filename).st_size > 100 * 1024 * 1024:
print(f'{filename}: ordinary Git blob exceeds 100 MiB.')
failed = True
continue
if filename in ignored or filename in LARGE_TEXT_FILES:
continue
size_kb = math.ceil(os.stat(filename).st_size / 1024)
if size_kb > max_kb:
print(f"{filename} ({size_kb} KB) exceeds {max_kb} KB.")
failed = True
return int(failed)
def main() -> int:
parser = argparse.ArgumentParser(description="Check that tracked files do not exceed a size limit.")
parser.add_argument("filenames", nargs="*")
parser.add_argument("--maxkb", type=int, default=500, help="maximum allowable size in KiB")
args = parser.parse_args()
return check_added_large_files(args.filenames, args.maxkb)
if __name__ == "__main__":
raise SystemExit(main())