Skip to content

Commit 654b47f

Browse files
committed
No need to use 'get_db_entry'
1 parent 0f5f431 commit 654b47f

1 file changed

Lines changed: 11 additions & 8 deletions

File tree

src/murfey/workflows/clem/register_align_and_merge_results.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,9 @@
88
from typing import Optional
99

1010
from pydantic import BaseModel, field_validator
11-
from sqlmodel import Session
11+
from sqlmodel import Session, select
1212

1313
from murfey.util.db import CLEMImageSeries
14-
from murfey.workflows.clem import get_db_entry
1514

1615
logger = logging.getLogger("murfey.workflows.clem.register_align_and_merge_results")
1716

@@ -83,12 +82,16 @@ def register_align_and_merge_result(
8382
try:
8483
# Register items in database if not already present
8584
try:
86-
clem_img_series: CLEMImageSeries = get_db_entry(
87-
db=murfey_db,
88-
table=CLEMImageSeries,
89-
session_id=session_id,
90-
series_name=result.series_name,
91-
)
85+
if not (
86+
clem_img_series := murfey_db.exec(
87+
select(CLEMImageSeries)
88+
.where(CLEMImageSeries.session_id == session_id)
89+
.where(CLEMImageSeries.series_name == result.series_name)
90+
).one_or_none()
91+
):
92+
clem_img_series = CLEMImageSeries(
93+
session_id=session_id, series_name=result.series_name
94+
)
9295
clem_img_series.composite_created = True
9396
murfey_db.add(clem_img_series)
9497
murfey_db.commit()

0 commit comments

Comments
 (0)