adding a QC status message and bumped version number

TEM_comms/__init__.py

@@ -5,6 +5,7 @@
 from . import tile
 from . import ui
 from . import montage
+from . import qc
 
 
 topics = {
@@ -34,4 +35,5 @@
     "ui.setup": ui.Setup,
     "montage.start": montage.Start,
     "montage.finished": montage.Finished,
+    "qc.status": qc.Status,
 }

TEM_comms/qc.py

@@ -0,0 +1,6 @@
+from pigeon import BaseMessage
+from typing import Literal
+
+
+class Status(BaseMessage):
+    state: Literal["GOOD", "STOP_AT_END", "STOP_NOW"]

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "pigeon-tem-comms"
-version = "0.4.0"
+version = "0.4.1"
 authors = [
   { name="Cameron Devine", email="[email protected]" },
 ]

tests/test_qc.py

@@ -0,0 +1,8 @@
+from TEM_comms.qc import Status
+from pydantic import ValidationError
+import pytest
+
+
+def test_bad_status():
+    with pytest.raises(ValidationError):
+        Status(state="BAD")