Move pubsub module from GORT and add tests (#10)

* Move pubsub module from GORT and add tests

* Update changelog

* Remove use of ReprEnum which is only supported in 3.13

* Install RabbitMQ in the test workflow

* Another attempt at fixing the StrEnum issue

* Fix call to read_input_registers

* Import Self from typing_extensions

* Remove pinning of numpy

* One more attempt to fix the enum issue

.github/workflows/test.yml

@@ -32,6 +32,11 @@ jobs:
         with:
           enable-cache: true
 
+      - name: Setup RabbitMQ
+        continue-on-error: true
+        run: |
+          sudo apt-get install --assume-yes rabbitmq-server
+
       - name: Install dependencies
         run: |
           uv sync --no-dev --frozen --extra schedule
@@ -40,6 +45,9 @@ jobs:
         run: |
           uv pip install pytest pytest-mock pytest-asyncio pytest-cov
           uv run pytest
+        env:
+          PYTEST_RABBITMQ_CTL: '/usr/lib/rabbitmq/bin/rabbitmqctl'
+          PYTEST_RABBITMQ_SERVER: '/usr/lib/rabbitmq/bin/rabbitmq-server'
 
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v5

CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## Next version
+
+### 🚀 New
+
+* [#10](https://vscode.dev/github/sdss/lvmopstools/pull/10) Added a `pubsub` module with tools to emit and subscribe to events using RabbitMQ.
+
+
 ## 0.4.4 - December 5, 2024
 
 ### ✨ Improved

pyproject.toml

@@ -17,6 +17,9 @@ dependencies = [
     "httpx>=0.27.2",
     "polars>=1.13.0",
     "typing-extensions>=4.12.2",
+    "aio-pika>=9.5.3",
+    "pydantic>=2.10.3",
+    "strenum>=0.4.15",
 ]
 
 [project.optional-dependencies]
@@ -61,6 +64,7 @@ dev-dependencies = [
     "sphinx-autobuild>=2021.3.14",
     "sphinx-autodoc-typehints>=1.23.2",
     "ruff>=0.6.1",
+    "pytest-rabbitmq>=3.1.1",
 ]
 
 [tool.ruff]

src/lvmopstools/data/config.yaml

@@ -4,6 +4,11 @@ rabbitmq:
 
 api: http://10.8.38.21:8090/api
 
+pubsub:
+  connection_string: amqp://guest:guest@localhost:5672
+  exchange_name: lvmops
+  routing_key: data
+
 devices:
   thermistors:
     host: 10.8.38.180

src/lvmopstools/devices/ion.py

@@ -11,7 +11,7 @@
 import asyncio
 import warnings
 
-from typing import TypedDict
+from typing import TypedDict, cast
 
 from drift import Drift
 from drift.convert import data_to_float32
@@ -61,10 +61,12 @@ async def _read_one_ion_controller(ion_config: dict) -> dict[str, IonPumpDict]:
             signal_address = camera_config["signal_address"]
             # on_off_address = camera_config["on_off_address"]
 
-            signal = await drift.client.read_input_registers(signal_address, 2)
-            # onoff = await drift.client.read_input_registers(on_off_address, 1)
+            signal = await drift.client.read_input_registers(signal_address, count=2)
+            # onoff = await drift.client.read_input_registers(on_off_address, count=1)
 
-            diff_volt = data_to_float32(tuple(signal.registers))
+            registers = cast(tuple[int, int], tuple(signal.registers))
+
+            diff_volt = data_to_float32(registers)
             pressure = convert_pressure(diff_volt)
 
             # onoff_status = bool(onoff.registers[0])