ensure directory is present

lightkurve · Jul 11, 2024 · c24ae32 · c24ae32
1 parent 93aefd1
commit c24ae32
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,4 @@
-src/lkprf/data
+src/lkprf/data/*.fits
 sandbox
 
 # Byte-compiled / optimized / DLL files

diff --git a/src/lkprf/data.py b/src/lkprf/data.py
@@ -54,6 +54,8 @@ def _download_file(url, file_path):
         try:
             response = requests.get(url)
             response.raise_for_status()  # Ensure the request was successful
+            if not os.path.isdir('/'.join(file_path.split('/')[:-1])):
+                os.mkdir('/'.join(file_path.split('/')[:-1]))
             with open(file_path, "wb") as f:
                 f.write(response.content)
         except requests.exceptions.HTTPError as http_err: