Add overwatcher switch

pyproject.toml

@@ -42,6 +42,7 @@ python-multipart = "^0.0.6"
 astropy = "^6.0.0"
 astroplan = "^0.9.1"
 polars = "^0.20.16"
+redis = {version = "^5.0.3", extras = ["hiredis"]}
 
 [tool.poetry.group.dev.dependencies]
 ipython = ">=8.0.0"

src/lvmapi/app.py

@@ -11,7 +11,7 @@
 from fastapi import FastAPI
 
 from lvmapi import auth
-from lvmapi.routers import ephemeris, slack, spectrographs, telescopes
+from lvmapi.routers import ephemeris, overwatcher, slack, spectrographs, telescopes
 
 
 app = FastAPI()
@@ -20,6 +20,7 @@
 app.include_router(spectrographs.router)
 app.include_router(slack.router)
 app.include_router(ephemeris.router)
+app.include_router(overwatcher.router)
 
 
 @app.get("/")

src/lvmapi/routers/overwatcher.py

@@ -0,0 +1,48 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# @Author: José Sánchez-Gallego (gallegoj@uw.edu)
+# @Date: 2024-03-25
+# @Filename: overwatcher.py
+# @License: BSD 3-clause (http://www.opensource.org/licenses/BSD-3-Clause)
+
+from __future__ import annotations
+
+from fastapi import APIRouter
+
+from lvmapi.tools import get_redis_connection
+
+
+router = APIRouter(prefix="/overwatcher", tags=["overwatcher"])
+
+
+@router.get("/")
+async def get_overwatcher():
+    """Not implemented."""
+
+    return {}
+
+
+@router.get("/enabled", description="Is the overwatcher enabled?")
+def get_overwatcher_enabled() -> bool:
+    """Returns whether the overwatcher is enabled."""
+
+    redis = get_redis_connection()
+
+    enabled = redis.get("overwatcher:enabled")
+    assert enabled is None or isinstance(enabled, str)
+
+    if enabled is None:
+        return False
+
+    return bool(int(enabled))
+
+
+@router.put("/enabled/{enabled}", description="Enable or disable the overwatcher")
+def put_overwatcher_enabled(enabled: bool) -> bool:
+    """Enables or disables the overwatcher."""
+
+    redis = get_redis_connection()
+    redis.set("overwatcher:enabled", int(enabled))
+
+    return enabled

src/lvmapi/tools/__init__.py

@@ -10,4 +10,6 @@
 
 from .influxdb import *
 from .rabbitmq import *
+from .redis import *
+from .schedule import *
 from .spectrograph import *

src/lvmapi/tools/redis.py

@@ -0,0 +1,20 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# @Author: José Sánchez-Gallego (gallegoj@uw.edu)
+# @Date: 2024-03-25
+# @Filename: redis.py
+# @License: BSD 3-clause (http://www.opensource.org/licenses/BSD-3-Clause)
+
+from __future__ import annotations
+
+import redis
+
+
+__all__ = ["get_redis_connection"]
+
+
+def get_redis_connection() -> redis.Redis:
+    """Returns a connection to the Redis server."""
+
+    return redis.Redis(host="10.8.38.26", port=6379, db=0, decode_responses=True)