142 solenoid_current_override=None, toroids_current_override=None):
143 """
144 Create RunParams by reading from IS via the WEBDAQ REST API.
145
146 This uses the webis_server REST API to fetch run parameters, avoiding
147 direct dependencies on TDAQ libraries. The API endpoint is determined by:
148 1. The webdaq_base parameter if provided
149 2. The TDAQ_WEBDAQ_BASE environment variable
150
151 The IS objects accessed are:
152 - RunParams.RunParams: run_number, det_mask, timeSOR, trigger_type, etc.
153 - Magnets.Magnets: SolenoidCurrent, ToroidsCurrent
154
155 Args:
156 partition: The partition name (default: from TDAQ_PARTITION env var)
157 webdaq_base: Base URL for webis_server (default: from TDAQ_WEBDAQ_BASE env var)
158 strict: If True, raise an exception if IS read fails (for --online-environment)
159 solenoid_current_override: If provided, skip IS fetch for solenoid (command-line override)
160 toroids_current_override: If provided, skip IS fetch for toroids (command-line override)
161
162 Returns:
163 RunParams instance with values from IS, or defaults if unavailable
164
165 Raises:
166 RuntimeError: If strict=True and IS read fails
167 """
168 import requests
169
170
171 if webdaq_base is None:
172 webdaq_base = os.environ.get('TDAQ_WEBDAQ_BASE')
173
174 if not webdaq_base:
175 msg = "TDAQ_WEBDAQ_BASE not set, cannot read from IS"
176 if strict:
177 raise RuntimeError(msg + " (required for --online-environment)")
178 log.warning(msg + ". Using defaults.")
179 return cls()
180
181
182 if partition is None:
183 partition = os.environ.get('TDAQ_PARTITION', 'ATLAS')
184
185 log.info("Reading run parameters from IS via WEBDAQ: %s (partition=%s)", webdaq_base, partition)
186
187 params = {}
188
189
190
191
192 try:
193 url = f"{webdaq_base}/info/current/{partition}/is/RunParams/RunParams.RunParams?format=compact"
194 log.debug("Fetching RunParams from: %s", url)
195
196 response = requests.get(url, timeout=10)
197 if response.status_code == 200:
198 response_data = response.json()
199 log.debug("RunParams response from IS: %s", response_data)
200
201
202 if isinstance(response_data, list) and len(response_data) >= 4:
203 runparams = response_data[3]
204 else:
205 runparams = response_data
206
207 log.debug("RunParams data: %s", runparams)
208
209
210 if 'run_number' in runparams:
211 params['run_number'] = int(runparams['run_number'])
212 if 'lumiblock' in runparams:
213 params['lb_number'] = int(runparams['lumiblock'])
214 if 'det_mask' in runparams:
215 params['detector_mask'] = runparams['det_mask']
216 if 'timeSOR' in runparams:
217 sor_time = runparams['timeSOR']
218
219 if '.' not in sor_time:
220 sor_time += '.000000'
221 params['sor_time'] = sor_time
222 if 'beam_type' in runparams:
223 params['beam_type'] = int(runparams['beam_type'])
224 if 'beam_energy' in runparams:
225 params['beam_energy'] = int(runparams['beam_energy'])
226 if 'run_type' in runparams:
227 params['run_type'] = runparams['run_type']
228 if 'trigger_type' in runparams:
229 params['trigger_type'] = int(runparams['trigger_type'])
230 if 'recording_enabled' in runparams:
231 params['recording_enabled'] = runparams['recording_enabled'] in ('1', 'true', 'True', True, 1)
232
233 log.info("Got run parameters from IS: run=%s, lb=%s",
234 params.get('run_number'), params.get('lb_number'))
235 else:
236 msg = f"Failed to fetch RunParams from IS: HTTP {response.status_code}"
237 if strict:
238 raise RuntimeError(msg + " (required for --online-environment)")
239 log.warning(msg)
240
241 except requests.exceptions.RequestException as e:
242 msg = f"Error fetching RunParams from IS: {e}"
243 if strict:
244 raise RuntimeError(msg + " (required for --online-environment)")
245 log.warning(msg)
246 except (ValueError, KeyError) as e:
247 msg = f"Error parsing RunParams from IS: {e}"
248 if strict:
249 raise RuntimeError(msg + " (required for --online-environment)")
250 log.warning(msg)
251
252
253
254
255 have_solenoid_override = solenoid_current_override is not None
256 have_toroids_override = toroids_current_override is not None
257
258 if have_solenoid_override:
259 params['solenoid_current'] = solenoid_current_override
260 log.info("Using solenoid_current=%.1f from command line override", solenoid_current_override)
261 if have_toroids_override:
262 params['toroids_current'] = toroids_current_override
263 log.info("Using toroids_current=%.1f from command line override", toroids_current_override)
264
265
266 if not (have_solenoid_override and have_toroids_override):
267 try:
268 url = f"{webdaq_base}/info/current/{partition}/is/Magnets/Magnets.Magnets?format=compact"
269 log.debug("Fetching Magnets from: %s", url)
270
271 response = requests.get(url, timeout=10)
272 if response.status_code == 200:
273 response_data = response.json()
274 log.debug("Magnets response from IS: %s", response_data)
275
276 magnets = response_data[3] if isinstance(response_data, list) and len(response_data) >= 4 else response_data
277 log.debug("Magnets data: %s", magnets)
278
279
280
281 if not have_solenoid_override:
282 params['solenoid_current'] = float(magnets['SolenoidCurrent']['value'])
283 if not have_toroids_override:
284 params['toroids_current'] = float(magnets['ToroidsCurrent']['value'])
285
286 log.info("Got magnet currents from IS: solenoid=%s, toroids=%s",
287 params.get('solenoid_current'), params.get('toroids_current'))
288 elif strict:
289 raise RuntimeError(f"Magnets not available from IS: HTTP {response.status_code} "
290 "(required for --online-environment, use --solenoid-current and --toroids-current to override)")
291 else:
292 log.debug("Magnets not available from IS: HTTP %d", response.status_code)
293
294 except requests.exceptions.RequestException as e:
295 if strict:
296 raise RuntimeError(f"Error fetching Magnets from IS: {e} "
297 "(required for --online-environment, use --solenoid-current and --toroids-current to override)")
298 log.debug("Error fetching Magnets from IS: %s", e)
299 except (ValueError, KeyError, TypeError) as e:
300 if strict:
301 raise RuntimeError(f"Error parsing Magnets from IS: {e} "
302 "(required for --online-environment, use --solenoid-current and --toroids-current to override)")
303 log.debug("Error parsing Magnets from IS: %s", e)
304
305
306 if strict and 'run_number' not in params:
307 raise RuntimeError("Failed to get run_number from IS (required for --online-environment)")
308
309 return cls(**params)
310
311
1.14.0