spectre-core 0.0.16__py3-none-any.whl → 0.0.18__py3-none-any.whl

spectre_core/config/__init__.py

@@ -6,7 +6,8 @@
 """General `spectre` package configurations."""
 
 from ._paths import (
-    get_spectre_data_dir_path, get_batches_dir_path, get_configs_dir_path, get_logs_dir_path
+    get_spectre_data_dir_path, get_batches_dir_path, get_configs_dir_path, 
+    get_logs_dir_path, trim_spectre_data_dir_path
 )
 from ._time_formats import (
     TimeFormat
@@ -14,5 +15,5 @@ from ._time_formats import (
 
 __all__ = [
     "get_spectre_data_dir_path", "get_batches_dir_path", "get_configs_dir_path", "get_logs_dir_path",
-    "TimeFormat"
+    "TimeFormat", "trim_spectre_data_dir_path"
 ]

spectre_core/config/_paths.py

@@ -115,3 +115,35 @@ def get_configs_dir_path(
     :return: The directory path for configuration files.
     """
     return _CONFIGS_DIR_PATH
+
+
+def trim_spectre_data_dir_path(
+    full_path: str
+) -> str:
+    """Remove the `SPECTRE_DATA_DIR_PATH` prefix from a full file path.
+
+    This function returns the relative path of `full_path` with respect to
+    `SPECTRE_DATA_DIR_PATH`. It is useful for trimming absolute paths
+    to maintain consistency across different environments where the base
+    directory might differ.
+
+    :param full_path: The full file path to be trimmed.
+    :return: The relative path with `SPECTRE_DATA_DIR_PATH` removed.
+    """
+    return os.path.relpath(full_path, _SPECTRE_DATA_DIR_PATH)
+
+
+def add_spectre_data_dir_path(
+    rel_path: str
+) -> str:
+    """Prepend the `SPECTRE_DATA_DIR_PATH` prefix to a relative file path.
+
+    This function constructs an absolute path by joining the given relative
+    path with `SPECTRE_DATA_DIR_PATH`. It is useful for converting stored
+    relative paths back into full paths within the mounted directory.
+
+    :param rel_path: The relative file path to be appended.
+    :return: The full file path prefixed with `SPECTRE_DATA_DIR_PATH`.
+    """
+    return os.path.join(_SPECTRE_DATA_DIR_PATH, rel_path)
+

spectre_core/wgetting/_callisto.py

@@ -8,7 +8,9 @@ import shutil
 import gzip
 from datetime import datetime
 
-from spectre_core.config import get_spectre_data_dir_path, get_batches_dir_path, TimeFormat
+from spectre_core.config import (
+    get_spectre_data_dir_path, get_batches_dir_path, TimeFormat, trim_spectre_data_dir_path
+)
 
 from enum import Enum
 
@@ -129,30 +131,37 @@ def _get_batch_path(
 
 def _unzip_file_to_batches(
     gz_path: str
-) -> None:
+) -> str:
     """
     Decompress a `.fit.gz` file and save it as a `.fits` batch file.
 
     :param gz_path: Path to the `.fit.gz` file.
+    :return: The file path of the newly created batch file, relative to `SPECTRE_DATA_DIR_PATH`.
     """
     fits_path = _get_batch_path(gz_path)
     with gzip.open(gz_path, "rb") as f_in, open(fits_path, "wb") as f_out:
         shutil.copyfileobj(f_in, f_out)
+        return trim_spectre_data_dir_path(f_out.name)
 
 
 def _unzip_to_batches(
     tmp_dir: str
-) -> None:
+) -> list[str]:
     """
     Decompress all `.gz` files in a temporary directory and save them as `spectre`
     batch files.
 
     :param tmp_dir: Path to the temporary directory containing `.gz` files.
+    :return: A list of file names of all newly created batch files, relative to `SPECTRE_DATA_DIR_PATH`.
     """
+    batch_file_names = []
     for entry in os.scandir(tmp_dir):
         if entry.is_file() and entry.name.endswith(".gz"):
-            _unzip_file_to_batches(entry.path)
+            batch_file_names.append( _unzip_file_to_batches(entry.path) )
             os.remove(entry.path)
+    shutil.rmtree(tmp_dir)
+    return batch_file_names
+            
 
 
 def _wget_callisto_data(
@@ -186,7 +195,7 @@ def download_callisto_data(
     year: int, 
     month: int, 
     day: int
-) -> None:
+) -> list[str]:
     """
     Download and decompress e-Callisto FITS files, saving them as `spectre` batch files.
 
@@ -194,6 +203,7 @@ def download_callisto_data(
     :param year: Year of the observation.
     :param month: Month of the observation.
     :param day: Day of the observation.
+    :return: A list of file names of all newly created batch files, relative to `SPECTRE_DATA_DIR_PATH`.
     """
     tmp_dir = os.path.join(get_spectre_data_dir_path(), "tmp")
     # if there are any residual files in the temporary directory, remove them.
@@ -202,5 +212,4 @@ def download_callisto_data(
     os.makedirs(tmp_dir, exist_ok=True)
     
     _wget_callisto_data(instrument_code.value, year, month, day, tmp_dir)
-    _unzip_to_batches(tmp_dir)
-    shutil.rmtree(tmp_dir)
+    return sorted( _unzip_to_batches(tmp_dir) )

{spectre_core-0.0.16.dist-info → spectre_core-0.0.18.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spectre-core
-Version: 0.0.16
+Version: 0.0.18
 Summary: The core Python package used by the spectre program. 
 Maintainer-email: Jimmy Fitzpatrick <jcfitzpatrick12@gmail.com>
 License:                     GNU GENERAL PUBLIC LICENSE

{spectre_core-0.0.16.dist-info → spectre_core-0.0.18.dist-info}/RECORD

@@ -20,8 +20,8 @@ spectre_core/capture_configs/_pconstraints.py,sha256=bnl1m6M9iyo5KOsPKT_arwrrAZb
 spectre_core/capture_configs/_pnames.py,sha256=3bJhT4kM5WuiH9SSs4UyBcq_mDybzx-bRiWJcIOqeLA,1975
 spectre_core/capture_configs/_ptemplates.py,sha256=Pepzs7eq31WPdTYmIICJjIVOzf-j5LdA2SD_f2RUZ54,25609
 spectre_core/capture_configs/_pvalidators.py,sha256=fRlFL3rTIpDU2G4iUKp3Lb_jFX5ddHNuUbY5hTzZQQg,10325
-spectre_core/config/__init__.py,sha256=CVddibAIWU4cNerXCxfeUn_cvnWWDBm4pWeUifqM6Ko,502
-spectre_core/config/_paths.py,sha256=Qn0KVe27P6Bo65MfHe4LpH_ryPvLgbTxKjc4MTw1BB8,4090
+spectre_core/config/__init__.py,sha256=TwQAiDUJLWuwx4IFlStG7Mvb9EFU3u82z5ilPfVH6e0,565
+spectre_core/config/_paths.py,sha256=k9jyPlaFHlM2pAwqPWlpivdc0jCiFFpy4tOny1uxc64,5224
 spectre_core/config/_time_formats.py,sha256=gS0j5zIvBhnV7KMYvTloloIbVwmCYn8MMKn3zNeQ4Xc,857
 spectre_core/jobs/__init__.py,sha256=WKTvxvpciedm6tsKjU02iXJhIdNsMDt-BnMVwVme2Bo,412
 spectre_core/jobs/_jobs.py,sha256=gGpxsLZZ7EdXBYGH-r_pKnRGWSapr78E5SK_VnulaGg,3844
@@ -70,9 +70,9 @@ spectre_core/spectrograms/_array_operations.py,sha256=79vddwWqR5i6OkeD5L_84t8svs
 spectre_core/spectrograms/_spectrogram.py,sha256=WhHEt_QpmzspDqYlzdZcJ8CAXxRfs8-JfP0T3NHpjLQ,28205
 spectre_core/spectrograms/_transform.py,sha256=WZ5jAe3bOpNldxHDSHPf8Q_1ifBdWqXB_mlF6DL1VuE,11734
 spectre_core/wgetting/__init__.py,sha256=UkS0Z0wuuqpoZ1EL35wJcDpjBiAaZgdZ7064yGESxNE,341
-spectre_core/wgetting/_callisto.py,sha256=m9II6ayswrY4h_ST3TodioZ4C865lN_uOF-BozvjZAg,6952
-spectre_core-0.0.16.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-spectre_core-0.0.16.dist-info/METADATA,sha256=7CfIEsTWqo1hw0GvOBc0NKv1o5jIezp0_WcpK5xAbx4,42100
-spectre_core-0.0.16.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-spectre_core-0.0.16.dist-info/top_level.txt,sha256=-UsyjpFohXgZpgcZ9QbVeXhsIyF3Am8RxNFNDV_Ta2Y,13
-spectre_core-0.0.16.dist-info/RECORD,,
+spectre_core/wgetting/_callisto.py,sha256=B2cb1sqF7SGgHfJ2YDOY-6hbJAqQfwK3NH1NS0Jwfxg,7469
+spectre_core-0.0.18.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+spectre_core-0.0.18.dist-info/METADATA,sha256=Agk1kfSTqOqArQD6bCUQmVK8n3g07fXSnguK0MR4d5E,42100
+spectre_core-0.0.18.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+spectre_core-0.0.18.dist-info/top_level.txt,sha256=-UsyjpFohXgZpgcZ9QbVeXhsIyF3Am8RxNFNDV_Ta2Y,13
+spectre_core-0.0.18.dist-info/RECORD,,