Merge pull request #3 from arangodb/parameters_update

Parameters update
arangodb · Jul 22, 2022 · cce0a78 · cce0a78
2 parents 779fe8a + abba8f1
commit cce0a78
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/adb_cloud_connector/utils.py b/adb_cloud_connector/utils.py
@@ -2,7 +2,7 @@
 import os
 import time
 from json.decoder import JSONDecodeError
-from typing import Any, Dict
+from typing import Any, Dict, Optional
 
 import requests
 from requests.exceptions import HTTPError
@@ -11,7 +11,10 @@
 dir_path = os.path.dirname(os.path.realpath(__file__))
 
 
-def get_temp_credentials() -> Json:
+def get_temp_credentials(
+    tutorialName: Optional[str] = None,
+    credentialProvider: str = "https://tutorials.arangodb.cloud:8529/_db/_system/tutorialDB/tutorialDB",
+) -> Json:
     creds_file = f"{dir_path}/data/creds.json"
 
     try:
@@ -26,8 +29,9 @@ def get_temp_credentials() -> Json:
 
     except (JSONDecodeError, HTTPError):
         print("Log: requesting new credentials...")
-        url = "https://tutorials.arangodb.cloud:8529/_db/_system/tutorialDB/tutorialDB"
-        response = requests.post(url, data=json.dumps({}))
+        url = credentialProvider
+        body = {"tutorialName": tutorialName} if tutorialName else "{}"
+        response = requests.post(url, data=json.dumps(body))
         response.raise_for_status()
 
         data: Json = response.json()

diff --git a/tests/test_utils.py b/tests/test_utils.py
@@ -3,6 +3,7 @@
 
 def test_get_temp_credentials() -> None:
     new_creds = get_temp_credentials()
+    print(new_creds)
     assert new_creds.keys() == {
         "url",
         "hostname",