B |‹d,ã@s|dZddlZddlmZddlmZddlmZ ddl m Z ej   dd ¡Zd d „Zefd d „Zefdd„Zedfdd„ZdS)zQuery methods for DQSEGDB2 éN)Úsegments)Úrequests)Útarget_audienceé)ÚapiÚDEFAULT_SEGMENT_SERVERzhttps://segments.ligo.orgcOstj||Ž}| ¡| ¡S)N)Ú igwn_requestsÚgetÚraise_for_statusÚjson)ÚargsÚkwargsÚresponse©rú[/work/yifan.wang/ringdown/master-ringdown-env/lib/python3.7/site-packages/dqsegdb2/query.pyÚ _get_json"s rc s.t |ˆ¡}t|f|Žd}‡fdd„|DƒS)aÕQuery for all defined flags for the given ``ifo`` Parameters ---------- ifo : `str` the interferometer prefix for which to query host : `str`, optional the URL of the database, defaults to ``DEFAULT_SEGMENT_SERVER`` Returns ------- flags : `set` the set of all define flag names in the format ``{ifo}:{name}`` Examples -------- >>> from dqsegdb2.query import query_names >>> query_names('G1') Úresultscsh|]}d ˆ|¡’qS)z{0}:{1})Úformat)Ú.0Úname)Úiforrú ?szquery_names..)rZname_query_urlr)rÚhostÚrequest_kwargsÚurlÚnamesr)rrÚ query_names(s rcKs2| dd¡\}}t |||¡}tt|f|ŽdƒS)aÞQuery for defined versions for the given flag Parameters ---------- flag : `str` the name for which to query host : `str`, optional the URL of the database, defaults to ``DEFAULT_SEGMENT_SERVER`` Returns ------- versions : `list` of `int` the list of defined versions for the given flag Examples -------- >>> from dqsegdb2.query import query_versions >>> query_versions('G1:GEO-SCIENCE') [1, 2, 3] ú:rÚversion)ÚsplitrZversion_query_urlÚsortedr)ÚflagrrrrrrrrÚquery_versionsBsr"Tc Kstt t t|ƒt|ƒ¡g¡}y | dd¡\}}} t| ƒg} WnJtk r†| d¡rf| dd¡d}| dd¡\}}t ||d} YnXt t ¡t ¡||| dd} |rº|  dt |ƒ¡t jf|Ž¤} xœtt| ƒƒD]Œ\} } tj|||| ||d }t|| d }xJd D]B}| | t ttj| |¡ƒ¡¡|r| | ¡|@| |<qW|  |¡| rÖd | d <qÖWWd QRX| S)aßQuery for segments for the given flag in a ``[start, stop)`` interval Parameters ---------- flag : `str` the name for which to query, see _Notes_ for information on how versionless-flags are queried. start : `int` the GPS start time. end : `int` the GPS end time. host : `str`, optional the URL of the database, defaults to ``DEFAULT_SEGMENT_SERVER``. coalesce : `bool`, optional if `True`, coalesce the segmentlists returned by the server, and restrict them to lie fully within the ``[start, end)`` request segment, otherwise return the 'raw' result, default: `True`. Returns ------- segmentdict : `dict` a `dict` with the following keys - ``'ifo'`` - the interferometer prefix (`str`) - ``'name'`` - the flag name (`str`) - ``'version'`` - the flag version (`int`) - ``'known'`` - the known segments (`~ligo.segments.segmentlist`) - ``'active'`` - the active segments (`~ligo.segments.segmentlist`) - ``'metadata'`` - a `dict` of flag information (`dict`) - ``'query_information'`` - a `dict` of query information (`dict`) Notes ----- If ``flag`` is given without a version (e.g. ``'X1:FLAG-NAME'``) or the version is given as ``'*'`` (e.g. ``'X1:FLAG-NAME:*'``) the result of the query will be the intersection of queries over all versions found in the database. In that case the ``'metadata'`` and ``'query_information'`` in the output will be preserved for the highest version number only. Examples -------- >>> from dqsegdb2.query import query_segments >>> query_segments('G1:GEO-SCIENCE:1', 1000000000, 1000001000) réz:*rr)r)ÚknownÚactiverrrZtoken_audience)ÚstartÚend)Úsession)r%r$Nr)rZ segmentlistÚsegmentÚfloatrÚintÚ ValueErrorÚendswithÚrsplitr"ÚdictÚ setdefaultÚscitoken_audiencerÚSessionÚ enumerater rZsegment_query_urlrÚextendÚmapÚpopÚcoalesceÚupdate)r!r&r'rr7rÚrequestrrrÚversionsÚoutÚsessÚirÚresultÚkeyrrrÚquery_segments]sF:        r@)Ú__doc__ÚosÚligorZigwn_auth_utilsrrZigwn_auth_utils.scitokensrr1ÚrÚenvironr0rrrr"r@rrrrÚs