Faster NEOS download and auto removal of NEOS image after installation (

#478)
commaai · Dec 21, 2018 · 4610a6e · 4610a6e
1 parent 067f1bf
commit 4610a6e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/installer/updater/updater b/installer/updater/updater
diff --git a/selfdrive/manager.py b/selfdrive/manager.py
@@ -41,9 +41,9 @@ def unblock_stdout():
     os._exit(os.wait()[1])
 
 if __name__ == "__main__":
-  if os.path.isfile("/init.qcom.rc") \
-      and (not os.path.isfile("/VERSION") or int(open("/VERSION").read()) < 8):
-
+  neos_update_required = os.path.isfile("/init.qcom.rc") \
+    and (not os.path.isfile("/VERSION") or int(open("/VERSION").read()) < 8)
+  if neos_update_required:
     # update continue.sh before updating NEOS
     if os.path.isfile(os.path.join(BASEDIR, "scripts", "continue.sh")):
       from shutil import copyfile
@@ -54,6 +54,9 @@ def unblock_stdout():
     subprocess.check_call(["git", "clean", "-xdf"], cwd=BASEDIR)
     os.system(os.path.join(BASEDIR, "installer", "updater", "updater"))
     raise Exception("NEOS outdated")
+  elif os.path.isdir("/data/neoupdate"):
+    from shutil import rmtree
+    rmtree("/data/neoupdate")
 
   unblock_stdout()