From ab30c7d1177dd5325390176304d1a053389c3aa2 Mon Sep 17 00:00:00 2001
From: "duncan.macleod" <duncan.macleod@ligo.org>
Date: Fri, 8 Apr 2022 14:46:41 +0100
Subject: [PATCH] ui: handle port numbers in default token_audience

add default audiences with and without the port suffix
---
 gwdatafind/ui.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/gwdatafind/ui.py b/gwdatafind/ui.py
index 5ca620b..bf4759d 100644
--- a/gwdatafind/ui.py
+++ b/gwdatafind/ui.py
@@ -115,10 +115,13 @@ def get(url, *args, **kwargs):
     if url.startswith("http://") and requests.__version__ < "2.15.0":
         # workaround https://github.com/psf/requests/issues/4025
         kwargs.setdefault("cert", False)
-    kwargs.setdefault("token_audience", [
-        "{0.scheme}://{0.netloc}".format(urlparse(url)),
+    scheme, netloc = urlparse(url)[:2]
+    host = netloc.split(":", 1)[0]  # remove the port
+    kwargs.setdefault("token_audience", list({
+        f"{scheme}://{netloc}",
+        f"{scheme}://{host}",
         "ANY",
-    ])
+    }))
     kwargs.setdefault("token_scope", "read:/frames")
     return _get(url, *args, **kwargs)
 
-- 
GitLab