From cd6eaf7be75f99d1760c44ff4de27606cbd402db Mon Sep 17 00:00:00 2001 From: Julian Stirling Date: Wed, 9 Jul 2025 21:19:55 +0000 Subject: [PATCH] Apply suggestions from code review of branch pass-portal-to-stop --- src/openflexure_microscope_server/server/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/openflexure_microscope_server/server/__init__.py b/src/openflexure_microscope_server/server/__init__.py index 5a8412c6..7f54b05d 100644 --- a/src/openflexure_microscope_server/server/__init__.py +++ b/src/openflexure_microscope_server/server/__init__.py @@ -1,6 +1,7 @@ from __future__ import annotations from typing import Optional, Callable +from functools import wraps from copy import copy import labthings_fastapi as lt @@ -30,6 +31,7 @@ def set_shutdown_function(shutdown_function: Callable[[], None]): original_handler = Server.handle_exit + @wraps(Server.handle_exit) def handle_exit(*args, **kwargs): shutdown_function() original_handler(*args, **kwargs)