Client class can use a different base url

Author: dannywillems

example/example_client.py

@@ -6,7 +6,8 @@
 
 
 API_KEY = decouple.config("API_KEY")
-CLIENT = Client(api_key=API_KEY)
+BASE_URL = decouple.config("LEAKIX_HOST", default=None)
+CLIENT = Client(api_key=API_KEY, base_url=BASE_URL)
 
 
 def example_get_host_filter_plugin():

leakix/client.py

@@ -25,10 +25,11 @@ class HostResult(Model):
 
 
 class Client:
-    BASE_URL = "https://leakix.net"
-
-    def __init__(self, api_key: Optional[str] = None):
+    def __init__(self,
+                 api_key: Optional[str] = None,
+                 base_url: Optional[str] = "https://leakix.net"):
         self.api_key = api_key
+        self.base_url = base_url
         self.headers = {
             "Accept": "application/json",
             "User-agent": "leakix-client-python/%s" % __VERSION__,
@@ -43,7 +44,7 @@ def get(self, scope: Scope, queries: Optional[List[Query]] = None):
             serialized_query = [q.serialize() for q in queries]
             serialized_query = " ".join(serialized_query)
             serialized_query = "%s" % serialized_query
-        url = "%s/search" % self.BASE_URL
+        url = "%s/search" % self.base_url
         r = requests.get(
             url,
             params={"scope": scope.value, "q": serialized_query},
@@ -74,7 +75,7 @@ def get_leak(self, queries: Optional[List[Query]] = None):
         return r
 
     def get_host(self, ipv4: str):
-        url = "%s/host/%s" % (self.BASE_URL, ipv4)
+        url = "%s/host/%s" % (self.base_url, ipv4)
         r = requests.get(url, headers=self.headers)
         if r.status_code == 200:
             response_json = r.json()

tests/test_client.py

@@ -30,7 +30,7 @@ def test_get_host_success(client):
         # remove .json
         ipv4 = f.name[:-5]
         with requests_mock.Mocker() as m:
-            url = "%s/host/%s" % (client.BASE_URL, ipv4)
+            url = "%s/host/%s" % (client.base_url, ipv4)
             m.get(url, json=res_json, status_code=200)
             response = client.get_host(ipv4)
             assert response.is_success()
@@ -47,7 +47,7 @@ def test_get_host_404(client):
         ipv4 = f.name[:-5]
         client = Client()
         with requests_mock.Mocker() as m:
-            url = "%s/host/%s" % (client.BASE_URL, ipv4)
+            url = "%s/host/%s" % (client.base_url, ipv4)
             m.get(url, json=res_json, status_code=404)
             response = client.get_host(ipv4)
             assert response.is_error()
@@ -60,7 +60,7 @@ def test_get_host_429(client, fake_ipv4):
     status_code = 429
     res_json = {"reason": "rate-limit", "status": "error"}
     with requests_mock.Mocker() as m:
-        url = "%s/host/%s" % (client.BASE_URL, fake_ipv4)
+        url = "%s/host/%s" % (client.base_url, fake_ipv4)
         m.get(url, json=res_json, status_code=status_code)
         response = client.get_host(fake_ipv4)
         assert response.is_error()