make cohort middleware url configurable

Author: tianj7

config.ini

@@ -1,3 +1,4 @@
 [DEFAULT]
 ARGO_ACCESS_METHOD = access
 ARGO_HOST = http://argo-argo-workflows-server.argo.svc.cluster.local:2746
+COHORT_DEFINITION_BY_SOURCE_AND_TEAM_PROJECT_URL = http://cohort-middleware-service/cohortdefinition-stats/by-source-id/{}/by-team-project?team-project={}

src/argowrapper/auth/utils.py

@@ -1,13 +1,18 @@
 import requests
 from argowrapper import logger
+from argowrapper.constants import (
+    COHORT_DEFINITION_BY_SOURCE_AND_TEAM_PROJECT_URL,
+)
 
 
 def get_cohort_ids_for_team_project(token, source_id, team_project):
     header = {"Authorization": token, "cookie": "fence={}".format(token)}
-    url = "http://cohort-middleware-service/cohortdefinition-stats/by-source-id/{}/by-team-project?team-project={}".format(
+    url1 = "http://cohort-middleware-service/cohortdefinition-stats/by-source-id/{}/by-team-project?team-project={}".format(
+        source_id, team_project
+    )
+    url = COHORT_DEFINITION_BY_SOURCE_AND_TEAM_PROJECT_URL.format(
         source_id, team_project
     )
-
     try:
         r = requests.get(url=url, headers=header)
         r.raise_for_status()

src/argowrapper/constants.py

@@ -20,6 +20,9 @@
 logger.info(f"Access method: {config['DEFAULT']['ARGO_ACCESS_METHOD']}")
 
 ARGO_HOST: Final = config["DEFAULT"]["ARGO_HOST"]
+COHORT_DEFINITION_BY_SOURCE_AND_TEAM_PROJECT_URL: Final = config["DEFAULT"][
+    "COHORT_DEFINITION_BY_SOURCE_AND_TEAM_PROJECT_URL"
+]
 TEST_WF: Final = "test.yaml"
 WF_HEADER: Final = "header.yaml"
 ARGO_NAMESPACE: Final = "argo"