Merge pull request #3010 from esdc-esac-esa-int/ESA_ehst_fix_get_datalabs

EHSTPCR-1181: fix for get_datalabs_path

Author: bsipocz

CHANGES.rst

@@ -219,7 +219,7 @@ esa.hubble
 
 - New method ``get_datalabs_path`` to return the complete path of a file in
   datalabs by combining the datalabs volume path with the path of the file
-  in the table ehst.artifact [#2998]
+  in the table ehst.artifact [#2998, #3010]
 
 esa.jwst
 ^^^^^^^^

astroquery/esa/hubble/core.py

@@ -1018,6 +1018,10 @@ def get_datalabs_path(self, filename, default_volume=None):
         The complete path of the file name in Datalabs
         """
 
+        # FITS files are always compressed
+        if filename.endswith('.fits'):
+            filename = f"{filename}.gz"
+
         query = f"select file_path from ehst.artifact where file_name = '{filename}'"
         job = self.query_tap(query=query)
         if job is None:

astroquery/esa/hubble/tests/test_esa_hubble_remote.py

@@ -130,6 +130,10 @@ def test_retrieve_fits_from_program(self):
                                                folder=str(self.temp_folder_for_fits.name))
         assert len(os.listdir(self.temp_folder_for_fits.name)) > 0
 
-    def test_get_datalabs_path(self):
+    def test_get_datalabs_path_image(self):
         result = esa_hubble.get_datalabs_path(filename='ib4x04ivq_flt.jpg', default_volume=None)
         assert result == '/data/user/hub_hstdata_i/i/b4x/04/ib4x04ivq_flt.jpg'
+
+    def test_get_datalabs_path_fits(self):
+        result = esa_hubble.get_datalabs_path(filename='ib4x04ivq_flt.fits', default_volume=None)
+        assert result == '/data/user/hub_hstdata_i/i/b4x/04/ib4x04ivq_flt.fits.gz'