Add command too to get the path to the scheduler conf script from the EFD and github

Author: ehneilsen

pyproject.toml

@@ -62,6 +62,7 @@ dev = [
 prenight = "schedview.app.prenight.prenight:main"
 scheduler_dashboard = "schedview.app.scheduler_dashboard.scheduler_dashboard_app:main"
 obs_version_at_time = "schedview.app.obs_version_at_time:version_at_time_cli"
+scheduler_config_at_time = "schedview.app.scheduler_config_at_time:scheduler_config_at_time_cli"
 
 [tool.setuptools.dynamic]
 version = { attr = "setuptools_scm.get_version" }

schedview/app/scheduler_config_at_time.py

@@ -0,0 +1,48 @@
+import argparse
+import asyncio
+
+from astropy.time import Time
+
+import schedview.collect
+
+GITHUB_API_URL_BASE = "https://api.github.com/repos"
+
+
+def scheduler_config_at_time_cli():
+
+    # Get "now" so we can make it the default.
+    now = Time.now()
+    now_isot = now.isot
+    assert isinstance(now.isot, str)
+
+    # Prepare argument parsing
+    parser = argparse.ArgumentParser(
+        prog="scheduler_config_at_time",
+        description="Get the scheduler configuration script path the observatory as of a given time.",
+    )
+    parser.add_argument("instrument", type=str, help="The instrument being scheduled.")
+    parser.add_argument(
+        "--datetime",
+        type=str,
+        default=f"{now_isot[:19]}Z",
+        help="UTC time in ISO 8601 T format: YYYY-MM-DDTHH:mm:SSZ",
+    )
+
+    # Actually parse the arguments.
+    args = parser.parse_args()
+
+    # Do the call
+    time_cut = Time(args.datetime)
+    ts_config_ocs_version = schedview.collect.get_version_at_time("ts_config_ocs", time_cut)
+
+    sal_indexes = schedview.collect.SAL_INDEX_GUESSES[args.instrument]
+    loop = asyncio.get_event_loop()
+    scheduler_config = loop.run_until_complete(
+        schedview.collect.get_scheduler_config(ts_config_ocs_version, sal_indexes, time_cut)
+    )
+
+    print(scheduler_config)
+
+
+if __name__ == "__main__":
+    scheduler_config_at_time_cli()

schedview/collect/base/__init__.py

@@ -2,6 +2,7 @@
 
 from ..efd import (
     SAL_INDEX_GUESSES,
+    get_scheduler_config,
     get_version_at_time,
     make_efd_client,
     make_version_table_for_time,

schedview/collect/efd.py

@@ -6,6 +6,8 @@
 from typing import Literal
 
 import pandas as pd
+import requests
+import yaml
 from astropy.time import Time, TimeDelta
 from lsst_efd_client import EfdClient
 
@@ -298,3 +300,42 @@ def get_version_at_time(item: str, time_cut: Time | None = None, max_age: TimeDe
         )
 
     return version
+
+
+async def get_scheduler_config(
+    ts_config_ocs_version: str, sal_indexes: tuple[int, ...] | None = None, time_cut: Time | None = None
+) -> str:
+
+    latest_config_df = await query_latest_in_efd_topic(
+        topic="lsst.sal.Scheduler.logevent_configurationApplied",
+        fields=["SchedulerId", "configurations", "salIndex", "schemaVersion", "url", "version"],
+        sal_indexes=sal_indexes,
+        time_cut=time_cut,
+        num_records=1,
+    )
+    config_fname = latest_config_df["configurations"].iloc[0].split(",")[-1]
+    schema_version = latest_config_df["schemaVersion"].iloc[0]
+
+    scheduler_config_ocs_url = "/".join(
+        [
+            "https://raw.githubusercontent.com",
+            "lsst-ts",
+            "ts_config_ocs",
+            ts_config_ocs_version,
+            "Scheduler",
+            schema_version,
+            config_fname,
+        ]
+    )
+
+    response = requests.get(scheduler_config_ocs_url, allow_redirects=True)
+    scheduler_config_ocs = yaml.safe_load(response.content.decode("utf-8"))
+
+    scheduler_config = scheduler_config_ocs["auxtel"]["feature_scheduler_driver_configuration"][
+        "scheduler_config"
+    ]
+
+    relative_scheduler_config = scheduler_config[
+        scheduler_config.find("/ts_config_ocs/Scheduler/feature_scheduler") + 1 :
+    ]
+    return relative_scheduler_config

schedview/collect/local/__init__.py

@@ -5,6 +5,7 @@
 
 from ..efd import (
     SAL_INDEX_GUESSES,
+    get_scheduler_config,
     get_version_at_time,
     make_efd_client,
     make_version_table_for_time,

schedview/collect/summit/__init__.py

@@ -2,6 +2,7 @@
 
 from ..efd import (
     SAL_INDEX_GUESSES,
+    get_scheduler_config,
     get_version_at_time,
     make_efd_client,
     make_version_table_for_time,

schedview/collect/tucson/__init__.py

@@ -2,6 +2,7 @@
 
 from ..efd import (
     SAL_INDEX_GUESSES,
+    get_scheduler_config,
     get_version_at_time,
     make_efd_client,
     make_version_table_for_time,

schedview/collect/usdf/__init__.py

@@ -2,6 +2,7 @@
 
 from ..efd import (
     SAL_INDEX_GUESSES,
+    get_scheduler_config,
     get_version_at_time,
     make_efd_client,
     make_version_table_for_time,

schedview/collect/usdfdev/__init__.py

@@ -2,6 +2,7 @@
 
 from ..efd import (
     SAL_INDEX_GUESSES,
+    get_scheduler_config,
     get_version_at_time,
     make_efd_client,
     make_version_table_for_time,