Update the user agent

Author: MislavReversingLabs

ReversingLabs/SDK/a1000.py

@@ -6,6 +6,7 @@
 """
 
 import datetime
+import inspect
 import requests
 import time
 from urllib import parse
@@ -104,7 +105,6 @@ def __init__(self, host, username=None, password=None, token=None, fields_v2=__F
             token = self.__get_token(username, password)
 
         self._headers = {
-            "User-Agent": self._user_agent,
             "Authorization": "Token {token}".format(token=token)
         }
         self._fields_v2 = fields_v2
@@ -291,7 +291,7 @@ def submit_url_for_analysis(self, url_string, crawler=None, archive_password=Non
 
         response = self.__post_request(
             url=url,
-            data=data,
+            data=data
         )
 
         self.__raise_on_error(response)
@@ -341,7 +341,7 @@ def check_submitted_url_status(self, task_id):
 
         url = self._url.format(endpoint=endpoint)
 
-        response = self.__get_request(url=url)
+        response = self.__get_request(url=url, method_name=inspect.currentframe().f_code.co_name)
 
         self.__raise_on_error(response)
 
@@ -489,7 +489,10 @@ def get_summary_report_v2(self, sample_hashes, retry=True, fields=None, include_
             "skip_reanalysis": str(skip_reanalysis).lower()
         }
 
-        response = self.__post_request(url=url, data=data)
+        response = self.__post_request(
+            url=url,
+            data=data
+        )
 
         self.__raise_on_error(response)
 
@@ -628,7 +631,10 @@ def get_detailed_report_v2(self, sample_hashes, retry=False, fields=None, skip_r
 
             data["include_networkthreatintelligence"] = str(include_networkthreatintelligence).lower()
 
-        response = self.__post_request(url=url, data=data)
+        response = self.__post_request(
+            url=url,
+            data=data
+        )
         self.__raise_on_error(response)
 
         return response
@@ -803,7 +809,10 @@ def reanalyze_samples_v2(self, hash_input, titanium_cloud=False, titanium_core=F
             "rl_cloud_sandbox_platform": rl_cloud_sandbox_platform
         }
 
-        response = self.__post_request(url=url, data=data)
+        response = self.__post_request(
+            url=url,
+            data=data
+        )
 
         self.__raise_on_error(response)
 
@@ -976,15 +985,18 @@ def delete_samples(self, hash_input):
 
             data = {"hash_values": hash_input}
 
-            response = self.__post_request(url=url, data=data)
+            response = self.__post_request(
+                url=url,
+                data=data,
+                method_name=f"{self.__class__.__name__} {inspect.currentframe().f_code.co_name}"
+            )
 
         else:
             raise WrongInputError("hash_input parameter must be a single hash string or "
                                   "a list of hash strings of the same type.")
 
         self.__raise_on_error(response)
 
-
         return response
 
     def check_sample_removal_status_v2(self, task_id):
@@ -1235,7 +1247,10 @@ def set_classification(self, sample_hash, classification, system, risk_score=Non
 
         url = self._url.format(endpoint=endpoint)
 
-        response = self.__post_request(url=url, data=data)
+        response = self.__post_request(
+            url=url,
+            data=data
+        )
 
         self.__raise_on_error(response)
 
@@ -2470,6 +2485,9 @@ def __get_request(self, url, params=None):
             :return: response
             :rtype: requests.Response
         """
+        self._headers["User-Agent"] = (f"{self._user_agent}; {self.__class__.__name__} "
+                                       f"{inspect.currentframe().f_back.f_code.co_name}")
+
         response = requests.get(
             url=url,
             verify=self._verify,
@@ -2492,6 +2510,9 @@ def __post_request(self, url, post_json=None, files=None, data=None, params=None
             :return: response
             :rtype: requests.Response
         """
+        self._headers["User-Agent"] = (f"{self._user_agent}; "
+                                       f"{self.__class__.__name__} {inspect.currentframe().f_back.f_code.co_name}")
+
         response = requests.post(
             url=url,
             json=post_json,
@@ -2512,6 +2533,9 @@ def __delete_request(self, url, post_json=None):
         :return: response
         :rtype: requests.Response
         """
+        self._headers["User-Agent"] = (f"{self._user_agent}; {self.__class__.__name__} "
+                                       f"{inspect.currentframe().f_back.f_code.co_name}")
+
         response = requests.delete(
             url=url,
             json=post_json,