Add method to change visit end time in multigrid controller and all child rsync processes

Author: d-j-hatton

src/murfey/client/multigrid_control.py

@@ -149,6 +149,11 @@ def finalise(self):
         for p in self.rsync_processes.keys():
             self._finalise_rsyncer(p)
 
+    def update_visit_time(self, new_end_time: datetime):
+        self.visit_end_time = new_end_time
+        for rp in self.rsync_processes.values():
+            rp._end_time = new_end_time
+
     def _start_rsyncer_multigrid(
         self,
         source: Path,

src/murfey/instrument_server/api.py

@@ -214,6 +214,13 @@ def stop_multigrid_watcher(session_id: MurfeySessionID, label: str):
     watchers[label].request_stop()
 
 
+@router.post("/sessions/{session_id}/multigrid_controller/visit_end_time/{timestamp}")
+def update_multigrid_controller_visit_end_time(
+    session_id: MurfeySessionID, timestamp: float
+):
+    controllers[session_id].update_visit_time(datetime.fromtimestamp(timestamp))
+
+
 class RsyncerSource(BaseModel):
     source: Path
     label: str

src/murfey/server/api/instrument.py

@@ -394,6 +394,29 @@ async def finalise_session(session_id: MurfeySessionID, db=murfey_db):
     return data
 
 
+@router.post("/sessions/{session_id}/multigrid_controller/visit_end_time/{timestamp}")
+async def update_visit_end_time(
+    session_id: MurfeySessionID, timestamp: float, db=murfey_db
+):
+    data = {}
+    instrument_name = (
+        db.exec(select(Session).where(Session.id == session_id)).one().instrument_name
+    )
+    machine_config = get_machine_config(instrument_name=instrument_name)[
+        instrument_name
+    ]
+    if machine_config.instrument_server_url:
+        async with aiohttp.ClientSession() as clientsession:
+            async with clientsession.post(
+                f"{machine_config.instrument_server_url}{url_path_for('api.router', 'update_multigrid_controller_visit_end_time', session_id=session_id, timestamp=timestamp)}",
+                headers={
+                    "Authorization": f"Bearer {instrument_server_tokens[session_id]['access_token']}"
+                },
+            ) as resp:
+                data = await resp.json()
+    return data
+
+
 @router.post("/sessions/{session_id}/abandon_session")
 async def abandon_session(session_id: MurfeySessionID, db=murfey_db):
     data = {}