Files
dragonpilot/release/create_casync_agnos_release.py
Justin Newberry 44bb4eb521 add tool to create casync agnos updates from existing manifest (#32124)
* wip

* wip

* add agnos prefix

* add mkdir

* tool to create agnos updates
2024-04-08 11:34:05 -07:00

53 lines
1.6 KiB
Python

import argparse
import json
import pathlib
import tempfile
from openpilot.common.basedir import BASEDIR
from openpilot.system.hardware.tici.agnos import StreamingDecompressor
from openpilot.system.updated.casync.common import create_casync_from_file
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="creates a casync release")
parser.add_argument("output_dir", type=str, help="output directory for the channel")
parser.add_argument("version", type=str, help="version of agnos this is")
parser.add_argument("--manifest", type=str, help="json manifest to create agnos release from", \
default=str(pathlib.Path(BASEDIR) / "system/hardware/tici/agnos.json"))
args = parser.parse_args()
output_dir = pathlib.Path(args.output_dir)
output_dir.mkdir(parents=True, exist_ok=True)
manifest_file = pathlib.Path(args.manifest)
with tempfile.NamedTemporaryFile() as entry_file:
entry_path = pathlib.Path(entry_file.name)
with open(manifest_file) as f:
manifest = json.load(f)
for entry in manifest:
print(f"creating casync agnos build from {entry}")
downloader = StreamingDecompressor(entry['url'])
CHUNK_SIZE=16*1024*1024 # 16mb
size = entry["size"]
cur = 0
with open(entry_path, "wb") as f:
while True:
to_read = min((size - cur), CHUNK_SIZE)
print(f"{cur/size*100:06.2f}% {to_read}")
if not to_read:
break
f.write(downloader.read(to_read))
cur += to_read
create_casync_from_file(entry_path, output_dir, f"agnos-{args.version}-{entry['name']}")