#!/usr/bin/env python3
# -*- coding: utf-8 -*-
'''
rundb - Helper classes for retrieving information from the RunDB
----------------------------------------------------------------
This modules contains classes useful to deal with the RunDB:
* `RunDB`, a simple API class to just get run information from the RunDB
'''
import requests
import getpass
[docs]class RunDB:
"""
Simple API class to just get run information from the RunDB.
Parameters:
apikey (str): RunDB API key (see `here <https://questions.belle2.org/question/9847/obtaining-api-key-for-rundb/>`_
how to get one)
username (str): DESY username
"""
#: URL of where the RunDB is hosted
URL = "https://rundb.belle2.org"
def __init__(self, apikey=None, username=None):
"""Create an object and setup authentication."""
self._session = requests.Session()
if apikey is None:
# If no specific username use the local system username
if username is None:
username = getpass.getuser()
# If we don't have an api key use desy username/password
self._session.auth = (username, getpass.getpass(f"DESY password ({username}): "))
else:
# Otherwise use the api key
self._session.headers.update({'Authorization': f'Bearer {apikey}'})
# And request json output ...
self._session.headers.update({'Content-Type': 'application/json'})
def _pagination(self, request):
"""Deal with API pagination of an initial request to the API.
It will return all the objects from all pages lazily requesting new pages
as objects are consumed. Will work for all list requests to the server.
"""
while True:
# check the return value and raise exception on error
request.raise_for_status()
# and otherwise get the json
result = request.json()
# and return the objects one by one by yielding objects from the list
# of results
yield from result['results']
# check if there's a next page, if not done
if result['next'] is None:
break
# otherwise continue with the next page
# yees, global variable ...
request = self._session.get(result['next'])
[docs] def get_run_info(self, **search_params):
"""Return the run information from the run registry.
All arguments are forwarded to the run registry ``/run/`` method
documented at the following `link <https://rundb.belle2.org/rest/v1/swagger/>`_.
Please check there for up to date documentation, at the time of
this writing the supported arguments are:
* min_experiment (int)
* min_run (int)
* max_experiment (int)
* max_run (int)
* min_date (iso8601 date string, e.g. 2020-05-06)
* max_date (iso8601 date string, e.g. 2020-05-06)
* all_detectors_running (bool)
* expand (bool): If true return full run objects, not just a summary
links to the run objects
If ``expand=False`` you can request the full objects for each run by calling
`get_details` with the returned run summary object as argument.
``expand=False`` is much faster if no further details are needed but
getting the details in a separate step for many many runs will be slow
so depending on how many runs are selected one or the other may be
faster.
"""
req = self._session.get(f'{self.URL}/rest/v1/runs/', params=search_params)
return self._pagination(req)
[docs] def get_details(self, run_summary):
"""
Return details for a run summary object returned from `get_run_info`
if ``expand`` was not set to True
Parameters:
run_summary: a run summary object returned from `get_run_info`
"""
# Get the url object
req = self._session.get(run_summary['url'])
# Raise an exception in case of any error
req.raise_for_status()
# And return the json object
return req.json()