for visr allow overriding the pv

DiamondLightSource · Sep 18, 2024 · 024aad9 · 024aad9
1 parent 92e3edb
commit 024aad9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/src/dodal/beamlines/b01_1.py b/src/dodal/beamlines/b01_1.py
@@ -14,7 +14,7 @@
 from dodal.log import set_beamline as set_log_beamline
 from dodal.utils import get_beamline_name
 
-BL = get_beamline_name("c01")  # noqa: F821
+BL = get_beamline_name("c01", force_default=True)  # noqa: F821
 set_log_beamline(BL)
 set_utils_beamline(BL)
 

diff --git a/src/dodal/utils.py b/src/dodal/utils.py
@@ -68,7 +68,9 @@
 AnyDeviceFactory: TypeAlias = V1DeviceFactory | V2DeviceFactory
 
 
-def get_beamline_name(default: str) -> str:
+def get_beamline_name(default: str, force_default: bool = False) -> str:
+    if force_default:
+        return default
     return environ.get("BEAMLINE") or default
 
 
@@ -149,6 +151,7 @@ def make_all_devices(
     """
     if isinstance(module, str) or module is None:
         module = import_module(module or __name__)
+    print(module.BL)
     factories = collect_factories(module, include_skipped)
     devices: tuple[dict[str, AnyDevice], dict[str, Exception]] = invoke_factories(
         factories, **kwargs