feat(apiclient): allow to specify additional libraries via setUserAgent

Author: KaiSchwarz-cnic

hexonet/apiconnector/apiclient.py

@@ -133,14 +133,18 @@ def getUserAgent(self):
             self.__ua = "%s (%s; %s; rv:%s) python/%s" % (pid, pf, arch, self.getVersion(), pyv)
         return self.__ua
 
-    def setUserAgent(self, pid, rv):
+    def setUserAgent(self, pid, rv, modules=[]):
         """
         Possibility to customize default user agent to fit your needs by given string and revision
         """
+        s = " "
+        mods = ""
+        if len(modules) > 0:
+            mods += " " + s.join(modules)
         pyv = platform.python_version()
         pf = platform.system()
         arch = platform.architecture()[0]
-        self.__ua = "%s (%s; %s; rv:%s) python-sdk/%s python/%s" % (pid, pf, arch, rv, self.getVersion(), pyv)
+        self.__ua = "%s (%s; %s; rv:%s)%s python-sdk/%s python/%s" % (pid, pf, arch, rv, mods, self.getVersion(), pyv)
         return self
 
     def getVersion(self):

tests/test_apiclient.py

@@ -141,6 +141,12 @@ def test_apiclientmethods():
     assert isinstance(cl2, AC) is True
     assert cl.getUserAgent() == ua
 
+    mods = ["reg/2.6.2", "ssl/7.2.2", "dc/8.2.2"]
+    ua = "%s (%s; %s; rv:%s) %s %s/%s python/%s" % (pid, pf, arch, rv, " ".join(mods), pid2, cl.getVersion(), pyv)
+    cl2 = cl.setUserAgent(pid, rv, mods)
+    assert isinstance(cl2, AC) is True
+    assert cl.getUserAgent() == ua
+
     # #.setURL()
     tmp = ISPAPI_CONNECTION_URL_PROXY
     url = cl.setURL(tmp).getURL()