[TASK] added context proxy: so that .put has a timeout

If the variable the value shall be placed at does not exist,
put will not timeout. Proxy uses asyncio.wait_for

Author: hzb-ps

src/dt4acc/custom_epics/ioc/handlers.py

@@ -1,15 +1,15 @@
 from bact_twin_architecture.bl.command_rewriter import CommandRewriter
 # from dt4acc.custom_epics.ioc.liasion_translation_manager import build_managers
-from p4p.client.asyncio import Context
 
 from .liasion_translation_manager import build_managers
+from ..utils.context_proxy import ContextProxy
 from ...core.accelerators.pyat_accelerator import setup_accelerator
 from ...core.command import UpdateManager
 from ...core.utils.logger import get_logger
 
 logger = get_logger()
 
-ctx = Context("pva")  # Create a context for EPICS PVA (PV Access)
+ctx = ContextProxy("pva")  # Create a context for EPICS PVA (PV Access)
 
 #: todo replace soon by database service
 

src/dt4acc/custom_epics/ioc/tasks.py

@@ -2,12 +2,11 @@
 import getpass
 from datetime import datetime
 
-from p4p.client.asyncio import Context
-
+from ..utils.context_proxy import ContextProxy
 from ...core.utils.logger import get_logger
 from ...core.views.shared_view import get_view_instance
 
-ctx = Context("pva")
+ctx = ContextProxy("pva")
 logger = get_logger()
 view = get_view_instance()
 heartbeat_task = None  # Global variable to store the heartbeat task reference

src/dt4acc/custom_epics/utils/context_proxy.py

@@ -0,0 +1,14 @@
+import asyncio
+
+from p4p.client.asyncio import Context
+
+class ContextProxy:
+    def __init__(self, context):
+        assert context == "pva"
+        self.context = Context(context)
+
+    async def put(self, *args, timeout=5, **kwargs):
+        return await asyncio.wait_for(
+            self.context.put(*args, **kwargs),
+            timeout=timeout
+        )

src/dt4acc/custom_epics/views/bpm_data.py

@@ -2,13 +2,13 @@
 from typing import Sequence
 
 import numpy as np
-from p4p.client.asyncio import Context
 
+from ..utils.context_proxy import ContextProxy
 from ...core.utils.logger import get_logger
 
 logger = get_logger()
 
-ctx = Context("pva")
+ctx = ContextProxy("pva")
 
 
 class BeamPositionPVs:

src/dt4acc/custom_epics/views/calculation_result_view.py

@@ -13,8 +13,8 @@
 import asyncio
 
 import numpy as np
-from p4p.client.asyncio import Context
 
+from ..utils.context_proxy import ContextProxy
 from ...core.model.element_upate import ElementUpdate
 from ...core.model.orbit import Orbit
 from ...core.model.twiss import TwissWithAggregatedKValues
@@ -23,7 +23,7 @@
 
 logger = get_logger()
 
-ctx = Context("pva")
+ctx = ContextProxy("pva")
 
 
 async def update_orbit_pv(pv_name, orbit_result):

src/dt4acc/custom_epics/views/create_or_update_pv.py

@@ -1,9 +1,8 @@
-from p4p.client.asyncio import Context
-
+from ..utils.context_proxy import ContextProxy
 from ...core.utils.logger import get_logger
 logger = get_logger()
 
-ctx = Context("pva")
+ctx = ContextProxy("pva")
 
 
 async def update_or_create_pv(element, pv_name, value, value_type, initial_type):

src/dt4acc/custom_epics/views/repeated_publishing_view.py

@@ -3,17 +3,18 @@
 from datetime import datetime
 import numpy as np
 
+from ..utils.context_proxy import ContextProxy
 from ...core.interfaces.view_interface import ViewInterface
 from ...core.model.orbit import Orbit
 from ...core.model.twiss import TwissWithAggregatedKValues, TuneData
 from ...core.utils.logger import get_logger
 from ...core.utils.periodic_publisher import PeriodicPublisher
 from ..data.constants import special_pvs
-from p4p.client.asyncio import Context
+
 
 logger = get_logger()
 
-ctx = Context("pva")
+ctx = ContextProxy("pva")
 
 
 class ExtractBPMFromOrbitFilter: