enabled user-agent logging in request headers

Author: fatihkurtoglu

scaleapi/__init__.py

@@ -1,8 +1,10 @@
 import requests
+import platform
 
 from .tasks import Task
 from .batches import Batch
 from .projects import Project
+from ._version import __version__
 
 TASK_TYPES = [
     'annotation',
@@ -63,6 +65,10 @@ class Batchlist(Paginator):
 class ScaleClient(object):
     def __init__(self, api_key):
         self.api_key = api_key
+        self._headers = {
+            "Content-Type": "application/json",
+            "User-Agent": _generate_useragent()
+        }
 
     def _getrequest(self, endpoint, params=None):
         """Makes a get request to an endpoint.
@@ -73,7 +79,7 @@ def _getrequest(self, endpoint, params=None):
         """
         params = params or {}
         r = requests.get(SCALE_ENDPOINT + endpoint,
-                         headers={"Content-Type": "application/json"},
+                         headers=self._headers,
                          auth=(self.api_key, ''), params=params)
 
         if r.status_code == 200:
@@ -97,7 +103,7 @@ def _postrequest(self, endpoint, payload=None):
         """
         payload = payload or {}
         r = requests.post(SCALE_ENDPOINT + endpoint, json=payload,
-                          headers={"Content-Type": "application/json"},
+                          headers=self._headers,
                           auth=(self.api_key, ''))
 
         if r.status_code == 200:
@@ -212,6 +218,15 @@ def update_project(self, project_name, **kwargs):
         projectdata = self._postrequest('projects/%s/setParams' % project_name, payload=kwargs)
         return projectdata
 
+def _generate_useragent():
+    try:
+        python_version = platform.python_version()
+        os_platform = platform.platform()
+
+        user_agent = '%s/%s Python/%s OS/%s' % (__name__, __version__, python_version, os_platform)
+        return user_agent
+    except:
+        return "scaleapi-python-client"
 
 def _AddTaskTypeCreator(task_type):
     def create_task_wrapper(self, **kwargs):