missions implementation

Author: snbianco

astroquery/mast/missions.py

@@ -503,8 +503,19 @@ def filter_products(self, products, *, extension=None, **filters):
                 warnings.warn(f"Column '{colname}' not found in product table.", InputWarning)
                 continue
 
-            vals = [vals] if isinstance(vals, str) else vals
-            col_mask = np.isin(products[colname], vals)
+            col_data = products[colname]
+            # If the column is an integer or float, treat differently
+            if col_data.dtype.kind in 'if' and isinstance(vals, str):
+                try:
+                    col_mask = utils.parse_numeric_product_filter(vals)(col_data)
+                except ValueError:
+                    warnings.warn(f"Could not parse numeric filter '{vals}' for column '{colname}'.", InputWarning)
+                    continue
+            else:
+                if isinstance(vals, str):
+                    vals = [vals]
+                col_mask = np.isin(col_data, vals)
+
             filter_mask &= col_mask
 
         # Return filtered products

astroquery/mast/utils.py

@@ -6,12 +6,12 @@
 Miscellaneous functions used throughout the MAST module.
 """
 
+import re
 import warnings
-import numpy as np
 
+import numpy as np
 import requests
 import platform
-
 from astropy.coordinates import SkyCoord
 from astropy.table import Table
 from astropy import units as u
@@ -345,3 +345,36 @@ def remove_duplicate_products(data_products, uri_key):
                  f"Only returning {len(unique_products)} unique product(s).")
 
     return unique_products
+
+
+def parse_numeric_product_filter(val):
+    """
+    Parses a numeric product filter value and returns a function that can be used to filter
+    a column of a product table.
+
+    Parameters
+    ----------
+    val : str
+        The filter value as a string. It can be a single number, a range in the form of "start..end",
+        or a comparison operator followed by a number (e.g., ">=10", "<5", ">100.5", etc.).
+
+    Returns
+    -------
+    response : function
+        A function that takes a column of a product table and returns a boolean mask indicating
+        which rows satisfy the filter condition.
+    """
+    range_pattern = re.compile(r'[+-]?(\d+(\.\d*)?|\.\d+)\.\.[+-]?(\d+(\.\d*)?|\.\d+)')
+    if val.startswith('>='):
+        return lambda col: col >= float(val[2:])
+    elif val.startswith('<='):
+        return lambda col: col <= float(val[2:])
+    elif val.startswith('>'):
+        return lambda col: col > float(val[1:])
+    elif val.startswith('<'):
+        return lambda col: col < float(val[1:])
+    elif range_pattern.fullmatch(val):
+        start, end = map(float, val.split('..'))
+        return lambda col: (col >= start) & (col <= end)
+    else:
+        return lambda col: col == float(val)