Merge pull request #105 from UpCloudLtd/modifiable-user-agent

API: user agent to class variable

Author: ajmyyra

upcloud_api/api.py

@@ -2,6 +2,7 @@
 
 import requests
 
+from upcloud_api import __version__
 from upcloud_api.errors import UpCloudAPIError
 
 
@@ -11,6 +12,7 @@ class API:
     """
 
     api_root = 'https://api.upcloud.com/1.3'
+    user_agent = f'upcloud-python-api/{__version__}'
 
     def __init__(self, token, timeout=None):
         """
@@ -29,7 +31,7 @@ def api_request(self, method, endpoint, body=None, params=None, timeout=-1):
             raise Exception('Invalid/Forbidden HTTP method')
 
         url = f'{self.api_root}{endpoint}'
-        headers = {'Authorization': self.token, 'User-Agent': self._get_user_agent()}
+        headers = {'Authorization': self.token, 'User-Agent': self.user_agent}
 
         if body:
             data = json.dumps(body)
@@ -91,8 +93,3 @@ def __error_middleware(self, res, res_json):
             )
 
         return res_json
-
-    def _get_user_agent(self) -> str:
-        from upcloud_api import __version__
-
-        return f'upcloud-python-api/{__version__}'