support eu api domain

Author: aviau

flareio/api_client.py

@@ -15,16 +15,31 @@
 from flareio.version import __version__ as _flareio_version
 
 
+ALLOWED_API_DOMAINS: set[str] = (
+    "api.flare.io",
+    "api.eu.flare.io",
+)
+
+
 class FlareApiClient:
     def __init__(
         self,
         *,
         api_key: str,
         tenant_id: t.Optional[int] = None,
         session: t.Optional[requests.Session] = None,
+        api_domain: str | None = None,
     ) -> None:
         if not api_key:
             raise Exception("API Key cannot be empty.")
+
+        api_domain = api_domain or "api.flare.io"
+        if api_domain not in ALLOWED_API_DOMAINS:
+            raise Exception(
+                f"Invalid API domain: {api_domain}. Only {ALLOWED_API_DOMAINS} are supported."
+            )
+        self.api_domain: str = api_domain
+
         self._api_key: str = api_key
         self._tenant_id: t.Optional[int] = tenant_id
 
@@ -93,7 +108,7 @@ def generate_token(self) -> str:
             }
 
         resp = self._session.post(
-            "https://api.flare.io/tokens/generate",
+            f"https://{self.api_domain}/tokens/generate",
             json=payload,
             headers={
                 "Authorization": self._api_key,
@@ -128,12 +143,12 @@ def _request(
         json: t.Optional[t.Dict[str, t.Any]] = None,
         headers: t.Optional[t.Dict[str, t.Any]] = None,
     ) -> requests.Response:
-        url = urljoin("https://api.flare.io", url)
+        url = urljoin(f"https://{self.api_domain}", url)
 
         netloc: str = urlparse(url).netloc
-        if not netloc == "api.flare.io":
+        if not netloc == self.api_domain:
             raise Exception(
-                f"Client was used to access {netloc=} at {url=}. Only the domain api.flare.io is supported."
+                f"Client was used to access {netloc=} at {url=}. Only the domain {self.api_domain} is supported."
             )
 
         headers = {

tests/test_api_client_creation.py

@@ -14,6 +14,15 @@ def test_create_client() -> None:
     FlareApiClient(api_key="test")
 
 
+def test_create_client_eu() -> None:
+    FlareApiClient(api_key="test", api_domain="api.eu.flare.io")
+
+
+def test_create_client_bad_api_domain() -> None:
+    with pytest.raises(Exception, match="Invalid API domain"):
+        FlareApiClient(api_key="test", api_domain="bad.com")
+
+
 def test_create_client_empty_api_key() -> None:
     with pytest.raises(Exception, match="API Key cannot be empty."):
         FlareApiClient(

tests/test_api_client_endpoints.py

@@ -82,6 +82,20 @@ def test_get_path_only() -> None:
         assert mocker.last_request.url == "https://api.flare.io/hello/test"
 
 
+def test_get_eu_domain() -> None:
+    client = get_test_client(
+        api_domain="api.eu.flare.io",
+    )
+    with requests_mock.Mocker() as mocker:
+        mocker.register_uri(
+            "GET",
+            "https://api.eu.flare.io/hello/test",
+            status_code=200,
+        )
+        client.get("/hello/test")
+        assert mocker.last_request.url == "https://api.eu.flare.io/hello/test"
+
+
 def test_get_user_agent() -> None:
     client = get_test_client()
     with requests_mock.Mocker() as mocker:

tests/utils.py

@@ -9,17 +9,21 @@ def get_test_client(
     *,
     tenant_id: t.Optional[int] = None,
     authenticated: bool = True,
+    api_domain: str | None = None,
 ) -> FlareApiClient:
     client = FlareApiClient(
         api_key="test-api-key",
         tenant_id=tenant_id,
+        api_domain=api_domain,
     )
 
+    mocked_api_domain = api_domain or "api.flare.io"
+
     if authenticated:
         with requests_mock.Mocker() as mocker:
             mocker.register_uri(
                 "POST",
-                "https://api.flare.io/tokens/generate",
+                f"https://{mocked_api_domain}/tokens/generate",
                 json={
                     "token": "test-token-hello",
                 },