Update download script

fontanf · Apr 1, 2024 · c37c3be · c37c3be
1 parent 7459ddc
commit c37c3be
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/scripts/download_data.py b/scripts/download_data.py
@@ -2,10 +2,12 @@
 import os
 import pathlib
 
-gdown.download(id="1RzBH4LWSqHGdE-0SIpZZCluV6MLZ6b4f")
-os.system("7z x star_observation_scheduling.7z -odata")
-pathlib.Path("star_observation_scheduling.7z").unlink()
 
-gdown.download(id="1ag21EFO0VzZlUccjTSVJ9tlT0uh7DbkS")
-os.system("7z x flexible_star_observation_scheduling.7z -odata")
-pathlib.Path("flexible_star_observation_scheduling.7z").unlink()
+def download(id):
+    gdown.download(id=id, output="data.7z")
+    os.system("7z x data.7z -odata")
+    pathlib.Path("data.7z").unlink()
+
+
+download("1RzBH4LWSqHGdE-0SIpZZCluV6MLZ6b4f")
+download("1ag21EFO0VzZlUccjTSVJ9tlT0uh7DbkS")