Skip to content

Commit aed6c98

Browse files
committed
Remove some path construction that is specific to DLS directory structure
1 parent d171395 commit aed6c98

File tree

3 files changed

+19
-10
lines changed

3 files changed

+19
-10
lines changed

src/murfey/server/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2813,7 +2813,7 @@ def feedback_callback(header: dict, message: dict) -> None:
28132813
dose_per_frame=message["dose_per_frame"],
28142814
gain_ref=(
28152815
str(machine_config.rsync_basepath / message["gain_ref"])
2816-
if message["gain_ref"]
2816+
if message["gain_ref"] and machine_config.data_transfer_enabled
28172817
else message["gain_ref"]
28182818
),
28192819
voltage=message["voltage"],

src/murfey/server/api/__init__.py

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1232,7 +1232,7 @@ async def request_tomography_preprocessing(
12321232
"fm_dose": proc_file.dose_per_frame,
12331233
"gain_ref": (
12341234
str(machine_config.rsync_basepath / proc_file.gain_ref)
1235-
if proc_file.gain_ref
1235+
if proc_file.gain_ref and machine_config.data_transfer_enabled
12361236
else proc_file.gain_ref
12371237
),
12381238
"fm_int_file": proc_file.eer_fractionation_file,
@@ -1509,14 +1509,22 @@ async def write_eer_fractionation_file(
15091509
machine_config = get_machine_config(instrument_name=instrument_name)[
15101510
instrument_name
15111511
]
1512-
file_path = (
1513-
Path(machine_config.rsync_basepath)
1514-
/ (machine_config.rsync_module or "data")
1515-
/ str(datetime.datetime.now().year)
1516-
/ secure_filename(visit_name)
1517-
/ "processing"
1518-
/ secure_filename(fractionation_params.fractionation_file_name)
1519-
)
1512+
if machine_config.eer_fractionation_file_template:
1513+
file_path = Path(
1514+
machine_config.eer_fractionation_file_template.format(
1515+
visit=secure_filename(visit_name),
1516+
year=str(datetime.datetime.now().year),
1517+
)
1518+
) / secure_filename(fractionation_params.fractionation_file_name)
1519+
else:
1520+
file_path = (
1521+
Path(machine_config.rsync_basepath)
1522+
/ (machine_config.rsync_module or "data")
1523+
/ str(datetime.datetime.now().year)
1524+
/ secure_filename(visit_name)
1525+
/ "processing"
1526+
/ secure_filename(fractionation_params.fractionation_file_name)
1527+
)
15201528
if file_path.is_file():
15211529
return {"eer_fractionation_file": str(file_path)}
15221530

src/murfey/util/config.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ class MachineConfig(BaseModel):
2727
create_directories: Dict[str, str] = {"atlas": "atlas"}
2828
analyse_created_directories: List[str] = []
2929
gain_reference_directory: Optional[Path] = None
30+
eer_fractionation_file_template: str = ""
3031
processed_directory_name: str = "processed"
3132
gain_directory_name: str = "processing"
3233
node_creator_queue: str = "node_creator"

0 commit comments

Comments
 (0)