Skip to content

Commit 421c0d5

Browse files
authored
Include GET endpoint to get session processing parameters (#536)
For the web UI
1 parent 886361e commit 421c0d5

File tree

3 files changed

+39
-1
lines changed

3 files changed

+39
-1
lines changed
Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
from logging import getLogger
2+
3+
from fastapi import APIRouter
4+
from pydantic import BaseModel
5+
from sqlmodel import Session, select
6+
7+
from murfey.server.murfey_db import murfey_db
8+
from murfey.util.db import SessionProcessingParameters
9+
10+
logger = getLogger("murfey.server.api.processing_parameters")
11+
12+
router = APIRouter()
13+
14+
15+
class EditableSessionProcessingParameters(BaseModel):
16+
gain_ref: str
17+
dose_per_frame: float
18+
eer_fractionation_file: str
19+
symmetry: str
20+
21+
22+
@router.get("sessions/{session_id}/session_processing_parameters")
23+
def get_session_processing_parameters(
24+
session_id: int, db: Session = murfey_db
25+
) -> EditableSessionProcessingParameters:
26+
proc_params = db.exec(
27+
select(SessionProcessingParameters).where(
28+
SessionProcessingParameters.session_id == session_id
29+
)
30+
).one()
31+
return EditableSessionProcessingParameters(
32+
gain_ref=proc_params.gain_ref,
33+
dose_per_frame=proc_params.dose_per_frame,
34+
eer_fractionation_file=proc_params.eer_fractionation_file,
35+
symmetry=proc_params.symmetry,
36+
)

src/murfey/server/demo_api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,6 @@
2828
_flush_grid_square_records,
2929
_flush_tomography_preprocessing,
3030
_murfey_id,
31-
_register_picked_particles_use_diameter,
3231
feedback_callback,
3332
get_hostname,
3433
get_microscope,
@@ -91,6 +90,7 @@
9190
)
9291
from murfey.util.processing_params import default_spa_parameters
9392
from murfey.util.state import global_state
93+
from murfey.workflows.spa.picking import _register_picked_particles_use_diameter
9494

9595
log = logging.getLogger("murfey.server.demo_api")
9696

src/murfey/server/main.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
import murfey.server.api.display
1818
import murfey.server.api.hub
1919
import murfey.server.api.instrument
20+
import murfey.server.api.processing_parameters
2021
import murfey.server.api.spa
2122
import murfey.server.websocket
2223
import murfey.util.models
@@ -73,6 +74,7 @@ class Settings(BaseSettings):
7374
app.include_router(murfey.server.api.display.router)
7475
app.include_router(murfey.server.api.instrument.router)
7576
app.include_router(murfey.server.api.hub.router)
77+
app.include_router(murfey.server.api.processing_parameters.router)
7678
app.include_router(murfey.server.websocket.ws)
7779

7880
# Search external packages for additional routers to include in Murfey

0 commit comments

Comments
 (0)