Make base URL more dynamic

Author: kesmit13

singlestoredb/management/manager.py

@@ -62,6 +62,7 @@ def __init__(
         )
         if not new_access_token:
             raise ManagementError(msg='No management token was configured.')
+
         self._is_jwt = not access_token and new_access_token and is_jwt(new_access_token)
         self._sess = requests.Session()
         self._sess.headers.update({
@@ -70,10 +71,25 @@ def __init__(
             'Accept': 'application/json',
             'User-Agent': f'SingleStoreDB-Python/{client_version}',
         })
+
+        # The base URL must be able to be set dynamically using the default_base_url
+        # class variable or the environment variable SINGLESTOREDB_MANAGEMENT_BASE_URL.
+        # Check to see if the current default_base_url or the environment variable
+        # is different from the default_base_url and use that if so.
+        config_base_url = config.get_option('management.base_url')
+        env_base_url = os.environ.get('SINGLESTOREDB_MANAGEMENT_BASE_URL', '').strip()
+        if base_url:
+            pass
+        elif config_base_url and config_base_url != type(self).default_base_url:
+            base_url = config_base_url
+        elif env_base_url and env_base_url != type(self).default_base_url:
+            base_url = env_base_url
+
         self._base_url = urljoin(
             base_url or type(self).default_base_url,
             version or type(self).default_version,
         ) + '/'
+
         self._params: Dict[str, str] = {}
         if organization_id:
             self._params['organizationID'] = organization_id