obtain survey/program from headers if necessary

desihub · Dec 10, 2024 · 1d55bc9 · 1d55bc9
1 parent 59be658
commit 1d55bc9
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/py/specprodDB/load.py b/py/specprodDB/load.py
@@ -1024,6 +1024,9 @@ def convert(cls, data, survey=None, program=None, row_index=None):
             if check_columns[column] is None:
                 if column.upper() in data.colnames:
                     log.info("Obtaining '%s' from input data file.", column)
+                elif column.upper() in data.meta:
+                    log.info("Obtaining '%s' from input data header.", column)
+                    default_columns[column] = data.meta[column.upper()]
                 else:
                     msg = "Could not obtain '%s' from input data file."
                     log.critical(msg, column)
@@ -1218,7 +1221,10 @@ def convert(cls, data, survey=None, program=None, tileid=None, night=None,
         for column in check_columns:
             if check_columns[column] is None:
                 if column.upper() in data.colnames:
-                    log.info("Obtaining '%s' from input data file.", column)
+                    log.info("Obtaining '%s' from input data table.", column)
+                elif column.upper() in data.meta:
+                    log.info("Obtaining '%s' from input data header.", column)
+                    default_columns[column] = data.meta[column.upper()]
                 else:
                     msg = "Could not obtain '%s' from input data file."
                     log.critical(msg, column)