mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-11-08 13:33:42 +08:00
firestar5683
54 lines
1.6 KiB
Python
Executable File
54 lines
1.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
import os
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
|
|
HERE = os.path.abspath(os.path.dirname(__file__))
|
|
ROOT = os.path.abspath(os.path.join(HERE, "../.."))
|
|
|
|
blacklist = [
|
|
".git/",
|
|
".venv/",
|
|
r"^\.github/workflows/",
|
|
|
|
"matlab.*.md",
|
|
|
|
# Obsolete checkout metadata is never packaged
|
|
".lfsconfig",
|
|
".git$",
|
|
".gitmodules",
|
|
]
|
|
|
|
# Dependency sources are ordinary tracked files. A gitlink cannot be packaged
|
|
# without a separate checkout, so fail rather than silently omitting its source.
|
|
def release_files(root: str, include_big_model: bool = False) -> list[bytes]:
|
|
entries = subprocess.check_output(["git", "ls-files", "--stage", "-z"], cwd=root).split(b"\0")
|
|
files = []
|
|
for entry in entries:
|
|
if not entry:
|
|
continue
|
|
|
|
metadata, tracked_file = entry.split(b"\t", 1)
|
|
mode, _, stage = metadata.split()
|
|
rf = os.fsdecode(tracked_file)
|
|
if mode == b"160000":
|
|
raise ValueError(f"Cannot package gitlink: {rf}; dependency sources must be tracked files")
|
|
if stage != b"0":
|
|
raise ValueError(f"Cannot package unresolved merge: {rf}")
|
|
if rf == "openpilot/selfdrive/modeld/models/big_driving_tinygrad.pkl":
|
|
continue
|
|
if not include_big_model and rf.startswith("openpilot/selfdrive/modeld/models/big_"):
|
|
continue
|
|
blacklisted = any(re.search(p, rf) for p in blacklist)
|
|
if blacklisted:
|
|
continue
|
|
|
|
files.append(tracked_file)
|
|
return files
|
|
|
|
|
|
if __name__ == "__main__":
|
|
for tracked_file in release_files(ROOT, bool(os.getenv("INCLUDE_BIG_MODEL"))):
|
|
sys.stdout.buffer.write(tracked_file + b"\0")
|