Skip to content

Commit 9aa2dc6

Browse files
committed
fix: add disable_healthcheck kwarg to create_dataproxy
1 parent a47b3d9 commit 9aa2dc6

1 file changed

Lines changed: 14 additions & 8 deletions

File tree

src/ga4gh/vrs/dataproxy.py

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -251,6 +251,7 @@ def __init__(self, base_url: str, disable_healthcheck: bool = False) -> None:
251251
"""Initialize REST-based dataproxy instance.
252252
253253
:param base_url: root URL to server
254+
:param disable_healthcheck: Whether healthcheck should be disabled
254255
"""
255256
super().__init__()
256257
self.base_url = f"{base_url}/{self.rest_version}/"
@@ -344,16 +345,19 @@ def _isoformat(o: datetime.datetime) -> str:
344345
# self.base_url = base_url
345346

346347

347-
def create_dataproxy(uri: str | None = None) -> _DataProxy:
348+
def create_dataproxy(
349+
uri: str | None = None, disable_healthcheck: bool = False
350+
) -> _DataProxy:
348351
"""Create a dataproxy from uri or GA4GH_VRS_DATAPROXY_URI
349352
350-
Currently accepted URI schemes:
351-
352-
* seqrepo+file:///path/to/seqrepo/root
353-
* seqrepo+:../relative/path/to/seqrepo/root
354-
* seqrepo+http://localhost:5000/seqrepo
355-
* seqrepo+https://somewhere:5000/seqrepo
353+
:param: uri: Dataproxy URI
354+
Currently accepted URI schemes:
356355
356+
* seqrepo+file:///path/to/seqrepo/root
357+
* seqrepo+:../relative/path/to/seqrepo/root
358+
* seqrepo+http://localhost:5000/seqrepo
359+
* seqrepo+https://somewhere:5000/seqrepo
360+
:param disable_healthcheck: Whether healthcheck should be disabled in REST dataproxy
357361
:raise ValueError: if URI doesn't match recognized schemes, e.g. is missing provider
358362
prefix (`"seqrepo+"`)
359363
"""
@@ -379,7 +383,9 @@ def create_dataproxy(uri: str | None = None) -> _DataProxy:
379383
sr = SeqRepo(root_dir=parsed_uri.path)
380384
dp = SeqRepoDataProxy(sr)
381385
elif proto in ("http", "https"):
382-
dp = SeqRepoRESTDataProxy(uri[len(provider) + 1 :])
386+
dp = SeqRepoRESTDataProxy(
387+
uri[len(provider) + 1 :], disable_healthcheck=disable_healthcheck
388+
)
383389
else:
384390
msg = f"SeqRepo URI scheme {parsed_uri.scheme} not implemented"
385391
raise ValueError(msg)

0 commit comments

Comments
 (0)