feat: append user agent to default user agent

closes #110
GIScience · Nov 16, 2023 · f81e9c8 · f81e9c8
1 parent ebfcbe8
commit f81e9c8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/ohsome/clients.py b/ohsome/clients.py
@@ -53,10 +53,11 @@ def __init__(
         else:
             self._base_api_url = OHSOME_BASE_API_URL
         self._cache = cache or []
-        if user_agent is not None:
-            self.user_agent = user_agent
-        else:
-            self.user_agent = "ohsome-py/{0}".format(OHSOME_VERSION)
+
+        if user_agent is None:
+            user_agent = "ohsome-py"
+        self.user_agent = f"ohsome-py/{OHSOME_VERSION} {user_agent}"
+
         self.__session = None
 
     def _session(self):

diff --git a/ohsome/test/test_client.py b/ohsome/test/test_client.py
@@ -43,7 +43,7 @@ def test_user_agent(base_client):
     resp = base_client._session().get(base_client._url)
     used_user_agent = resp.request.headers["user-agent"].split("/")
     assert used_user_agent[0] == "ohsome-py"
-    assert used_user_agent[1] == OHSOME_VERSION
+    assert used_user_agent[1] == f"{OHSOME_VERSION} ohsome-py"
 
 
 @pytest.mark.vcr