Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
36 changes: 36 additions & 0 deletions src/murfey/server/api/processing_parameters.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,36 @@
from logging import getLogger

from fastapi import APIRouter
from pydantic import BaseModel
from sqlmodel import Session, select

from murfey.server.murfey_db import murfey_db
from murfey.util.db import SessionProcessingParameters

logger = getLogger("murfey.server.api.processing_parameters")

router = APIRouter()


class EditableSessionProcessingParameters(BaseModel):
gain_ref: str
dose_per_frame: float
eer_fractionation_file: str
symmetry: str


@router.get("sessions/{session_id}/session_processing_parameters")
def get_session_processing_parameters(
session_id: int, db: Session = murfey_db
) -> EditableSessionProcessingParameters:
proc_params = db.exec(
select(SessionProcessingParameters).where(
SessionProcessingParameters.session_id == session_id
)
).one()
return EditableSessionProcessingParameters(
gain_ref=proc_params.gain_ref,
dose_per_frame=proc_params.dose_per_frame,
eer_fractionation_file=proc_params.eer_fractionation_file,
symmetry=proc_params.symmetry,
)
2 changes: 1 addition & 1 deletion src/murfey/server/demo_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,7 +28,6 @@
_flush_grid_square_records,
_flush_tomography_preprocessing,
_murfey_id,
_register_picked_particles_use_diameter,
feedback_callback,
get_hostname,
get_microscope,
Expand Down Expand Up @@ -91,6 +90,7 @@
)
from murfey.util.processing_params import default_spa_parameters
from murfey.util.state import global_state
from murfey.workflows.spa.picking import _register_picked_particles_use_diameter

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

Expand Down
2 changes: 2 additions & 0 deletions src/murfey/server/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
import murfey.server.api.display
import murfey.server.api.hub
import murfey.server.api.instrument
import murfey.server.api.processing_parameters
import murfey.server.api.spa
import murfey.server.websocket
import murfey.util.models
Expand Down Expand Up @@ -73,6 +74,7 @@ class Settings(BaseSettings):
app.include_router(murfey.server.api.display.router)
app.include_router(murfey.server.api.instrument.router)
app.include_router(murfey.server.api.hub.router)
app.include_router(murfey.server.api.processing_parameters.router)
app.include_router(murfey.server.websocket.ws)

# Search external packages for additional routers to include in Murfey
Expand Down