Fix/ap returns ra dec

datalab/datalab_session/analysis/target_position.py

@@ -0,0 +1,27 @@
+from urllib.error import HTTPError
+
+from django.contrib.auth.models import User
+
+from datalab.datalab_session.exceptions import ClientAlertException
+from datalab.datalab_session.utils.file_utils import get_fits_header
+from datalab.datalab_session.utils.filecache import FileCache
+from datalab.datalab_session.utils.fits_metadata import target_radec_from_header
+
+
+def target_position(input: dict, user: User):
+  try:
+    file_path = FileCache().get_fits(input['basename'], input.get('source', 'archive'), user)
+    sci_header = get_fits_header(file_path, 'SCI')
+  except TimeoutError:
+    raise ClientAlertException(f"Download of {input['basename']} FITs timed out")
+  except TypeError as e:
+    raise ClientAlertException(e)
+  except HTTPError:
+    raise ClientAlertException(f"No FITs file found for this image")
+
+  try:
+    ra, dec = target_radec_from_header(sci_header)
+  except ValueError as e:
+    raise ClientAlertException(e)
+
+  return {'ra': ra, 'dec': dec}

datalab/datalab_session/tests/test_analysis.py

@@ -10,7 +10,7 @@
 import numpy as np
 from numpy.testing import assert_almost_equal
 
-from datalab.datalab_session.analysis import centroiding, line_profile, source_catalog
+from datalab.datalab_session.analysis import centroiding, line_profile, source_catalog, target_position
 from datalab.datalab_session.data_operations import light_curve as light_curve_module
 
 
@@ -76,6 +76,20 @@ def test_source_catalog(self, mock_file_cache):
         self.assertAlmostEqual(output[0]['mag'], expected_mag)
         self.assertAlmostEqual(output[0]['magerr'], expected_magerr)
 
+    @mock.patch('datalab.datalab_session.analysis.target_position.get_fits_header')
+    @mock.patch('datalab.datalab_session.analysis.target_position.FileCache')
+    def test_target_position(self, mock_file_cache, mock_get_fits_header):
+        mock_file_cache.return_value.get_fits.return_value = self.analysis_fits_1_path
+        mock_get_fits_header.return_value = {"CAT-RA": "18:07:23.824", "CAT-DEC": "-07:28:36.39"}
+
+        output = target_position.target_position({
+            'basename': 'fits_1',
+            'source': 'archive'
+        }, None)
+
+        self.assertAlmostEqual(output['ra'], 271.84927, places=4)
+        self.assertAlmostEqual(output['dec'], -7.47677, places=4)
+
     @staticmethod
     def write_catalog_without_radec(path, ra, dec, celestial_wcs=True):
         """ Writes a fits whose CAT HDU has no ra/dec columns, so source_catalog has to derive

datalab/datalab_session/views.py

@@ -12,6 +12,7 @@
 from datalab.datalab_session.analysis.get_jpg import get_jpg
 from datalab.datalab_session.analysis.raw_data import raw_data
 from datalab.datalab_session.analysis.wcs import wcs
+from datalab.datalab_session.analysis.target_position import target_position
 from datalab.datalab_session.exceptions import ClientAlertException
 
 log = logging.getLogger()
@@ -38,7 +39,8 @@ class AnalysisView(RetrieveAPIView):
         "get-tif": get_tif,
         "get-jpg": get_jpg,
         "raw-data": raw_data,
-        "wcs": wcs
+        "wcs": wcs,
+        "target-position": target_position,
     }
 
     def post(self, request, action):