Add process column to Session table to record if processing is requested or not

Author: d-j-hatton

src/murfey/server/api/__init__.py

@@ -1973,6 +1973,17 @@ def create_session(instrument_name: str, visit: str, name: str, db=murfey_db) ->
     return sid
 
 
+@router.post("/sessions/{session_id}")
+def update_session(
+    session_id: MurfeySessionID, process: bool = True, db=murfey_db
+) -> None:
+    session = db.exec(select(Session).where(session_id == session_id)).one()
+    session.process = process
+    db.add(session)
+    db.commit()
+    return None
+
+
 @router.put("/sessions/{session_id}/current_gain_ref")
 def update_current_gain_ref(
     session_id: MurfeySessionID, new_gain_ref: CurrentGainRef, db=murfey_db

src/murfey/server/api/processing_parameters.py

@@ -1,6 +1,7 @@
 from logging import getLogger
 from typing import Optional
 
+import sqlalchemy
 from fastapi import APIRouter, Depends
 from pydantic import BaseModel
 from sqlmodel import Session, select
@@ -24,12 +25,15 @@ class EditableSessionProcessingParameters(BaseModel):
 @router.get("/sessions/{session_id}/session_processing_parameters")
 def get_session_processing_parameters(
     session_id: MurfeySessionID, db: Session = murfey_db
-) -> EditableSessionProcessingParameters:
-    proc_params = db.exec(
-        select(SessionProcessingParameters).where(
-            SessionProcessingParameters.session_id == session_id
-        )
-    ).one()
+) -> Optional[EditableSessionProcessingParameters]:
+    try:
+        proc_params = db.exec(
+            select(SessionProcessingParameters).where(
+                SessionProcessingParameters.session_id == session_id
+            )
+        ).one()
+    except sqlalchemy.exc.NoResultFound:
+        return None
     return EditableSessionProcessingParameters(
         gain_ref=proc_params.gain_ref,
         dose_per_frame=proc_params.dose_per_frame,

src/murfey/util/db.py

@@ -48,6 +48,7 @@ class Session(SQLModel, table=True):  # type: ignore
     started: bool = Field(default=False)
     current_gain_ref: str = Field(default="")
     instrument_name: str = Field(default="")
+    process: bool = Field(default=True)
 
     # CLEM Workflow