Merge pull request #21 from ElevenPaths/task/i122703-add_sendToAVDate

(resolves #122703) update Tacyt python sdk and test to include new fi…

Author: Iván Ramos Muñoz

tacyt/TacytApp.py

@@ -120,14 +120,16 @@ def get_RSS_info(self, filter_id):
         """
         return super(TacytApp, self).get_rss_info(filter_id=filter_id)
 
-    def upload_app(self, apk_file, tagName=None):
+    def upload_app(self, apk_file, tagName=None, send_to_av_date=None):
         """Upload an app file to Tacyt
         :param apk_file: path to file apk
         :param tagName: an optional tag to labeled the uploaded app.
+        :param send_to_av_date: an optional date to delay sending apps to third party antivirus like Opswat.
         :return: Response object.
         """
         return super(TacytApp, self).upload_app(app_path=apk_file,
-                                                tag_name=tagName)
+                                                tag_name=tagName,
+                                                send_to_av_date=send_to_av_date)
 
     def getEngineVersion(self, date=None, engineId=None, lang=None):
         """Search an engine and its associated vulnerabilities.

tacyt_sdk/authorization/auth.py

@@ -157,7 +157,7 @@ def http_get(self, url, headers=None):
             response = Response(json_string=res.content)
             res.raise_for_status()
         except requests.HTTPError as e:
-            logger.error(e.message, exc_info=True)
+            logger.error(str(e), exc_info=True)
             if not response.error:
                 response.error = Error({"code": res.status_code,
                                         "message": res.content})
@@ -185,7 +185,7 @@ def http_post(self, url, headers=None, body=None):
             response = Response(json_string=res.content)
             res.raise_for_status()
         except requests.HTTPError as e:
-            logger.error(e.message, exc_info=True)
+            logger.error(str(e), exc_info=True)
             if not response.error:
                 response.error = Error({"code": res.status_code,
                                         "message": res.content})
@@ -239,5 +239,5 @@ def http_post_file(self, url, headers, file_stream, file_name, data=None):
             response = Response(json_string=res.content)
             res.raise_for_status()
         except requests.HTTPError as e:
-            logger.error(e.message, exc_info=True)
+            logger.error(str(e), exc_info=True)
         return response

tacyt_sdk/client.py

@@ -266,10 +266,11 @@ def compare_apps(self, apps, include_details):
         result = CompareRequest(apps, include_details)
         return self.http_post(self.API_COMPARER_URL, body=result.as_dict())
 
-    def upload_app(self, app_path, tag_name=None):
+    def upload_app(self, app_path, tag_name=None, send_to_av_date=None):
         """Upload an app file to Tacyt
         :param app_path: path to file apk
         :param tag_name: an optional tag to labeled the uploaded app.
+        :param send_to_av_date: an optional date to delay sending apps to third party antivirus like Opswat.
         :return: Response object.
         """
         try:
@@ -284,9 +285,10 @@ def upload_app(self, app_path, tag_name=None):
                     headers,
                     file_stream,
                     file_name,
-                    data={"tagName": tag_name})
+                    data={"tagName": tag_name,
+                          "sendToAVDate": send_to_av_date})
         except IOError as e:
-            logger.error(e.message, exc_info=True)
+            logger.error(str(e), exc_info=True)
             response = None
         return response