feat: custom user-agent

pyproject.toml

@@ -27,6 +27,9 @@ build-backend = "poetry.core.masonry.api"
 version_toml = [
     "pyproject.toml:tool.poetry.version"
 ]
+version_variable = [
+    "threatx_api_client/__init__.py:__version__"
+]
 major_on_zero = false
 branch = "main"
 upload_to_vcs_release = true

threatx_api_client/__init__.py

@@ -3,12 +3,15 @@
 import aiohttp
 
 from threatx_api_client.exceptions import (
+    TXAPIError,
     TXAPIIncorrectCommandError,
     TXAPIIncorrectEnvironmentError,
     TXAPIIncorrectTokenError,
-    TXAPIResponseError, TXAPIError,
+    TXAPIResponseError,
 )
 
+__version__ = "1.2.0"
+
 
 class Client:
     """Main API Client class."""
@@ -27,6 +30,10 @@ def __init__(self, api_env, api_key):
         self.api_env = api_env
         self.api_key = api_key
 
+        self.headers = {
+            "User-Agent": f"ThreatX-API-Client/{__version__}"
+        }
+
         self.parallel_requests = 10
 
         self.base_url = self.__get_api_env_host()
@@ -81,7 +88,7 @@ async def __process_response(self, path: str, available_commands: list, payloads
             if payload.get("command") not in available_commands:
                 raise TXAPIIncorrectCommandError(payload.get("command"))
 
-        async with aiohttp.ClientSession(base_url=self.base_url) as session:
+        async with aiohttp.ClientSession(base_url=self.base_url, headers=self.headers) as session:
             responses = await asyncio.gather(*(
                 self.__post(
                     session,
@@ -100,7 +107,7 @@ async def __login(self):
         if not self.api_key:
             raise TXAPIIncorrectTokenError("Please provide TX API Key.")
 
-        async with aiohttp.ClientSession(base_url=self.base_url) as session:
+        async with aiohttp.ClientSession(base_url=self.base_url, headers=self.headers) as session:
             response = await asyncio.gather(
                 self.__post(
                     session,