From a0896655157ce21d56d42fdf96fef8f94037245c Mon Sep 17 00:00:00 2001 From: Ell Bogat Date: Wed, 4 Dec 2024 14:59:18 -0500 Subject: [PATCH] add mast download for specific file suffixes --- spaceKLIP/psflib.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/spaceKLIP/psflib.py b/spaceKLIP/psflib.py index 0bdfc78..57a76e0 100644 --- a/spaceKLIP/psflib.py +++ b/spaceKLIP/psflib.py @@ -424,9 +424,24 @@ def get_sciref_files(sci_target, refdb, idir=None, def download_mast(ref_db,token=None, overwrite=False,exists_ok=True, progress=False, verbose=False, + suffix=None, # e.g. 'calints' base_dir=os.path.join('DATA','MAST_DOWNLOAD')): - for fname in list(ref_db.FILENAME): + fnames = list(ref_db.FILENAME) + + # Update file suffix if provided + if not suffix == None: + new_suffix = suffix.strip('_') + + for ff,fname in enumerate(fnames): + fname_split = fname.split('_') + new_fname = '_'.join(fname_split[:-1]) + f'_{new_suffix}.fits' + + fnames[ff] = new_fname + + # Download each file + for fname in fnames: + mast.get_mast_filename(fname, outputdir=base_dir, overwrite=overwrite, exists_ok=exists_ok,