Fix: use api_url argument

main.py

@@ -6,10 +6,10 @@ from requests import HTTPError
 from utils import get_existing_results, save_results
 
 
-def main(analytics_token):
+def main(analytics_token, api_url):
     try:
         # Get existing results
-        existing_results = get_existing_results(analytics_token)
+        existing_results = get_existing_results(api_url, analytics_token)
 
         my_engine_results = existing_results["my_engine"]
 
@@ -23,10 +23,10 @@ def main(analytics_token):
         ]
 
         # Send result to rights engine
-        save_results(analytics_token, {"result": result})
+        save_results(api_url, analytics_token, {"result": result})
     except HTTPError as error:
         print(error.response.status_code, error.response.text)
 
 
 if __name__ == "__main__":
-    main(sys.argv[1])
+    main(sys.argv[1], sys.argv[2])

utils.py

 import requests
 
-API_URL = "http://127.0.0.1:8003"
 PAGE_SIZE = 100
 
 
-def get_statements_page(analytics_token, last_object_id=None):
+def get_statements_page(api_url, analytics_token, last_object_id=None):
     payload = (
         {"page_size": PAGE_SIZE, "last_object_id": last_object_id}
         if last_object_id
         else {"page_size": PAGE_SIZE}
     )
     response = requests.post(
-        f"{API_URL}/api/v1/provider/data",
+        f"{api_url}/api/v1/provider/data",
         headers={"Authorization": f"Basic {analytics_token}"},
         json=payload,
     )
@@ -21,12 +20,12 @@ def get_statements_page(analytics_token, last_object_id=None):
     return payload.get("statements", [])
 
 
-def get_statements(analytics_token):
+def get_statements(api_url, analytics_token):
     statements = []
     read_all_pages = False
     last_object_id = None
     while not read_all_pages:
-        page_statements = get_statements_page(analytics_token, last_object_id)
+        page_statements = get_statements_page(api_url, analytics_token, last_object_id)
         statements.extend(page_statements)
         if len(page_statements) == 0:
             read_all_pages = True
@@ -36,9 +35,9 @@ def get_statements(analytics_token):
     return statements
 
 
-def get_existing_results(analytics_token):
+def get_existing_results(api_url, analytics_token):
     response = requests.get(
-        f"{API_URL}/api/v1/provider/results",
+        f"{api_url}/api/v1/provider/results",
         headers={"Authorization": f"Basic {analytics_token}"},
     )
     response.raise_for_status()
@@ -46,9 +45,9 @@ def get_existing_results(analytics_token):
     return data
 
 
-def save_results(analytics_token, results):
+def save_results(api_url, analytics_token, results):
     response = requests.post(
-        f"{API_URL}/api/v1/provider/store-result",
+        f"{api_url}/api/v1/provider/store-result",
         headers={"Authorization": f"Basic {analytics_token}"},
         json=results,
     )