From b75d7d55c4a123aa71ed77073f6247428ece240e Mon Sep 17 00:00:00 2001
From: Shanika Galaudage <shanika.galaudage@ligo.org>
Date: Thu, 9 May 2019 21:58:19 -0500
Subject: [PATCH] Add option for GraceDB service_url
---
bilby/gw/utils.py | 9 +++++++--
1 file changed, 7 insertions(+), 2 deletions(-)
diff --git a/bilby/gw/utils.py b/bilby/gw/utils.py
index e0f3da9c5..22210684f 100644
--- a/bilby/gw/utils.py
+++ b/bilby/gw/utils.py
@@ -461,7 +461,7 @@ def get_gracedb(gracedb, outdir, duration, calibration, detectors, query_types=N
return candidate, cache_files
-def gracedb_to_json(gracedb, cred=None, outdir=None):
+def gracedb_to_json(gracedb, cred=None, service_url='https://gracedb.ligo.org/api/', outdir=None):
""" Script to download a GraceDB candidate
Parameters
@@ -470,6 +470,10 @@ def gracedb_to_json(gracedb, cred=None, outdir=None):
The UID of the GraceDB candidate
cred:
Credentials for authentications, see ligo.gracedb.rest.GraceDb
+ service_url:
+ The url of the GraceDB candidate
+ GraceDB 'https://gracedb.ligo.org/api/' (default)
+ GraceDB-playground 'https://gracedb-playground.ligo.org/api/'
outdir: str, optional
If given, a string identfying the location in which to store the json
"""
@@ -478,8 +482,9 @@ def gracedb_to_json(gracedb, cred=None, outdir=None):
from ligo.gracedb.rest import GraceDb
logger.info('Initialise client and attempt to download')
+ logger.info('Fetching from {}'.format(service_url))
try:
- client = GraceDb(cred=cred)
+ client = GraceDb(cred=cred, service_url=service_url)
except IOError:
raise ValueError(
'Failed to authenticate with gracedb: check your X509 '
--
GitLab